Clément Bonnet

Steve. Cursor for Minecraft

jax-llm-examples. Minimal yet performant LLM examples in pure JAX

ARC-AGI-2.

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

lang-jepa. Python

synalinks-skills. Claude skills for Synalinks OSS

mlp.c. simple implementation of an MLP model in pure C

Kinetix. Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.

flax. Flax is a neural network library for JAX that is designed for flexibility.

cartpole.c. Simple example of developing an RL environment in pure C

tchoupy. Time-saving tool for unit conversions and dimensional quantity computations

alien. ALIEN is a CUDA-powered artificial life simulation program.

Craftax. (Crafter + NetHack) in JAX. ICML 2024 Spotlight.

cogment-lab. A toolkit for practical Human-AI cooperation research

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

outer-value-function-meta-rl. Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function

refiners. A microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation

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

jaxprvis. Python

ARC-AGI. The Abstraction and Reasoning Corpus

pgx. ♟️ Vectorized RL game environments in JAX

nucleotide-transformer. Foundation Models for Genomics & Transcriptomics

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

brax. Massively parallel rigidbody physics simulation on accelerator hardware.

poppy. :hibiscus: Population-Based Reinforcement Learning for Combinatorial Optimization

catx. 🐈‍⬛ Contextual bandits library for continuous action trees with smoothing in JAX

gymnax. RL Environments in JAX 🌍

QDax. Accelerated Quality-Diversity

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

bio-datasets. Free collection of Bio datasets and embeddings

deepchain-apps. A library for deploying App on deepchain.bio

bio-transformers. bio-transformers is a wrapper on top of the ESM/Protbert model, trained on millions on proteins and used to predict embeddings.

jax. Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

transformers. 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

rl-sample-efficiency. Implementation of Reinforcement Learning algorithms in Python

nlp-tutorial. Natural Language Processing Tutorial for Deep Learning Researchers

reinforcement-learning. Implementation of Reinforcement Learning Algorithms. Python, OpenAI Gym, Tensorflow. Exercises and Solutions to accompany Sutton's Book and David Silver's course.

awesome-rl. Reinforcement learning resources curated

DeepLearningFrameworks. Demo of running NNs across different frameworks