PyPI page
Author:
Google DeepMind
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:
10,506
Downloads last week:
60,106
Downloads last month:
163,583
7-day smoothing