AlphaFold 3 inference pipeline.
Python 8.6k 1.4k
Multi-Joint dynamics with Contact. A general purpose physics simulator.
C++ 15.5k 1.8k
Open source code for AlphaFold 2.
Python 14.9k 3k
A library of reinforcement learning components and agents
Python 4.1k 553
JAX-based neural network library
Python 3.3k 302
This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook 15.2k 2.9k
TORAX: Tokamak transport simulation in JAX
A collection of formalized statements of conjectures in Lean.
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Real-time behaviour synthesis with MuJoCo, using Predictive Control
A library for generative social simulation
PIX is an image processing library in JAX, for JAX.