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
  ☆37Updated 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.
  ☆19Updated 4 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆33Updated 4 months ago
• FLAIROx / cultural-accumulation
  ☆12Updated 11 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆33Updated 7 months ago
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆41Updated 6 months ago
• okarthikb / state-space-models
  ☆27Updated 11 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 8 months ago
• upiterbarg / diff_history
  [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)
  ☆20Updated 10 months ago
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆58Updated 4 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 7 months ago
• PufferAI / PufferTank
  ☆39Updated last month
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆17Updated 3 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 3 weeks ago
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆24Updated 3 months ago
• samvelyan / minihack
  MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research
  ☆21Updated last week
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆11Updated 2 years ago
• benediktstroebl / agent-evals
  ☆21Updated 3 weeks ago
• google-deepmind / enn_acme
  ☆31Updated 2 years ago
• leloykun / modded-nanogpt
  NanoGPT (124M) quality in 2.67B tokens
  ☆28Updated last month
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆20Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆59Updated 6 months ago
• enajx / HyperNCA
  ☆39Updated 3 years ago
• nacloos / baba-is-ai
  ☆37Updated 9 months ago
• minosvasilias / simple_grpo
  Simple GRPO scripts and configurations.
  ☆58Updated 4 months ago
• EleutherAI / mdl
  Minimum Description Length probing for neural network representations
  ☆18Updated 4 months ago
• esennesh / dcpc_paper
  ☆12Updated last week