piwheels

distrax

Distrax: Probability distributions in JAX.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install distrax

Dependencies

• tfp-nightly
• jaxlib
• chex
• jax
• absl-py
• numpy

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.7	2025-09-01
distrax-0.1.7-py3-none-any.whl (305 KB)
0.1.6 yanked	2025-09-01
distrax-0.1.6-py3-none-any.whl (314 KB)
0.1.5	2023-11-22
distrax-0.1.5-py3-none-any.whl (312 KB)
0.1.4	2023-06-29
distrax-0.1.4-py3-none-any.whl (311 KB)
0.1.3	2023-02-14
distrax-0.1.3-py3-none-any.whl (311 KB)
0.1.2	2022-03-25
distrax-0.1.2-py3-none-any.whl (266 KB)
0.1.1	2022-02-25
distrax-0.1.1-py3-none-any.whl (237 KB)
0.1.0	2021-11-18
distrax-0.1.0-py3-none-any.whl (216 KB)
0.0.3	2021-11-18
distrax-0.0.3-py3-none-any.whl (216 KB)
0.0.2	2021-08-03
distrax-0.0.2-py3-none-any.whl (203 KB)
0.0.1	2021-04-13
distrax-0.0.1-py3-none-any.whl (192 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue