Package profile
distrax
- Summary: Distrax: Probability distributions in JAX.
- Author: Google DeepMind <distrax-dev@google.com>
- License: Apache Software License
- Homepage: https://github.com/google-deepmind/distrax
- Source: https://github.com/google-deepmind/distrax
- Number of releases: 13
- First release: 0.0.1 on 2021-04-13
- Latest release: 0.1.9 on 2026-06-12
- Latest release size: 306.1 KB (pure Python wheel)
Dependencies
Distrax has 12 dependencies, 6 of which optional.Dependent packages
| Package | Optional | Group |
|---|---|---|
| rlax | false |
Similar packages
- optaxA gradient processing and optimization library in JAX.
- chexChex: Testing made fun, in JAX!
- cluSet of libraries for ML training loops in JAX.
- flaxFlax: A neural network library for JAX designed for flexibility
- jaxlibXLA library for JAX
- pybloom-liveBloom filter: A Probabilistic data structure
- httpstanHTTP-based interface to Stan, a package for Bayesian inference.