codingfisch / flashrlLinks
Fast reinforcement learning π¨
β26Updated 2 months ago
Alternatives and similar repositories for flashrl
Users that are interested in flashrl are comparing it to the libraries listed below
Sorting:
- Clean RL implementation using MLXβ33Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.β21Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.β32Updated 4 months ago
- Efficient baselines for autocurricula in JAX.β196Updated last year
- A collection of matrix games in JAXβ12Updated 10 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ115Updated last year
- Implementation of Soft Actor Critic and some of its improvements in Pytorchβ60Updated 7 months ago
- Fast dataset format and loaderβ22Updated 8 months ago
- β84Updated last month
- β23Updated last year
- Repo to reproduce the First-Explore paper resultsβ38Updated 9 months ago
- β54Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learningβ72Updated last year
- Synchronized Curriculum Learning for RL Agentsβ113Updated last month
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasksβ35Updated 11 months ago
- β84Updated 11 months ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."β33Updated 3 months ago
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observβ¦β33Updated 3 months ago
- Minimal RLHF implementation built on top of minGPT.β30Updated last year
- A2C is a special case of PPO!β22Updated 3 years ago
- Simple repository for training small reasoning modelsβ40Updated 8 months ago
- GPT implementation in Flaxβ18Updated 3 years ago
- β13Updated last year
- Implementation of ReWiND, "Language-Guided Rewards Teach Robot Policies without New Demonstrations", from USC / Amazon Roboticsβ33Updated last month
- Causal Analysis of Agent Behavior for AI Safetyβ18Updated 2 years ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024β108Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).β68Updated 9 months ago
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Modelsβ65Updated 7 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGOβ64Updated last year