PyPI page
Home page
Author:
None
Summary:
Distrax: Probability distributions in JAX.
Latest version:
0.1.9
Required dependencies:
absl-py
|
chex
|
jax
|
jaxlib
|
numpy
|
tfp-nightly
Optional dependencies:
dm-haiku
|
flax
|
mock
|
optax
|
tensorflow
|
tensorflow-datasets
Downloads last day:
16,869
Downloads last week:
65,022
Downloads last month:
218,680