Google DeepMind

All

347 repositories

optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•247•1.9k•29•24•Updated Jul 23, 2025Jul 23, 2025
gemma
Public
Gemma open-weight LLM library, from Google DeepMind
Python
•
Apache License 2.0
•491•3.5k•43•45•Updated Jul 23, 2025Jul 23, 2025
dm-haiku
Public
JAX-based neural network library
machine-learning deep-neural-networks deep-learning neural-networks jax
Python
•
Apache License 2.0
•253•3.1k•75•21•Updated Jul 22, 2025Jul 22, 2025
chex
Public
Python
•
Apache License 2.0
•54•866•25•27•Updated Jul 22, 2025Jul 22, 2025
tf2jax
Public
Python
•
Apache License 2.0
•15•115•0•17•Updated Jul 22, 2025Jul 22, 2025
hybrid_rnns_reward_learning
Public
Jupyter Notebook
•
Apache License 2.0
•0•1•0•0•Updated Jul 22, 2025Jul 22, 2025
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•206•954•10•5•Updated Jul 22, 2025Jul 22, 2025
mujoco_warp
Public
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
nvidia-warp
Python
•
Apache License 2.0
•60•640•48•8•Updated Jul 22, 2025Jul 22, 2025
formal-conjectures
Public
A collection of formalized statements of conjectures in Lean.
formal-mathematics lean4
Lean
•
Apache License 2.0
•64•559•177•29•Updated Jul 22, 2025Jul 22, 2025
bagz
Public
Bagz is a format for storing a sequence of string records. It supports per-record compression and fast index-based lookup.
C++
•
Apache License 2.0
•1•16•0•0•Updated Jul 22, 2025Jul 22, 2025
sonnet
Public
TensorFlow-based neural network library
machine-learning deep-learning artificial-intelligence neural-networks tensorflow
Python
•
Apache License 2.0
•1.3k•9.9k•32•8•Updated Jul 22, 2025Jul 22, 2025
acme
Public
A library of reinforcement learning components and agents
research reinforcement-learning agents
Python
•
Apache License 2.0
•490•3.7k•66•14•Updated Jul 22, 2025Jul 22, 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
•20•247•5•1•Updated Jul 21, 2025Jul 21, 2025
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•1.1k•10k•126•66•Updated Jul 21, 2025Jul 21, 2025
alphagenome
Public
This API provides programmatic access to the AlphaGenome model developed by Google DeepMind.
Python
•
Apache License 2.0
•117•1.1k•1•0•Updated Jul 21, 2025Jul 21, 2025
levintreesearch_cm
Public
Racket
•
Apache License 2.0
•5•32•0•0•Updated Jul 21, 2025Jul 21, 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
•14•93•1•2•Updated Jul 21, 2025Jul 21, 2025
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•60•443•30•19•Updated Jul 21, 2025Jul 21, 2025
android_env
Public
RL research on Android devices.
android reinforcement-learning
Python
•
Apache License 2.0
•89•1.1k•0•10•Updated Jul 21, 2025Jul 21, 2025
igen
Public
Apache License 2.0
•0•0•0•0•Updated Jul 18, 2025Jul 18, 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
•26•281•13•10•Updated Jul 17, 2025Jul 17, 2025
xmanager
Public
A platform for managing machine learning experiments
Python
•
Apache License 2.0
•52•862•17•6•Updated Jul 17, 2025Jul 17, 2025
anthro-benchmark
Public
Python
•
Apache License 2.0
•2•6•0•0•Updated Jul 16, 2025Jul 16, 2025
alphafold
Public
Open source code for AlphaFold 2.
Python
•
Apache License 2.0
•2.4k•14k•272•35•Updated Jul 16, 2025Jul 16, 2025
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
•102•737•23•6•Updated Jul 15, 2025Jul 15, 2025
disentangled_rnns
Public
Python
•
Apache License 2.0
•5•24•0•6•Updated Jul 15, 2025Jul 15, 2025
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
•76•626•3•15•Updated Jul 15, 2025Jul 15, 2025
deepmind-research
Public
This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook
•
Apache License 2.0
•2.7k•14k•194•88•Updated Jul 15, 2025Jul 15, 2025
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•882•6.8k•23•1•Updated Jul 14, 2025Jul 14, 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
•704•4.2k•106•5•Updated Jul 13, 2025Jul 13, 2025