Google DeepMind

All

317 repositories

torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•50•416•36•7•Updated Apr 8, 2025Apr 8, 2025
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•227•1.9k•34•15•Updated Apr 8, 2025Apr 8, 2025
fusion_surrogates
Public
A library of surrogate transport models for tokamak fusion.
Python
•
Apache License 2.0
•1•4•0•2•Updated Apr 8, 2025Apr 8, 2025
gemma
Public
Gemma open-weight LLM library, from Google DeepMind
Jupyter Notebook
•
Apache License 2.0
•425•3.1k•30•38•Updated Apr 8, 2025Apr 8, 2025
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
•691•4k•102•8•Updated Apr 8, 2025Apr 8, 2025
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•952•9.2k•102•45•Updated Apr 8, 2025Apr 8, 2025
tf2jax
Public
Python
•
Apache License 2.0
•12•114•0•13•Updated Apr 8, 2025Apr 8, 2025
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•180•833•6•1•Updated Apr 8, 2025Apr 8, 2025
mujoco_warp
Public
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
Python
•
Apache License 2.0
•32•322•49•26•Updated Apr 8, 2025Apr 8, 2025
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
•60•1.8k•4•1•Updated Apr 8, 2025Apr 8, 2025
tapnet
Public
Tracking Any Point (TAP)
benchmark computer-vision deep-learning robotics point-tracking
Jupyter Notebook
•
Apache License 2.0
•137•1.4k•25•1•Updated Apr 7, 2025Apr 7, 2025
disentangled_rnns
Public
Python
•
Apache License 2.0
•3•13•0•4•Updated Apr 7, 2025Apr 7, 2025
loft
Public
LOFT: A 1 Million+ Token Long-Context Benchmark
Python
•
Apache License 2.0
•14•184•1•1•Updated Apr 7, 2025Apr 7, 2025
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
•978•4.5k•16•11•Updated Apr 7, 2025Apr 7, 2025
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•267•10•7•Updated Apr 7, 2025Apr 7, 2025
physics-IQ-benchmark
Public
Benchmarking physical understanding in generative video models
benchmark generative-models video-generation physical-understanding
Python
•
Other
•12•145•2•0•Updated Apr 4, 2025Apr 4, 2025
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•134•677•8•0•Updated Apr 3, 2025Apr 3, 2025
chex
Public
Python
•
Apache License 2.0
•49•843•23•23•Updated Apr 3, 2025Apr 3, 2025
questbench
Public
Python
•
Apache License 2.0
•0•5•0•0•Updated Apr 2, 2025Apr 2, 2025
mishax
Public
Python
•
Apache License 2.0
•8•127•0•0•Updated Mar 31, 2025Mar 31, 2025
mujoco_playground
Public
An open-source library for GPU-accelerated robot learning and sim-to-real transfer.
Jupyter Notebook
•
Apache License 2.0
•104•896•22•3•Updated Mar 31, 2025Mar 31, 2025
distrax
Public
Python
•
Apache License 2.0
•32•563•22•20•Updated Mar 28, 2025Mar 28, 2025
multiscope
Public
Real-time visualisation
Go
•
Apache License 2.0
•2•17•0•6•Updated Mar 27, 2025Mar 27, 2025
personality_in_llms
Public
Jupyter Notebook
•
Apache License 2.0
•2•6•1•0•Updated Mar 27, 2025Mar 27, 2025
simply
Public
1•2•0•0•Updated Mar 24, 2025Mar 24, 2025
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•788•6.3k•8•1•Updated Mar 24, 2025Mar 24, 2025
deepmind-research
Public
This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook
•
Apache License 2.0
•2.7k•14k•193•83•Updated Mar 21, 2025Mar 21, 2025
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
•5•70•0•2•Updated Mar 21, 2025Mar 21, 2025
unlearning_evaluation
Public
Python
•
Apache License 2.0
•5•16•0•1•Updated Mar 21, 2025Mar 21, 2025
clrs
Public
Jupyter Notebook
•
Apache License 2.0
•100•478•5•5•Updated Mar 20, 2025Mar 20, 2025