DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
Python 2.8k 510
TensorFlow-based neural network library
Python 9.3k 1.3k
This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook 10k 2.1k
A library of reinforcement learning components and agents
Python 2.6k 323
JAX-based neural network library
Python 1.8k 148
Open source code for AlphaFold.
Python 8.1k 1.3k
Optax is a gradient processing and optimization library for JAX.
tree is a library for working with nested data structures