All

400 repositories

envlogger
Public
A tool for recording RL trajectories.
reinforcement-learning logging
reinforcement-learning logging
Python
•
Apache License 2.0
•20•133•0•3•Updated Aug 21, 2026Aug 21, 2026
mujoco
Public
Multi-Joint dynamics with Contact. A general purpose physics simulator.
robotics physics mujoco
robotics physics mujoco
C++
•
Apache License 2.0
•1.7k•15k•166•183•Updated Aug 21, 2026Aug 21, 2026
formal-conjectures
Public
A collection of formalized statements of conjectures in Lean.
formal-mathematics lean4
formal-mathematics lean4
Lean
•
Apache License 2.0
•420•1.2k•756•307•Updated Aug 21, 2026Aug 21, 2026
torax
Public
TORAX: Tokamak transport simulation in JAX
Python
•
Other
•144•711•30•58•Updated Aug 21, 2026Aug 21, 2026
tf2jax
Public
Python
•
Apache License 2.0
•20•123•0•22•Updated Aug 21, 2026Aug 21, 2026
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
•11•67•2•6•Updated Aug 21, 2026Aug 21, 2026
mujoco_warp
Public
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
nvidia-warp mujoco-warp
nvidia-warp mujoco-warp
Python
•
Apache License 2.0
•199•1.4k•23•21•Updated Aug 21, 2026Aug 21, 2026
meltingpot
Public
A suite of test scenarios for multi-agent reinforcement learning.
multiagent-reinforcement-learning
multiagent-reinforcement-learning
Python
•
Apache License 2.0
•159•863•10•55•Updated Aug 21, 2026Aug 21, 2026
physics-IQ-benchmark
Public
Benchmarking physical understanding in generative video models
benchmark generative-models video-generation
benchmark generative-models video-generation physical-understanding
Python
•
Other
•42•331•0•9•Updated Aug 21, 2026Aug 21, 2026
qtqp
Public
Python
•
Apache License 2.0
•6•35•0•20•Updated Aug 21, 2026Aug 21, 2026
mujoco_mpc
Public
Real-time behaviour synthesis with MuJoCo, using Predictive Control
mpc model-predictive-control mujoco
mpc model-predictive-control mujoco mpc-control predictive-control motor-control
C++
•
Apache License 2.0
•279•1.7k•39•16•Updated Aug 21, 2026Aug 21, 2026
jax_privacy
Public
Algorithms for Privacy-Preserving Machine Learning in JAX
Python
•
Apache License 2.0
•66•191•7•10•Updated Aug 21, 2026Aug 21, 2026
gemma
Public
Gemma open-weight LLM library, from Google DeepMind
Python
•
Apache License 2.0
•1k•5.7k•128•191•Updated Aug 21, 2026Aug 21, 2026
regress-lm
Public
Library for sequence-to-sequence numeric prediction, applicable to any tokenizable input, and allows pretraining and fine-tuning over multiple tasks.
regression prediction sequence-to-sequence
regression prediction sequence-to-sequence quantitative
Python
•
Apache License 2.0
•64•352•0•11•Updated Aug 20, 2026Aug 20, 2026
amplio
Public
Amplio: A Lightweight Agent Harness for Robust and Long-Horizon Runs
Go
•
Apache License 2.0
•3•29•0•0•Updated Aug 20, 2026Aug 20, 2026
dm_control
Public
Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
machine-learning reinforcement-learning deep-learning
machine-learning reinforcement-learning deep-learning artificial-intelligence neural-networks physics-simulation mujoco
Python
•
Apache License 2.0
•762•4.7k•112•14•Updated Aug 20, 2026Aug 20, 2026
optax
Public
Optax is a gradient processing and optimization library for JAX.
machine-learning optimization
machine-learning optimization
Python
•
Apache License 2.0
•361•2.3k•29•80•Updated Aug 20, 2026Aug 20, 2026
superhuman
Public
Lean
•
Apache License 2.0
•82•789•8•10•Updated Aug 19, 2026Aug 19, 2026
alphafold3
Public
AlphaFold 3 inference pipeline.
Python
•
Apache License 2.0
•1.3k•8.5k•10•5•Updated Aug 19, 2026Aug 19, 2026
atomic_concept_edits
Public
Python
•
Apache License 2.0
•1•8•1•1•Updated Aug 19, 2026Aug 19, 2026
kfac-jax
Public
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
machine-learning optimization bayesian-deep-learning
machine-learning optimization bayesian-deep-learning
Python
•
Apache License 2.0
•33•330•11•13•Updated Aug 19, 2026Aug 19, 2026
representations4d
Public
Jupyter Notebook
•
Apache License 2.0
•12•183•2•0•Updated Aug 19, 2026Aug 19, 2026
simply
Public
Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.
Python
•
Apache License 2.0
•65•562•3•8•Updated Aug 19, 2026Aug 19, 2026
mujoco_playground
Public
An open-source library for GPU-accelerated robot learning and sim-to-real transfer.
Python
•
Apache License 2.0
•351•2.2k•65•36•Updated Aug 18, 2026Aug 18, 2026
dm_pix
Public
PIX is an image processing library in JAX, for JAX.
python machine-learning image
python machine-learning image computer-vision image-processing jax
Python
•
Apache License 2.0
•29•444•1•1•Updated Aug 18, 2026Aug 18, 2026
alphagenome
Public
This API provides programmatic access to the AlphaGenome model developed by Google DeepMind.
Python
•
Apache License 2.0
•272•2k•2•0•Updated Aug 17, 2026Aug 17, 2026
concordia
Public
A library for generative social simulation
multi-agent social-simulation agent-based-simulation
multi-agent social-simulation agent-based-simulation generative-agents
Python
•
Apache License 2.0
•359•1.6k•11•39•Updated Aug 14, 2026Aug 14, 2026
reaf
Public
Robotics Environment Authoring Framework
Python
•
Apache License 2.0
•3•13•0•0•Updated Aug 14, 2026Aug 14, 2026
multiscope
Public
Real-time visualisation
Go
•
Apache License 2.0
•8•35•0•6•Updated Aug 13, 2026Aug 13, 2026
open_spiel
Public
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
python games reinforcement-learning
python games reinforcement-learning cpp multiagent
C++
•
Apache License 2.0
•1.2k•5.4k•23•26•Updated Aug 12, 2026Aug 12, 2026

ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.

All

All

Repositories list

400 repositories