Library for efficient training and application of Machine Learning Interatomic Potentials (MLIP)

A LLM-friendly framework for translating dynamical equations to gymnasium-compatible RL environments.

Meta in-context learning for protein fitness prediction

A Framework for Comparing N Hyperparameter Optimizers on M Benchmarks.

Supporting code for our paper "Protein Sequence Modelling with Bayesian Flow Networks"

Official code repository for the paper "ProteinNPT: Improving Protein Property Prediction and Design with Non-Parametric Transformers"

A tool for aggregating and plotting MARL experiment data.

⚡ Flashbax: Accelerated Replay Buffers in JAX

Datasets with baselines for Offline MARL.

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

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

Companion webpage for the book "Bayesian Optimization" by Roman Garnett

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks