Alternatives and similar repositories for clean-rl-mlx:

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

• keraJLi / synthetic-gymnax
  ☆19Updated 7 months ago
• btnorman / First-Explore
  Repo to reproduce the First-Explore paper results
  ☆37Updated last month
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆30Updated 3 months ago
• PufferAI / PufferTank
  ☆29Updated last month
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆9Updated last year
• okarthikb / state-space-models
  ☆27Updated 6 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated 6 months ago
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆34Updated last month
• evanatyourservice / llm-jax
  Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
  ☆18Updated last week
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 3 weeks ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 7 months ago
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆47Updated 7 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆28Updated 2 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated this week
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆27Updated 3 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆28Updated 3 months ago
• upiterbarg / diff_history
  [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)
  ☆20Updated 5 months ago
• advait / c4a0
  Alpha-Zero Connect Four NN trained via self play
  ☆13Updated 4 months ago
• DramaCow / jaxued
  ☆67Updated 5 months ago
• RLG-Leiden / edugym
  ☆13Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆25Updated 6 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆25Updated 4 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆45Updated 7 months ago
• likenneth / q_probe
  Q-Probe: A Lightweight Approach to Reward Maximization for Language Models
  ☆40Updated 7 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆69Updated 5 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated last month
• enajx / HyperNCA
  ☆37Updated 2 years ago