PyPI page
Home page
Author:
None
Summary:
Distrax: Probability distributions in JAX.
Latest version:
0.1.8
Required dependencies:
absl-py
|
chex
|
jax
|
jaxlib
|
numpy
|
tfp-nightly
Optional dependencies:
dm-haiku
|
flax
|
mock
|
optax
|
tensorflow
|
tensorflow-datasets
Downloads last day:
6,853
Downloads last week:
43,062
Downloads last month:
192,197