Google DeepMind

All

292 repositories

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
•23•252•9•7•Updated Dec 14, 2024Dec 14, 2024
distrax
Public
Python
•
Apache License 2.0
•32•538•22•19•Updated Dec 14, 2024Dec 14, 2024
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
Jupyter Notebook
•
Apache License 2.0
•843•8.3k•75•36•Updated Dec 14, 2024Dec 14, 2024
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•35•385•37•7•Updated Dec 14, 2024Dec 14, 2024
graphcast
Public
weather weather-forecast
Python
•
Apache License 2.0
•664•5.4k•37•4•Updated Dec 13, 2024Dec 13, 2024
synthid-text
Public
Python
•
Apache License 2.0
•27•392•2•0•Updated Dec 13, 2024Dec 13, 2024
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
Python
•
Apache License 2.0
•195•1.7k•36•24•Updated Dec 13, 2024Dec 13, 2024
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Other
•652•5.6k•15•3•Updated Dec 13, 2024Dec 13, 2024
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•164•721•5•5•Updated Dec 13, 2024Dec 13, 2024
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•123•636•7•2•Updated Dec 12, 2024Dec 12, 2024
tf2jax
Public
Python
•
Apache License 2.0
•11•107•1•13•Updated Dec 12, 2024Dec 12, 2024
dm-haiku
Public
JAX-based neural network library
machine-learning deep-neural-networks deep-learning neural-networks jax
Python
•
Apache License 2.0
•232•2.9k•72•17•Updated Dec 12, 2024Dec 12, 2024
md4
Public
Code for https://arxiv.org/abs/2406.04329
Python
•
Apache License 2.0
•0•32•0•0•Updated Dec 11, 2024Dec 11, 2024
enn
Public
Python
•
Apache License 2.0
•57•298•8•8•Updated Dec 11, 2024Dec 11, 2024
dm_pix
Public
PIX is an image processing library in JAX, for JAX.
machine-learning image computer-vision image-processing jax python
Python
•
Apache License 2.0
•23•395•1•1•Updated Dec 11, 2024Dec 11, 2024
chex
Public
Python
•
Apache License 2.0
•44•798•21•20•Updated Dec 11, 2024Dec 11, 2024
gemma
Public
Open weights LLM from Google DeepMind.
Python
•
Apache License 2.0
•319•2.5k•7•16•Updated Dec 11, 2024Dec 11, 2024
mctx
Public
Monte Carlo tree search in JAX
monte-carlo-tree-search jax reinforcement-learning
Python
•
Apache License 2.0
•193•2.4k•0•1•Updated Dec 11, 2024Dec 11, 2024
mujoco_menagerie
Public
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
robotics mujoco
Python
•
Other
•222•1.5k•15•6•Updated Dec 11, 2024Dec 11, 2024
proactive_t2i_agents
Public
Code release for the paper, "Proactive Agents for Text-to-Image Generation under Uncertainty"
0•4•1•0•Updated Dec 11, 2024Dec 11, 2024
xmanager
Public
A platform for managing machine learning experiments
Python
•
Apache License 2.0
•45•826•12•0•Updated Dec 10, 2024Dec 10, 2024
materials_discovery
Public
Jupyter Notebook
•
Apache License 2.0
•145•912•18•7•Updated Dec 9, 2024Dec 9, 2024
lm_act
Public
LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrations
Python
•
Apache License 2.0
•2•4•0•0•Updated Dec 9, 2024Dec 9, 2024
dm_robotics
Public
Libraries, tools and tasks created and used at DeepMind Robotics.
Python
•
Apache License 2.0
•37•350•4•1•Updated Dec 9, 2024Dec 9, 2024
iris
Public
Python
•
Apache License 2.0
•3•4•0•1•Updated Dec 9, 2024Dec 9, 2024
ferminet
Public
An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Python
•
Apache License 2.0
•130•744•1•1•Updated Dec 9, 2024Dec 9, 2024
what_type_of_inference_is_planning
Public
Python
•
Apache License 2.0
•0•1•0•0•Updated Dec 7, 2024Dec 7, 2024
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
•675•3.9k•93•8•Updated Dec 6, 2024Dec 6, 2024
llms_can_learn_rules
Public
Python
•
Apache License 2.0
•1•2•0•0•Updated Dec 6, 2024Dec 6, 2024
geckonum_benchmark_t2i
Public
GeckoNum Benchmark for T2I Model Eval.
Apache License 2.0
•0•9•0•0•Updated Dec 5, 2024Dec 5, 2024