Reverse Dependencies of distrax

The following projects have a declared dependency on distrax:

• craftax — An open-world environment for training RL agents
• dopamax — Reinforcement learning in pure JAX.
• flashbax — Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
• gpjax-nightly — Didactic Gaussian processes in Jax.
• harmonic — Python package for efficient Bayesian evidence computation
• jaxfss — JAX/Flax implementation of finite-size scaling
• jaxkern-nightly — Kernels in Jax.
• jaxmarl — Multi-Agent Reinforcement Learning with JAX
• JaxUtils — Utility functions for JaxGaussianProcesses
• jaxutils-nightly — Utility functions for JaxGaussianProcesses
• kfac-jax — A Jax package for approximate curvature estimation and optimization using KFAC.
• modularbayes — Modular Bayesian Inference.
• MOGPJax — Didactic multi-output Gaussian processes in Jax.
• momaland — A standard API for Multi-Objective Multi-Agent Decision making and a diverse set of reference environments.
• probabilistic-reconciliation — Probabilistic reconciliation of time series forecasts
• queuinx — Queuinx: A library for performance evaluation in Jax
• rlax — A library of reinforcement learning building blocks in JAX.
• robo-transformers — RT-1, RT-1-X, Octo Robotics Transformer Model Inference
• sbijax — Simulation-based inference in JAX
• softclip — JAX/Flax implementation of softclip
• surjectors — Surjection layers for density estimation with normalizing flows
• xminigrid — JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid