Pinned Loading

1. Stoix Stoix Public

   🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

   Python 422 52

2. instadeepai/flashbax instadeepai/flashbax Public

   ⚡ Flashbax: Accelerated Replay Buffers in JAX

   Python 281 22

3. facebookresearch/aira-dojo facebookresearch/aira-dojo Public

   AIRA-dojo: a framework for developing and evaluating AI research agents

   Python 167 36

4. Stoa Stoa Public

   A JAX-Native Interface for Reinforcement Learning Environments

   Python 10 5