Google DeepMind

All

292 repositories

mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•849•8.4k•79•33•Updated Dec 24, 2024Dec 24, 2024
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•37•385•37•5•Updated Dec 24, 2024Dec 24, 2024
long-form-factuality
Public
Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
language benchmark metrics evaluation language-modeling dataset factuality large-language-models
Python
•
Other
•62•567•2•10•Updated Dec 23, 2024Dec 23, 2024
mathematics_dataset
Public
This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.
Python
•
Apache License 2.0
•250•1.8k•2•0•Updated Dec 23, 2024Dec 23, 2024
iris
Public
Python
•
Apache License 2.0
•3•6•0•3•Updated Dec 23, 2024Dec 23, 2024
gemma
Public
Open weights LLM from Google DeepMind.
Python
•
Apache License 2.0
•323•2.5k•7•20•Updated Dec 23, 2024Dec 23, 2024
mishax
Public
Python
•
Apache License 2.0
•5•109•0•0•Updated Dec 20, 2024Dec 20, 2024
alphafold
Public
Open source code for AlphaFold 2.
Python
•
Apache License 2.0
•2.3k•13k•254•28•Updated Dec 20, 2024Dec 20, 2024
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•668•5.7k•11•2•Updated Dec 20, 2024Dec 20, 2024
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•197•1.7k•37•23•Updated Dec 19, 2024Dec 19, 2024
clrs
Public
Jupyter Notebook
•
Apache License 2.0
•94•460•4•6•Updated Dec 19, 2024Dec 19, 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
•24•254•9•6•Updated Dec 19, 2024Dec 19, 2024
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•169•737•5•3•Updated Dec 19, 2024Dec 19, 2024
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•124•639•7•0•Updated Dec 19, 2024Dec 19, 2024
alphatensor_quantum
Public
Python
•
Apache License 2.0
•2•5•0•0•Updated Dec 19, 2024Dec 19, 2024
proactive_t2i_agents
Public
Code release for the paper, "Proactive Agents for Text-to-Image Generation under Uncertainty"
1•5•1•0•Updated Dec 18, 2024Dec 18, 2024
android_env
Public
RL research on Android devices.
android reinforcement-learning
Python
•
Apache License 2.0
•79•1k•1•8•Updated Dec 17, 2024Dec 17, 2024
graphcast
Public
weather weather-forecast
Python
•
Apache License 2.0
•679•5.5k•38•4•Updated Dec 17, 2024Dec 17, 2024
dm-haiku
Public
JAX-based neural network library
machine-learning deep-neural-networks deep-learning neural-networks jax
Python
•
Apache License 2.0
•233•2.9k•72•17•Updated Dec 17, 2024Dec 17, 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
•169•1.1k•21•12•Updated Dec 17, 2024Dec 17, 2024
tf2jax
Public
Python
•
Apache License 2.0
•12•109•0•13•Updated Dec 16, 2024Dec 16, 2024
open_spiel
Public
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
cpp multiagent python games reinforcement-learning
C++
•
Apache License 2.0
•942•4.3k•25•7•Updated Dec 16, 2024Dec 16, 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
•677•3.9k•93•8•Updated Dec 16, 2024Dec 16, 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
•55•1.7k•2•1•Updated Dec 16, 2024Dec 16, 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
•18•362•7•0•Updated Dec 16, 2024Dec 16, 2024
distrax
Public
Python
•
Apache License 2.0
•33•540•22•19•Updated Dec 14, 2024Dec 14, 2024
synthid-text
Public
Python
•
Apache License 2.0
•28•402•2•0•Updated Dec 13, 2024Dec 13, 2024
md4
Public
Code for https://arxiv.org/abs/2406.04329
Python
•
Apache License 2.0
•1•42•0•0•Updated Dec 11, 2024Dec 11, 2024
enn
Public
Python
•
Apache License 2.0
•59•299•8•8•Updated Dec 11, 2024Dec 11, 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
•24•396•1•1•Updated Dec 11, 2024Dec 11, 2024