Google DeepMind

All

294 repositories

gemma
Public
Open weights LLM from Google DeepMind.
Python
•
Apache License 2.0
•326•2.5k•7•18•Updated Jan 8, 2025Jan 8, 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•255•8•7•Updated Jan 8, 2025Jan 8, 2025
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•37•388•37•6•Updated Jan 8, 2025Jan 8, 2025
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•863•8.5k•82•38•Updated Jan 8, 2025Jan 8, 2025
proactive_t2i_agents
Public
Code release for the paper, "Proactive Agents for Text-to-Image Generation under Uncertainty"
Python
•
Apache License 2.0
•2•19•1•0•Updated Jan 7, 2025Jan 7, 2025
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•691•5.8k•16•2•Updated Jan 6, 2025Jan 6, 2025
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•126•642•7•0•Updated Jan 6, 2025Jan 6, 2025
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•199•1.8k•39•25•Updated Jan 6, 2025Jan 6, 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
•948•4.3k•21•7•Updated Jan 6, 2025Jan 6, 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
•678•3.9k•96•8•Updated Jan 6, 2025Jan 6, 2025
sustainable_behavior
Public
Jupyter Notebook
•
Apache License 2.0
•0•4•0•0•Updated Jan 1, 2025Jan 1, 2025
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•169•745•5•2•Updated Jan 1, 2025Jan 1, 2025
iris
Public
Python
•
Apache License 2.0
•4•7•0•1•Updated Dec 30, 2024Dec 30, 2024
librispeech-long
Public
LibriSpeech-Long is a benchmark dataset for long-form speech generation and processing. Released as part of "Long-Form Speech Generation with Spoken Language Models" (arXiv 2024).
Creative Commons Attribution 4.0 International
•1•44•0•0•Updated Dec 28, 2024Dec 28, 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
•64•570•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
•252•1.8k•2•0•Updated Dec 23, 2024Dec 23, 2024
mishax
Public
Python
•
Apache License 2.0
•5•115•0•0•Updated Dec 20, 2024Dec 20, 2024
alphafold
Public
Open source code for AlphaFold 2.
Python
•
Apache License 2.0
•2.3k•13k•255•28•Updated Dec 20, 2024Dec 20, 2024
clrs
Public
Jupyter Notebook
•
Apache License 2.0
•96•460•4•6•Updated Dec 19, 2024Dec 19, 2024
alphatensor_quantum
Public
Python
•
Apache License 2.0
•3•7•0•0•Updated Dec 19, 2024Dec 19, 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
•684•5.6k•41•5•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
•234•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
•172•1.1k•22•12•Updated Dec 17, 2024Dec 17, 2024
tf2jax
Public
Python
•
Apache License 2.0
•12•110•0•13•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
•19•369•8•0•Updated Dec 16, 2024Dec 16, 2024
distrax
Public
Python
•
Apache License 2.0
•33•541•22•19•Updated Dec 14, 2024Dec 14, 2024
synthid-text
Public
Python
•
Apache License 2.0
•30•405•2•0•Updated Dec 13, 2024Dec 13, 2024
md4
Public
Code for https://arxiv.org/abs/2406.04329
Python
•
Apache License 2.0
•2•47•1•0•Updated Dec 11, 2024Dec 11, 2024