Stars
L2O/NCO codes from CIAM Group at SUSTech, Shenzhen, China
For our ACL25 Paper: Can Language Models Replace Programmers? RepoCod Says ‘Not Yet’ - by Shanchao Liang and Yiran Hu and Nan Jiang and Lin Tan
An implementation of mLSTM from xLSTM in JAX
Basic tabular Q-learning agent built, run, visualised and plotted using DeepMind's AI Safety gridworld environments.
Data and starter notebooks for the locust breeding ground prediction hackaton
⚡ Flashbax: Accelerated Replay Buffers in JAX
🧭 COMPASS: Combinatorial Optimization with Policy Adaptation using Latent Space Search
Datasets with baselines for offline multi-agent reinforcement learning.
🧬 ManyFold: An efficient and flexible library for training and validating protein folding models
Accelerated Quality-Diversity
Foundation Models for Genomics & Transcriptomics
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX