Distrax

github.comtool

Reimplementation of TensorFlow Probability, containing probability distributions and bijectors.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Equinox github.com

   tool category #jax link ↗

   Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.

   
2. ▲
   JaxDF github.com

   tool category #jax link ↗

   Framework for differentiable simulators with arbitrary discretizations.

   
3. ▲
   JAX RL github.com

   tool category #jax link ↗

   Implementations of reinforcement learning algorithms.

   
4. ▲
   Diffrax github.com

   tool category #jax link ↗

   Numerical differential equation solvers in JAX.

5. ▲
   OTT-JAX github.com

   tool category #jax link ↗

   Optimal transport tools in JAX.

   
6. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

   

JAX › Libraries: “Reimplementation of TensorFlow Probability, containing probability distributions and bijectors.”