33 dependents
| Package | Description | Downloads/month |
|---|---|---|
| A library of reinforcement learning building blocks in JAX. | 37K | |
| Python library for RAD Embeddings, provably correct latent DFA representations. | 3K | |
| A Jax package for approximate curvature estimation and optimization using KFAC. | 3K | |
| Machine learning assisted marginal likelihood (Bayesian evidence) estimation for... | 2K | |
| A library containing a collection of distance and similarity measures for data a... | 2K | |
| Surjection layers for density estimation with normalizing flows | 2K | |
| A research-focused modular generative modeling library built on JAX/Flax NNX | 1K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| JAX implementation of Reinforcement Learning agents | 1K | |
| Opifex Computational Framework - JAX-native platform for scientific machine lear... | 903 | |
| Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral... | 850 | |
| A reinforcement learning environment for simulating target-MDP environments (Air... | 796 | |
| Vectorizable RL algorithms in pure JAX | 637 | |
| Towards Embodied AI with MuscleMimic: Unlocking full-body musculoskeletal motor ... | 485 | |
| Robotic Environments with jaX (REX) | 467 | |
| LEM slope stability analysis using JAX | 334 | |
| Imitation learning benchmark focusing on complex locomotion tasks using MuJoCo. | 287 | |
| TargetGym: Fast Setpoint tracking environments in Jax. | 273 | |
| This package provides tools to execute and implement Implicit Likelihood Inferen... | 255 | |
| Bayesian Data Analysis using PyMC and JAX | 255 | |
| Diffusion meets sampling | 222 | |
| A unified JAX/Flax framework for memory-augmented reinforcement learning with su... | 219 | |
| Python Boilerplate that contains all the code you need to create a Python packag... | 206 | |
| Queuinx: A library for performance evaluation in Jax | 175 | |
| A library for game-theoretic evaluation and ratings. | 154 | |
| Reinforcement learning in pure JAX. | 151 | |
| Automatically created RL Environment Editors | 144 | |
| Flow RL is a high-performance RL library with flow and diffusion models. | 120 | |
| Extensible offline reinforcement learning framework built with JAX, Flax, and Op... | 112 | |
| Partially Observable Benchmarks in JAX | 90 | |
| A library for learning the ground state energy using orbital free density functi... | 71 | |
| Synthetic gymnax environments | 69 | |
| Add your description here | 6 |
