lockwo

lockwo/distreqx

Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.

47 stars 9 forks 47 watchers Python Apache License 2.0

jax machine-learning statistics

View on GitHub Website

3 Open Issues Need Help Last updated: Jun 20, 2026

Open Issues Need Help

View All on GitHub

Bijector Checklist and Progress 2 days ago

enhancement help wanted

lockwo/distreqx

47

Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.

Python

#jax#machine-learning#statistics

Distribution Checklist and Progress 3 days ago

enhancement help wanted

lockwo/distreqx

47

Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.

Python

#jax#machine-learning#statistics

Improve tests 9 months ago

enhancement help wanted

lockwo/distreqx

47

Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.

Python

#jax#machine-learning#statistics