Google DeepMind

All

296 repositories

rlax
Public
Python
•
Apache License 2.0
•88•1.3k•8•8•Updated Jan 11, 2025Jan 11, 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 11, 2025Jan 11, 2025
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•199•1.8k•40•25•Updated Jan 11, 2025Jan 11, 2025
inli
Public
Creative Commons Attribution Share Alike 4.0 International
•0•0•0•0•Updated Jan 10, 2025Jan 10, 2025
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•39•388•37•6•Updated Jan 10, 2025Jan 10, 2025
questbench
Public
0•0•0•0•Updated Jan 10, 2025Jan 10, 2025
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•864•8.5k•85•36•Updated Jan 10, 2025Jan 10, 2025
searchless_chess
Public
Grandmaster-Level Chess Without Search
Python
•
Apache License 2.0
•29•537•0•1•Updated Jan 10, 2025Jan 10, 2025
mujoco_menagerie
Public
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
robotics mujoco
Python
•
Other
•234•1.6k•17•7•Updated Jan 10, 2025Jan 10, 2025
android_env
Public
RL research on Android devices.
android reinforcement-learning
Python
•
Apache License 2.0
•79•1k•1•8•Updated Jan 9, 2025Jan 9, 2025
ferminet
Public
An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Python
•
Apache License 2.0
•134•746•0•0•Updated Jan 9, 2025Jan 9, 2025
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•698•5.8k•11•2•Updated Jan 9, 2025Jan 9, 2025
gemma
Public
Open weights LLM from Google DeepMind.
Python
•
Apache License 2.0
•328•2.6k•7•17•Updated Jan 9, 2025Jan 9, 2025
tf2jax
Public
Python
•
Apache License 2.0
•12•110•0•13•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•21•1•0•Updated Jan 7, 2025Jan 7, 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
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
•949•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•5•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•748•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•45•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•256•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
graphcast
Public
weather weather-forecast
Python
•
Apache License 2.0
•685•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