Open source code for AlphaFold.
Python 12.2k 2.2k
Multi-Joint dynamics with Contact. A general purpose physics simulator.
Jupyter Notebook 7.6k 751
TensorFlow-based neural network library
Python 9.7k 1.3k
A library of reinforcement learning components and agents
Python 3.4k 423
JAX-based neural network library
Python 2.8k 230
This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook 13k 2.6k
TORAX: Tokamak transport simulation in JAX
A JAX research toolkit for building, editing, and visualizing neural networks.
Optax is a gradient processing and optimization library for JAX.
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
A tool for recording RL trajectories.
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Monte Carlo tree search in JAX
A library for generative social simulation