Repositories list

• actionengine

  Public
  A toolkit for building multimodal, streaming APIs and UIs

  C++

  •

  Apache License 2.0

  •7•67•0•13•Updated Jan 10, 2026Jan 10, 2026

• formal-conjectures

  Public
  A collection of formalized statements of conjectures in Lean.

  formal-mathematicslean4

  Lean

  •

  Apache License 2.0

  •175•759•588•56•Updated Jan 10, 2026Jan 10, 2026

• torax

  Public
  TORAX: Tokamak transport simulation in JAX

  Python

  •

  Other

  •106•595•39•26•Updated Jan 10, 2026Jan 10, 2026

• trecvit

  Public

  google-deepmind/trecvit’s past year of commit activity
  Python

  •

  Apache License 2.0

  •1•17•1•0•Updated Jan 10, 2026Jan 10, 2026

• mujoco_menagerie

  Public
  A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

  roboticsmujoco

  Python

  •

  Other

  •400•2.9k•31•13•Updated Jan 10, 2026Jan 10, 2026

• disentangled_rnns

  Public
  Python

  •

  Apache License 2.0

  •6•33•0•32•Updated Jan 10, 2026Jan 10, 2026

• jax_privacy

  Public
  Algorithms for Privacy-Preserving Machine Learning in JAX

  Python

  •

  Apache License 2.0

  •29•136•7•10•Updated Jan 10, 2026Jan 10, 2026

• mujoco

  Public
  Multi-Joint dynamics with Contact. A general purpose physics simulator.

  roboticsphysicsmujoco

  C++

  •

  Apache License 2.0

  •1.3k•12k•138•108•Updated Jan 9, 2026Jan 9, 2026

• neural_testbed

  Public
  Jupyter Notebook

  •

  Apache License 2.0

  •17•193•1•1•Updated Jan 9, 2026Jan 9, 2026

• mujoco_playground

  Public
  An open-source library for GPU-accelerated robot learning and sim-to-real transfer.

  Jupyter Notebook

  •

  Apache License 2.0

  •268•1.7k•47•8•Updated Jan 9, 2026Jan 9, 2026

• dm-haiku

  Public
  JAX-based neural network library

  machine-learningdeep-neural-networksdeep-learning+ 2neural-networksjax

  Python

  •

  Apache License 2.0

  •276•3.2k•75•24•Updated Jan 9, 2026Jan 9, 2026

• clrs

  Public
  Jupyter Notebook

  •

  Apache License 2.0

  •112•512•5•5•Updated Jan 9, 2026Jan 9, 2026

• enn

  Public
  Python

  •

  Apache License 2.0

  •61•312•8•8•Updated Jan 9, 2026Jan 9, 2026

• meqpy

  Public
  Python

  •

  Apache License 2.0

  •0•5•1•0•Updated Jan 9, 2026Jan 9, 2026

• meltingpot

  Public
  A suite of test scenarios for multi-agent reinforcement learning.

  multiagent-reinforcement-learning

  Python

  •

  Apache License 2.0

  •152•776•9•5•Updated Jan 9, 2026Jan 9, 2026

• videoprism

  Public
  Official repository for "VideoPrism: A Foundational Visual Encoder for Video Understanding" (ICML 2024)

  self-supervised-learningvision-transformervision-language-model+ 1video-foundation-model

  Python

  •

  Apache License 2.0

  •29•343•6•1•Updated Jan 9, 2026Jan 9, 2026

• sonnet

  Public
  TensorFlow-based neural network library

  machine-learningdeep-learningtensorflow+ 2artificial-intelligenceneural-networks

  Python

  •

  Apache License 2.0

  •1.3k•9.9k•32•7•Updated Jan 9, 2026Jan 9, 2026

• optax

  Public
  Optax is a gradient processing and optimization library for JAX.

  machine-learningoptimization

  Python

  •

  Apache License 2.0

  •303•2.1k•29•37•Updated Jan 9, 2026Jan 9, 2026

• gemma

  Public
  Gemma open-weight LLM library, from Google DeepMind

  Python

  •

  Apache License 2.0

  •626•3.9k•71•94•Updated Jan 9, 2026Jan 9, 2026

• mujoco_warp

  Public
  GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.

  nvidia-warpmujoco-warp

  Python

  •

  Apache License 2.0

  •109•986•42•31•Updated Jan 9, 2026Jan 9, 2026

• alphafold3

  Public
  AlphaFold 3 inference pipeline.

  Python

  •

  Other

  •1.1k•7.4k•19•4•Updated Jan 9, 2026Jan 9, 2026

• kfac-jax

  Public
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.

  machine-learningoptimizationbayesian-deep-learning

  Python

  •

  Apache License 2.0

  •28•304•13•11•Updated Jan 9, 2026Jan 9, 2026

• 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

  •107•761•23•7•Updated Jan 9, 2026Jan 9, 2026

• recoglab

  Public
  Python

  •

  Apache License 2.0

  •1•9•0•5•Updated Jan 9, 2026Jan 9, 2026

• image_obfuscation_benchmark

  Public
  Python

  •

  Apache License 2.0

  •0•24•1•7•Updated Jan 8, 2026Jan 8, 2026

• sam_edge

  Public
  Python

  •

  Apache License 2.0

  •1•22•0•7•Updated Jan 8, 2026Jan 8, 2026

• alphageometry

  Public
  Python

  •

  Apache License 2.0

  •555•4.7k•124•15•Updated Jan 8, 2026Jan 8, 2026

• max_product_noisy_or

  Public
  Python

  •

  Apache License 2.0

  •2•8•0•14•Updated Jan 8, 2026Jan 8, 2026

• transformer_grammars

  Public
  Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scale, TACL (2022)

  Python

  •

  Apache License 2.0

  •8•133•1•13•Updated Jan 8, 2026Jan 8, 2026

• distribution_shift_framework

  Public
  This repository contains the code of the distribution shift framework presented in A Fine-Grained Analysis on Distribution Shift (Wiles et al., 2022).

  machine-learningdeep-learningartificial-intelligence+ 3neural-networksjaxdistribution-shift

  Python

  •

  Apache License 2.0

  •8•84•1•13•Updated Jan 8, 2026Jan 8, 2026