Google DeepMind

All

358 repositories

torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•65•475•29•24•Updated Sep 13, 2025Sep 13, 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
•28•283•13•11•Updated Sep 12, 2025Sep 12, 2025
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
C++
•
Apache License 2.0
•1.1k•11k•123•76•Updated Sep 12, 2025Sep 12, 2025
mujoco_warp
Public
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
nvidia-warp
Python
•
Apache License 2.0
•72•719•39•13•Updated Sep 12, 2025Sep 12, 2025
regress-lm
Public
Library for text-to-text regression, applicable to any input string representation and allows pretraining and fine-tuning over multiple regression tasks.
Python
•
Apache License 2.0
•30•218•1•4•Updated Sep 12, 2025Sep 12, 2025
fusion_surrogates
Public
A library of surrogate transport models for tokamak fusion.
Python
•
Apache License 2.0
•1•11•1•2•Updated Sep 11, 2025Sep 11, 2025
formal-conjectures
Public
A collection of formalized statements of conjectures in Lean.
formal-mathematics lean4
Lean
•
Apache License 2.0
•77•620•199•31•Updated Sep 11, 2025Sep 11, 2025
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•214•1k•3•4•Updated Sep 11, 2025Sep 11, 2025
alphagenome
Public
This API provides programmatic access to the AlphaGenome model developed by Google DeepMind.
Python
•
Apache License 2.0
•141•1.2k•4•0•Updated Sep 11, 2025Sep 11, 2025
gemma
Public
Gemma open-weight LLM library, from Google DeepMind
Python
•
Apache License 2.0
•530•3.7k•47•54•Updated Sep 10, 2025Sep 10, 2025
unpuzzles_and_simple_reasoning
Public
Python
•
Apache License 2.0
•0•0•0•0•Updated Sep 9, 2025Sep 9, 2025
multiscope
Public
Real-time visualisation
Go
•
Apache License 2.0
•5•20•0•7•Updated Sep 9, 2025Sep 9, 2025
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•255•2k•33•23•Updated Sep 8, 2025Sep 8, 2025
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•934•7k•19•1•Updated Sep 8, 2025Sep 8, 2025
mishax
Public
Python
•
Apache License 2.0
•9•142•0•1•Updated Sep 8, 2025Sep 8, 2025
tf2jax
Public
Python
•
Apache License 2.0
•16•115•0•18•Updated Sep 6, 2025Sep 6, 2025
corr_faith
Public
Python
•
Apache License 2.0
•0•0•0•0•Updated Sep 6, 2025Sep 6, 2025
videoprism
Public
Official repository for "VideoPrism: A Foundational Visual Encoder for Video Understanding" (ICML 2024)
self-supervised-learning vision-transformer vision-language-model video-foundation-model
Python
•
Apache License 2.0
•24•294•5•1•Updated Sep 6, 2025Sep 6, 2025
neural_assets
Public
Python
•
Apache License 2.0
•0•15•0•0•Updated Sep 5, 2025Sep 5, 2025
vocap
Public
Apache License 2.0
•0•8•1•0•Updated Sep 5, 2025Sep 5, 2025
mujoco_playground
Public
An open-source library for GPU-accelerated robot learning and sim-to-real transfer.
Jupyter Notebook
•
Apache License 2.0
•197•1.4k•19•4•Updated Sep 5, 2025Sep 5, 2025
jax_privacy
Public
Algorithms for Privacy-Preserving Machine Learning in JAX
Python
•
Apache License 2.0
•15•96•2•0•Updated Sep 4, 2025Sep 4, 2025
xarray_jax
Public
Python
•
Apache License 2.0
•2•10•0•0•Updated Sep 4, 2025Sep 4, 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
•1k•4.7k•15•7•Updated Sep 2, 2025Sep 2, 2025
alphafold
Public
Open source code for AlphaFold 2.
Python
•
Apache License 2.0
•2.5k•14k•272•28•Updated Sep 2, 2025Sep 2, 2025
dm-haiku
Public
JAX-based neural network library
machine-learning deep-neural-networks deep-learning neural-networks jax
Python
•
Apache License 2.0
•255•3.1k•75•22•Updated Sep 2, 2025Sep 2, 2025
chex
Public
Python
•
Apache License 2.0
•56•881•25•26•Updated Sep 2, 2025Sep 2, 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
•714•4.2k•106•6•Updated Sep 2, 2025Sep 2, 2025
mctx
Public
Monte Carlo tree search in JAX
monte-carlo-tree-search jax reinforcement-learning
Python
•
Apache License 2.0
•208•2.5k•4•2•Updated Sep 2, 2025Sep 2, 2025
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•140•733•8•3•Updated Sep 2, 2025Sep 2, 2025