Pinned Loading

1. jumanji jumanji Public

   🕹️ A diverse suite of scalable reinforcement learning environments in JAX

   Python 728 91

2. Mava Mava Public

   🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

   Python 795 105

3. nucleotide-transformer nucleotide-transformer Public

   🧬 Nucleotide Transformer: Building and Evaluating Robust Foundation Models for Human Genomics

   Python 612 70

4. flashbax flashbax Public

   ⚡ Flashbax: Accelerated Replay Buffers in JAX

   Python 236 19

5. sebulba sebulba Public

   🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

   Python 57 5

6. manyfold manyfold Public

   🧬 ManyFold: An efficient and flexible library for training and validating protein folding models

   Python 81 9