© 2025-2026 PySpect
| Package | Optional | Group |
|---|---|---|
| brax | false | |
| clu | false | |
| dopamine-rl | false | |
| evojax | false | |
| evosax | false | |
| gymnax | false | |
| tensorflowjs | false | |
| transformer-engine-jax | false | |
| causalml | true | jax |
| crfm-helm | true | heim |
| deepmd-kit | true | jax |
| diffusers | true | flax |
| distrax | true | examples |
| dm-haiku | true | flax |
| e3nn-jax | true | dev |
| fiddle | true | testing |
| google-vizier | true | jax |
| gymnasium | true | jax |
| optax | true | docs |
| orbax-checkpoint | true | docs |
| outlines | true | test |
| penzai | true | docs |
| rlax | true | test |
| safetensors | true | jax |
| tfx | true | all |
| tyro | true | dev-nn |
| orbax-checkpoint |
| false |
| pyyaml | false |
| rich | false |
| tensorstore | false |
| treescope | false |
| typing-extensions | false |
| ale-py | true | testing |
| cloudpickle | true | testing |
| clu | true | testing |
| dm-haiku | true | docs |
| docutils | true | docs |
| einops | true | testing |
| grain | true | testing |
| gymnasium | true | testing |
| ipykernel | true | docs |
| ipython-genutils | true | docs |
| ipywidgets | true | docs |
| jaxlib | true | testing |
| jaxtyping | true | testing |
| jraph | true | testing |
| jupytext | true | docs |
| kagglehub | true | docs |
| keras | true | testing |
| matplotlib | true | docs |
| ml-collections | true | docs |
| mypy | true | testing |
| myst-nb | true | docs |
| nanobind | true | dev |
| nbstripout | true | docs |
| opencv-python | true | testing |
| orbax-export | true | docs |
| pre-commit | true | dev |
| protobuf | true | testing |
| pygments | true | docs |
| pytest | true | testing |
| pytest-cov | true | testing |
| pytest-custom-exit-code | true | testing |
| pytest-xdist | true | testing |
| pytype | true | testing |
| recommonmark | true | docs |
| scikit-build-core | true | dev |
| scikit-learn | true | docs |
| sentencepiece | true | testing |
| sphinx | true | docs |
| sphinx-book-theme | true | docs |
| sphinx-design | true | docs |
| tensorboard | true | testing |
| tensorflow | true | testing |
| tensorflow-datasets | true | testing |
| tensorflow-text | true | testing |
| torch | true | testing |
| tqdm | true | docs |