luchris429 / popjaxrl

Related projects: ⓘ

• DramaCow / jaxued
  ☆56Updated 3 weeks ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆27Updated last week
• jurgisp / memory-maze
  Evaluating long-term memory of reinforcement learning algorithms
  ☆129Updated last year
• MichalBortkiewicz / JaxGCRL
  ☆25Updated this week
• mklissa / dceo
  Learning diverse options through the Laplacian representation.
  ☆22Updated 8 months ago
• seohongpark / HILP
  Foundation Policies with Hilbert Representations (ICML 2024)
  ☆65Updated 5 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆59Updated last month
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆51Updated 5 months ago
• cassidylaidlaw / effective-horizon
  Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"
  ☆41Updated 2 months ago
• danijar / ninjax
  General Modules for JAX
  ☆57Updated last month
• keraJLi / rejax
  ☆141Updated 2 weeks ago
• twni2016 / Memory-RL
  When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)
  ☆49Updated 8 months ago
• vivekmyers / contrastive_planning
  Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
  ☆35Updated 2 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆59Updated last month
• Div99 / XQL
  Extreme Q-Learning: Max Entropy RL without Entropy
  ☆78Updated last year
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆17Updated last month
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆26Updated last month
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆10Updated 2 months ago
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆87Updated last month
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆189Updated 2 weeks ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆102Updated 3 weeks ago
• keraJLi / synthetic-gymnax
  ☆17Updated 3 months ago
• kvfrans / rlbase_stable
  ☆40Updated 2 months ago
• young-geng / JaxCQL
  Conservative Q learning in Jax
  ☆49Updated last year
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆58Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆39Updated 2 years ago
• proroklab / ffm
  Reinforcement Learning with Fast and Forgetful Memory
  ☆23Updated last year
• MichaelTMatthews / Craftax_Baselines
  ☆14Updated last month
• twitter-research / hyperbolic-rl
  ☆56Updated last year
• seohongpark / METRA
  METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)
  ☆49Updated 11 months ago