17 projects
id-echo
A fast distributed replay buffer for reinforcement learning.
id-mava
A Python library for Multi-Agent Reinforcement Learning.
id-marl-eval
A Python library for Multi-Agent Reinforcement Learning evaluation.
matrax
Matrix games in JAX
jumanji
A diverse suite of scalable reinforcement learning environments in JAX
stix-ml
A unifying framework for flows and diffusions using stochastic interpolants
catx
Contextual Bandits with Continuous Actions in JAX
mlip
Machine Learning Interatomic Potentials in JAX
alf_tools
A Python package with tools for performing active learning experiments.
instanovo
InstaNovo enables diffusion-powered de novo peptide sequencing in large scale proteomics experiments
e3j
Fast Euclid equivariant operations for GPU and TPU
chemreporter
Python tool for flexible and reproducible curation of MLIP datasets.
alf_core
A Python package for performing active learning experiments.
mlipaudit
Library and CLI tool for benchmarking ML Interatomic Potentials
flashbax
Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
winnow-fdr
Score calibration and false discovery estimation for de novo peptide sequencing.
e3j-ops
Fast Euclid equivariant operations for JAX
