PySpect
Home
Lists
Top stars
Rising stars
← Back to distrax
Dependencies for distrax
Frequently asked questions
© 2025-2026 PySpect
Dependencies of distrax (12).
Dependency
Optional
Group
absl-py
false
chex
false
jax
false
jaxlib
false
Dependent packages
Packages depending on distrax (1).
Package
Optional
Group
rlax
false
numpy
false
tfp-nightly
false
dm-haiku
true
examples
flax
true
examples
mock
true
test
optax
true
examples
tensorflow
true
examples
tensorflow-datasets
true
examples