Alternatives and similar repositories for flashrl

Users that are interested in flashrl are comparing it to the libraries listed below

• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆32Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆33Updated 9 months ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆11Updated 2 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 2 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆32Updated 6 months ago
• BricksRL / bricksrl
  BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO
  ☆62Updated 10 months ago
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆59Updated 5 months ago
• ttumiel / minRLHF
  Minimal RLHF implementation built on top of minGPT.
  ☆30Updated last year
• sai-prasanna / dreaming_of_many_worlds
  ☆23Updated 10 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆114Updated 11 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆191Updated 11 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• danijar / granular
  Fast dataset format and loader
  ☆22Updated 6 months ago
• danijar / elements
  Building blocks for productive research
  ☆59Updated last week
• Reytuag / transformerXL_PPO_JAX
  ☆81Updated 9 months ago
• kvfrans / lmpo
  ☆97Updated this week
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase
  ☆58Updated this week
• PufferAI / PufferTank
  ☆42Updated last month
• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆105Updated last week
• DramaCow / jaxued
  ☆82Updated 4 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆105Updated 10 months ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆65Updated 7 months ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆31Updated 3 weeks ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆73Updated 11 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 6 months ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆52Updated 11 months ago
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆97Updated 7 months ago