Alternatives and similar repositories for clean-rl-mlx

Users that are interested in clean-rl-mlx are comparing it to the libraries listed below

• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆38Updated 9 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 11 months ago
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆65Updated 7 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 8 months ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆35Updated 11 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆40Updated 8 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆196Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆26Updated 2 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 4 months ago
• jcoreyes / evolvingrl
  Supplementary Data for Evolving Reinforcement Learning Algorithms
  ☆46Updated 4 years ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase
  ☆76Updated this week
• facebookresearch / taskmet
  TaskMet Task-driven Metric Learning for Model Learning
  ☆19Updated last year
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆29Updated this week
• sai-prasanna / dreaming_of_many_worlds
  ☆23Updated last year
• jennyzzt / omni
  OMNI: Open-endedness via Models of human Notions of Interestingness
  ☆57Updated 8 months ago
• DHDev0 / Muzero-unplugged
  Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…
  ☆33Updated 3 months ago
• kyegomez / SelfExtend
  Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta
  ☆12Updated 11 months ago
• RLG-Leiden / edugym
  ☆15Updated 2 years ago
• RiddleHe / llm-interp
  A collection of lightweight interpretability scripts to understand how LLMs think
  ☆56Updated 2 weeks ago
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆131Updated this week
• okarthikb / state-space-models
  ☆28Updated last year
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆115Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆68Updated 9 months ago