Reverse Dependencies of jax-cuda12-plugin
The following projects have a declared dependency on jax-cuda12-plugin:
- jax — Differentiate, compile, and transform Numpy code.
- jimmy-vision — Implementations of popular vision models in Jax (Flax NNX)
1