21 dependents
| Package | Description | Downloads/month |
|---|---|---|
| JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library | 8K | |
| JAX implementation of Reinforcement Learning agents | 1K | |
| Fast, lightweight and parallelised simulation-based inference in JAX. | 894 | |
| Spyx: Spiking Neural Networks in JAX | 851 | |
| Recourse Explanation Library in JAX | 802 | |
| jax for quantum simulations | 648 | |
| Higher-Order Interactions | 454 | |
| Small jax-based python package for computing error in Monte Carlo approximate po... | 405 | |
| Environments for Generative Flow Networks in JAX | 357 | |
| Gravitational-wave astronomy in JAX | 340 | |
| LEM slope stability analysis using JAX | 334 | |
| Bayesian Data Analysis using PyMC and JAX | 255 | |
| 1D3V Particle in Cell Code in JAX | 235 | |
| A Bayesian version of Simultaneous Truth and Performance Level Estimation (STAPL... | 230 | |
| SPECTRAX: Hermite-Fourier Vlasov solver in JAX | 181 | |
| A research-oriented package for diffusion-based generative modeling with modular... | 139 | |
| Gauss-Legendre 6th-order implicit integrator and finite-difference helpers built... | 117 | |
| Walk through phase-space observations | 115 | |
| A general PDMP sampler in JAX | 103 | |
| Affine Invariant HMC and Nested Sampling in JAX | 30 | |
| Affine Invariant HMC and Nested Sampling in JAX | 1 |
