Skip to content
Change the repository type filter

All

    Repositories list

    • regress-lm

      Public
      Library for text-to-text regression, applicable to any input string representation and allows pretraining and fine-tuning over multiple regression tasks.
      Python
      Apache License 2.0
      53335010Updated Apr 13, 2026Apr 13, 2026
    • kfac-jax

      Public
      Second Order Optimization and Curvature Estimation with K-FAC in JAX.
      Python
      Apache License 2.0
      303191312Updated Apr 13, 2026Apr 13, 2026
    • torax

      Public
      TORAX: Tokamak transport simulation in JAX
      Python
      Other
      1216553035Updated Apr 13, 2026Apr 13, 2026
    • formal-conjectures

      Public
      A collection of formalized statements of conjectures in Lean.
      Lean
      Apache License 2.0
      257919754107Updated Apr 13, 2026Apr 13, 2026
    • tips

      Public
      TIPS (ICLR'25): Text-Image Pretraining with Spatial Awareness
      Jupyter Notebook
      Apache License 2.0
      1317011Updated Apr 13, 2026Apr 13, 2026
    • tf2jax

      Public
      Python
      Apache License 2.0
      19120021Updated Apr 13, 2026Apr 13, 2026
    • qtqp

      Public
      Python
      Apache License 2.0
      52703Updated Apr 13, 2026Apr 13, 2026
    • concordia

      Public
      A library for generative social simulation
      Python
      Apache License 2.0
      3041.3k923Updated Apr 11, 2026Apr 11, 2026
    • jax_privacy

      Public
      Algorithms for Privacy-Preserving Machine Learning in JAX
      Python
      Apache License 2.0
      4616375Updated Apr 10, 2026Apr 10, 2026
    • chex

      Public
      Python
      Apache License 2.0
      699352743Updated Apr 10, 2026Apr 10, 2026
    • mujoco

      Public
      Multi-Joint dynamics with Contact. A general purpose physics simulator.
      C++
      Apache License 2.0
      1.4k13k156160Updated Apr 10, 2026Apr 10, 2026
    • meltingpot

      Public
      A suite of test scenarios for multi-agent reinforcement learning.
      Python
      Apache License 2.0
      15581498Updated Apr 9, 2026Apr 9, 2026
    • gemma

      Public
      Gemma open-weight LLM library, from Google DeepMind
      Python
      Apache License 2.0
      8404.8k105138Updated Apr 9, 2026Apr 9, 2026
    • disentangled_rnns

      Public
      Python
      Apache License 2.0
      1241033Updated Apr 9, 2026Apr 9, 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
      6051738Updated Apr 8, 2026Apr 8, 2026
    • physics-IQ-benchmark

      Public
      Benchmarking physical understanding in generative video models
      Python
      Other
      2827642Updated Apr 8, 2026Apr 8, 2026
    • acme

      Public
      A library of reinforcement learning components and agents
      Python
      Apache License 2.0
      5364k6929Updated Apr 8, 2026Apr 8, 2026
    • dangerous-capability-evaluations

      Public
      Python
      Apache License 2.0
      569020Updated Apr 8, 2026Apr 8, 2026
    • calm

      Public
      Python
      Apache License 2.0
      105832Updated Apr 8, 2026Apr 8, 2026
    • long-form-factuality

      Public
      Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
      Python
      Other
      83680319Updated Apr 8, 2026Apr 8, 2026
    • corr_faith

      Public
      Python
      Apache License 2.0
      0501Updated Apr 8, 2026Apr 8, 2026
    • action_piece

      Public
      Python
      Apache License 2.0
      75322Updated Apr 8, 2026Apr 8, 2026
    • xarray_jax

      Public
      Python
      Apache License 2.0
      73810Updated Apr 7, 2026Apr 7, 2026
    • mujoco_menagerie

      Public
      A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
      Python
      Other
      4573.3k3519Updated Apr 7, 2026Apr 7, 2026
    • mujoco_warp

      Public
      GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
      Python
      Apache License 2.0
      1441.2k3417Updated Apr 7, 2026Apr 7, 2026
    • multiscope

      Public
      Real-time visualisation
      Go
      Apache License 2.0
      42301Updated Apr 7, 2026Apr 7, 2026
    • alphafold

      Public
      Open source code for AlphaFold 2.
      Python
      Apache License 2.0
      2.6k14k27527Updated Apr 7, 2026Apr 7, 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
      1077732310Updated Apr 7, 2026Apr 7, 2026
    • mujoco_playground

      Public
      An open-source library for GPU-accelerated robot learning and sim-to-real transfer.
      Python
      Apache License 2.0
      3051.9k5815Updated Apr 6, 2026Apr 6, 2026
    • HAI_selective_prediction

      Public
      Jupyter Notebook
      Apache License 2.0
      1600Updated Apr 6, 2026Apr 6, 2026
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.