Alternatives and similar repositories for clean-rl-mlx

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

• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆38Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆34Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated last year
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆66Updated 11 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆48Updated 11 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated 3 weeks ago
• FLAIROx / cultural-accumulation
  ☆16Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆75Updated 6 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated 2 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 7 months ago
• character-ai / MuKoe
  ☆55Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆28Updated 6 months ago
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆20Updated 11 months ago
• okarthikb / state-space-models
  ☆27Updated last year
• kyegomez / SelfExtend
  Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta
  ☆13Updated last year
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆136Updated last week
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated 2 years ago
• likenneth / q_probe
  Q-Probe: A Lightweight Approach to Reward Maximization for Language Models
  ☆41Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆73Updated last year
• NVlabs / gbrl_sb3
  GBRL-based Actor-Critic algorithms implemented in stable-baselines3
  ☆42Updated this week
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆29Updated last week
• HyperPotatoNeo / RSA
  ☆82Updated 3 months ago
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆64Updated 3 weeks ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• benediktstroebl / agent-evals
  ☆25Updated 8 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated last year
• facebookresearch / motif
  Intrinsic Motivation from Artificial Intelligence Feedback
  ☆135Updated 2 years ago