Skip to content
Change the repository type filter

All

    Repositories list

    • representations4d

      Public
      Jupyter Notebook
      25810Updated Dec 15, 2025Dec 15, 2025
    • mujoco

      Public
      Multi-Joint dynamics with Contact. A general purpose physics simulator.
      C++
      1.2k11k134104Updated Dec 15, 2025Dec 15, 2025
    • torax

      Public
      TORAX: Tokamak transport simulation in JAX
      Python
      1005804529Updated Dec 15, 2025Dec 15, 2025
    • meltingpot

      Public
      A suite of test scenarios for multi-agent reinforcement learning.
      Python
      14877183Updated Dec 15, 2025Dec 15, 2025
    • open_spiel

      Public
      OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
      C++
      1.1k4.9k1516Updated Dec 15, 2025Dec 15, 2025
    • kfac-jax

      Public
      Second Order Optimization and Curvature Estimation with K-FAC in JAX.
      Python
      282981314Updated Dec 15, 2025Dec 15, 2025
    • concordia

      Public
      A library for generative social simulation
      Python
      2451.1k66Updated Dec 15, 2025Dec 15, 2025
    • dm-haiku

      Public
      JAX-based neural network library
      Python
      2713.1k7524Updated Dec 15, 2025Dec 15, 2025
    • qtqp

      Public
      Python
      22311Updated Dec 15, 2025Dec 15, 2025
    • reverb

      Public
      Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
      C++
      107757237Updated Dec 15, 2025Dec 15, 2025
    • formal-conjectures

      Public
      A collection of formalized statements of conjectures in Lean.
      Lean
      14872956040Updated Dec 15, 2025Dec 15, 2025
    • physics-IQ-benchmark

      Public
      Benchmarking physical understanding in generative video models
      Python
      1922250Updated Dec 15, 2025Dec 15, 2025
    • A toolkit for building multimodal, streaming APIs and UIs
      C++
      760011Updated Dec 14, 2025Dec 14, 2025
    • mujoco_warp

      Public
      GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
      Python
      1059434417Updated Dec 14, 2025Dec 14, 2025
    • chex

      Public
      Python
      629022529Updated Dec 13, 2025Dec 13, 2025
    • simply

      Public
      Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.
      Python
      104421Updated Dec 13, 2025Dec 13, 2025
    • Jupyter Notebook
      716611Updated Dec 12, 2025Dec 12, 2025
    • disentangled_rnns

      Public
      Python
      633030Updated Dec 12, 2025Dec 12, 2025
    • alphagenome

      Public
      This API provides programmatic access to the AlphaGenome model developed by Google DeepMind.
      Python
      1641.3k20Updated Dec 12, 2025Dec 12, 2025
    • Algorithms for Privacy-Preserving Machine Learning in JAX
      Python
      2513524Updated Dec 12, 2025Dec 12, 2025
    • tf2jax

      Public
      Python
      17118021Updated Dec 11, 2025Dec 11, 2025
    • optax

      Public
      Optax is a gradient processing and optimization library for JAX.
      Python
      2932.1k3637Updated Dec 11, 2025Dec 11, 2025
    • distrax

      Public
      Python
      396102624Updated Dec 9, 2025Dec 9, 2025
    • rlax

      Public
      Python
      971.4k910Updated Dec 9, 2025Dec 9, 2025
    • 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
      4329905Updated Dec 7, 2025Dec 7, 2025
    • dangerous-capability-evaluations

      Public
      Python
      36207Updated Dec 6, 2025Dec 6, 2025
    • recoglab

      Public
      Python
      0905Updated Dec 6, 2025Dec 6, 2025
    • Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
      Python
      80659316Updated Dec 6, 2025Dec 6, 2025
    • image_obfuscation_benchmark

      Public
      Python
      02417Updated Dec 6, 2025Dec 6, 2025
    • materials_discovery

      Public
      Jupyter Notebook
      1741.1k219Updated Dec 6, 2025Dec 6, 2025