Repositories list

• Kinetix

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

  machine-learningreinforcement-learningphysics-engine

  Python

  •

  MIT License

  •6•168•0•1•Updated Mar 11, 2025Mar 11, 2025

• JaxMARL

  Public
  Multi-Agent Reinforcement Learning with JAX

  reinforcement-learningmultiagent-reinforcement-learningmarl

  Python

  •

  Apache License 2.0

  •104•531•6•1•Updated Mar 6, 2025Mar 6, 2025

• 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 17, 2025Feb 17, 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

  •3•55•1•0•Updated Jan 23, 2025Jan 23, 2025

• JaxGL

  Public
  Simple JAX Graphics Library.

  graphicsgraphics-libraryjax

  Python

  •

  MIT License

  •0•34•0•0•Updated Nov 3, 2024Nov 3, 2024

• jaxirl

  Public
  Contains JAX implementation of algorithms for inverse reinforcement learning

  Python

  •

  Apache License 2.0

  •1•69•2•0•Updated Aug 18, 2024Aug 18, 2024

• cultural-accumulation

  Public
  Jupyter Notebook

  •4•13•1•1•Updated Jul 16, 2024Jul 16, 2024

• behaviour-distillation

  Public
  Code for Behaviour Distillation (ICML 2024)

  Python

  •

  Apache License 2.0

  •0•3•0•0•Updated Jun 21, 2024Jun 21, 2024

• FLAIROx.github.io

  Public
  JavaScript

  •0•2•0•0•Updated May 20, 2024May 20, 2024

• popjym

  Public
  POPGym Library in JAX

  Python

  •

  Apache License 2.0

  •0•11•0•0•Updated Apr 15, 2024Apr 15, 2024

• rfos

  Public
  R-FOS

  Jupyter Notebook

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