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 10 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 9 months ago
• tyler-romero / microR1
  Simple repository for training small reasoning models
  ☆44Updated 8 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆65Updated 8 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• character-ai / MuKoe
  ☆54Updated last year
• FLAIROx / cultural-accumulation
  ☆14Updated last year
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆28Updated 3 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆197Updated 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
• okarthikb / state-space-models
  ☆28Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 4 months ago
• RiddleHe / llm-interp
  A collection of lightweight interpretability scripts to understand how LLMs think
  ☆61Updated this week
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆69Updated 10 months ago
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆133Updated last week
• likenneth / q_probe
  Q-Probe: A Lightweight Approach to Reward Maximization for Language Models
  ☆41Updated last year
• facebookresearch / llm-speedrunner
  The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…
  ☆109Updated 3 weeks ago
• upiterbarg / diff_history
  [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)
  ☆20Updated last year
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 4 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆53Updated 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
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years 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
• NVlabs / gbrl_sb3
  GBRL-based Actor-Critic algorithms implemented in stable-baselines3
  ☆38Updated this week
• TristanBilot / mlx-GCN
  MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).
  ☆24Updated last year
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated last year
• EleutherAI / mdl
  Minimum Description Length probing for neural network representations
  ☆20Updated 9 months ago