© 2025-2026 PySpect
| Package | Optional | Group |
|---|---|---|
| brax | false | |
| chex | false | |
| clu | false | |
| distrax | false | |
| dopamine-rl | false | |
| e3nn-jax | false | |
| evojax | false | |
| jax | false | |
| jaxite | false | |
| jaxns | false | |
| jaxopt | false | |
| jraph | false | |
| mujoco-mjx | false | |
| numpyro | false | |
| optax | false | |
| orbax-export | false | |
| rlax | false | |
| sella | false | |
| seqio | false | |
| seqio-nightly | false | |
| tensorflowjs | false | |
| botorch | true | fully-bayesian |
| bsuite | true | baselines-jax |
| crfm-helm | true | heim |
| datasets | true | jax |
| deepchem | true | jax |
| diffusers | true | flax |
| dm-haiku | true | jax |
| flax | true | testing |
| funsor | true | jax |
| google-vizier | true | jax |
| gymnasium | true | jax |
| iminuit | true | doc |
| jmp | true | jax |
| pot | true | backend-jax |
| pykeops | true | full |
| pymanopt | true | jax |
| pymbar | true | jax |
| pytensor | true | jax |
| safetensors | true | jax |
| scalene | true | test |
| tensorflow-probability | true | jax |
| tfp-nightly | true | jax |
| tfx | true | all |
| underthesea | true | voice |