Google DeepMind

All

286 repositories

torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•32•364•6•8•Updated Nov 14, 2024Nov 14, 2024
kfac-jax
Public
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
machine-learning optimization bayesian-deep-learning
Python
•
Apache License 2.0
•22•248•9•6•Updated Nov 14, 2024Nov 14, 2024
mujoco_menagerie
Public
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
robotics mujoco
Jupyter Notebook
•
Other
•198•1.4k•13•9•Updated Nov 14, 2024Nov 14, 2024
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•819•8.2k•73•34•Updated Nov 14, 2024Nov 14, 2024
gemma
Public
Open weights LLM from Google DeepMind.
Python
•
Apache License 2.0
•310•2.5k•8•15•Updated Nov 14, 2024Nov 14, 2024
android_env
Public
RL research on Android devices.
android reinforcement-learning
Python
•
Apache License 2.0
•75•1k•0•8•Updated Nov 14, 2024Nov 14, 2024
dm_control
Public
Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
machine-learning reinforcement-learning deep-learning artificial-intelligence neural-networks physics-simulation mujoco
Python
•
Apache License 2.0
•669•3.8k•93•8•Updated Nov 14, 2024Nov 14, 2024
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•480•4.6k•6•1•Updated Nov 14, 2024Nov 14, 2024
mujoco_mpc
Public
Real-time behaviour synthesis with MuJoCo, using Predictive Control
mpc model-predictive-control mujoco mpc-control predictive-control motor-control
C++
•
Apache License 2.0
•156•1k•14•12•Updated Nov 14, 2024Nov 14, 2024
synthid-text
Public
Python
•
Apache License 2.0
•25•343•1•0•Updated Nov 14, 2024Nov 14, 2024
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•162•675•4•4•Updated Nov 14, 2024Nov 14, 2024
nuclease_design
Public
ML-guided enzyme engineering
learning engineering machine protein
Jupyter Notebook
•
Apache License 2.0
•11•34•0•0•Updated Nov 14, 2024Nov 14, 2024
sonnet
Public
TensorFlow-based neural network library
machine-learning deep-learning artificial-intelligence neural-networks tensorflow
Python
•
Apache License 2.0
•1.3k•9.8k•32•6•Updated Nov 14, 2024Nov 14, 2024
evergreen-spec
Public
Apache License 2.0
•0•5•0•0•Updated Nov 13, 2024Nov 13, 2024
chex
Public
Python
•
Apache License 2.0
•44•788•21•19•Updated Nov 13, 2024Nov 13, 2024
clrs
Public
Jupyter Notebook
•
Apache License 2.0
•85•449•4•6•Updated Nov 13, 2024Nov 13, 2024
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•193•1.7k•33•19•Updated Nov 12, 2024Nov 12, 2024
dks
Public
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.
machine-learning tensorflow artificial-intelligence neural-networks jax neural-tangent-kernel deep-learning pytorch
Python
•
Apache License 2.0
•4•64•0•2•Updated Nov 12, 2024Nov 12, 2024
envlogger
Public
A tool for recording RL trajectories.
reinforcement-learning logging
Python
•
Apache License 2.0
•13•94•0•0•Updated Nov 12, 2024Nov 12, 2024
dm_pix
Public
PIX is an image processing library in JAX, for JAX.
machine-learning image computer-vision image-processing jax python
Python
•
Apache License 2.0
•23•389•1•0•Updated Nov 8, 2024Nov 8, 2024
dm-haiku
Public
JAX-based neural network library
machine-learning deep-neural-networks deep-learning neural-networks jax
Python
•
Apache License 2.0
•232•2.9k•74•18•Updated Nov 8, 2024Nov 8, 2024
webquest
Public
Apache License 2.0
•0•0•0•0•Updated Nov 6, 2024Nov 6, 2024
tabula_rasa_agents
Public
Jupyter Notebook
•
Apache License 2.0
•0•0•0•0•Updated Nov 4, 2024Nov 4, 2024
transformer_ngrams
Public
Jupyter Notebook
•
Apache License 2.0
•1•9•0•0•Updated Nov 4, 2024Nov 4, 2024
tf2jax
Public
Python
•
Apache License 2.0
•11•105•0•12•Updated Nov 4, 2024Nov 4, 2024
wikidomains
Public
Creative Commons Attribution Share Alike 4.0 International
•0•2•0•0•Updated Nov 4, 2024Nov 4, 2024
reverb
Public
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
C++
•
Apache License 2.0
•93•706•17•7•Updated Nov 4, 2024Nov 4, 2024
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•121•617•7•3•Updated Nov 4, 2024Nov 4, 2024
treescope
Public
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
visualization machine-learning neural-networks ndarray pretty-printing
Python
•
Apache License 2.0
•17•335•7•2•Updated Nov 1, 2024Nov 1, 2024
penzai
Public
A JAX research toolkit for building, editing, and visualizing neural networks.
visualization neural-networks interpretability fine-tuning jax
Python
•
Apache License 2.0
•52•1.7k•3•0•Updated Nov 1, 2024Nov 1, 2024