Google DeepMind
Change the repository type filter
All
Repositories list
286 repositories
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.
- A library for generative social simulation
- Real-time behaviour synthesis with MuJoCo, using Predictive Control
- Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
- TensorFlow-based neural network library
- Optax is a gradient processing and optimization library for JAX.
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
- PIX is an image processing library in JAX, for JAX.
- JAX-based neural network library
reverb
Public- A suite of test scenarios for multi-agent reinforcement learning.
treescope
PublicAn interactive HTML pretty-printer for machine learning research in IPython notebooks.- A JAX research toolkit for building, editing, and visualizing neural networks.