Repositories list

• envelope-bench

  Public
  A jax-native environment interface

  Python

  •

  MIT License

  •2•0•0•0•Updated Feb 17, 2026Feb 17, 2026

• website

  Public
  Administered by Matt. Branch main is automatically deployed to Netlify so please only push public stuff there.

  0•0•0•0•Updated Feb 16, 2026Feb 16, 2026

• envelope-backup

  Public
  A jax-native environment interface

  Python

  •

  MIT License

  •2•0•0•0•Updated Feb 11, 2026Feb 11, 2026

• IFG

  Public
  Official implementation of the Intent Factored Generation paper

  Python

  •

  Apache License 2.0

  •1•6•0•0•Updated Jan 27, 2026Jan 27, 2026

• Kinetix

  Public
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.

  machine-learningreinforcement-learningphysics-engine

  Python

  •

  MIT License

  •9•230•2•0•Updated Jan 24, 2026Jan 24, 2026

• JaxMARL

  Public
  Multi-Agent Reinforcement Learning with JAX

  reinforcement-learningmultiagent-reinforcement-learningmarl

  Python

  •

  Apache License 2.0

  •138•743•17•1•Updated Jan 14, 2026Jan 14, 2026

• ah2ac2

  Public
  Ad-Hoc Human-AI Coordination Challenge (AH2AC2)

  reinforcement-learningartificial-intelligencemultiagent-reinforcement-learning+ 2marlhuman-ai-collaboration

  Python

  •1•12•0•0•Updated Jul 8, 2025Jul 8, 2025

• FLAIROx.github.io

  Public
  JavaScript

  •0•2•0•0•Updated Jun 29, 2025Jun 29, 2025

• hugo-theme-flair

  Public
  Hugo theme for the FLAIR website

  HTML

  •

  MIT License

  •0•0•0•0•Updated Feb 12, 2025Feb 12, 2025

• jafar

  Public
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"

  deepmindgenievideo-generation+ 1icml-2024

  Python

  •

  Apache License 2.0

  •22•100•4•3•Updated Jan 23, 2025Jan 23, 2025

• JaxGL

  Public
  Simple JAX Graphics Library.

  graphicsgraphics-libraryjax

  Python

  •

  MIT License

  •1•36•0•0•Updated Nov 3, 2024Nov 3, 2024

• jaxirl

  Public
  Contains JAX implementation of algorithms for inverse reinforcement learning

  Python

  •

  Apache License 2.0

  •4•74•2•0•Updated Aug 18, 2024Aug 18, 2024

• cultural-accumulation

  Public
  Jupyter Notebook

  •6•16•1•1•Updated Jul 16, 2024Jul 16, 2024

• behaviour-distillation

  Public
  Code for Behaviour Distillation (ICML 2024)

  Python

  •

  Apache License 2.0

  •1•6•0•0•Updated Jun 21, 2024Jun 21, 2024

• popjym

  Public
  POPGym Library in JAX

  Python

  •

  Apache License 2.0

  •1•12•0•0•Updated Apr 15, 2024Apr 15, 2024

• rfos

  Public
  R-FOS

  Jupyter Notebook

  •0•0•0•0•Updated Feb 7, 2024Feb 7, 2024