andrew-silva / clean-rl-mlxLinks
Clean RL implementation using MLX
☆33Updated last year
Alternatives and similar repositories for clean-rl-mlx
Users that are interested in clean-rl-mlx are comparing it to the libraries listed below
Sorting:
- Code for minimum-entropy coupling.☆32Updated last year
- Repo to reproduce the First-Explore paper results☆38Updated 10 months ago
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆36Updated last year
- Simple repository for training small reasoning models☆45Updated 9 months ago
- Jax like function transformation engine but micro, microjax☆33Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 9 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated 2 years ago
- A collection of lightweight interpretability scripts to understand how LLMs think☆66Updated this week
- ☆15Updated last year
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models☆65Updated 8 months ago
- ☆28Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆89Updated last year
- ☆27Updated last year
- Efficient baselines for autocurricula in JAX.☆201Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆69Updated 10 months ago
- Minimum Description Length probing for neural network representations☆20Updated 9 months ago
- Collection of LLM completions for reasoning-gym task datasets☆30Updated 4 months ago
- Intrinsic Motivation from Artificial Intelligence Feedback☆132Updated 2 years ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆54Updated last year
- Scaling scaling laws with board games.☆53Updated 2 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆47Updated 4 years ago
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated last year
- Fast reinforcement learning 💨☆28Updated 4 months ago
- Learn online intrinsic rewards from LLM feedback☆45Updated 11 months ago
- Causal Analysis of Agent Behavior for AI Safety☆19Updated 2 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 5 months ago
- The Automated LLM Speedrunning Benchmark measures how well LLM agents can reproduce previous innovations and discover new ones in languag…☆111Updated last month
- A Gymnasium-based Environment of the Abstraction and Reasoning Corpus (ARC)☆68Updated last year