codingfisch / flashrlLinks
Fast reinforcement learning π¨
β28Updated 6 months ago
Alternatives and similar repositories for flashrl
Users that are interested in flashrl are comparing it to the libraries listed below
Sorting:
- Drop-in environment replacements that make your RL algorithm train faster.β21Updated last year
- Implementation of Soft Actor Critic and some of its improvements in Pytorchβ64Updated last month
- Clean RL implementation using MLXβ34Updated last year
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasksβ36Updated 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 8 months ago
- Minimal RLHF implementation built on top of minGPT.β32Updated last year
- Synchronized Curriculum Learning for RL Agentsβ121Updated this week
- Efficient baselines for autocurricula in JAX.β206Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ122Updated last year
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observβ¦β35Updated 7 months ago
- A simple, performant and scalable JAX-based world modeling codebase.β129Updated 3 weeks ago
- GPT implementation in Flaxβ18Updated 4 years ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).β73Updated last year
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."β36Updated 6 months ago
- Repo to reproduce the First-Explore paper resultsβ39Updated last year
- Implementation of ReWiND, "Language-Guided Rewards Teach Robot Policies without New Demonstrations", from USC / Amazon Roboticsβ35Updated 5 months ago
- β91Updated 2 weeks ago
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3β43Updated last week
- β90Updated last year
- A collection of matrix games in JAXβ13Updated last year
- Fast dataset format and loaderβ23Updated last month
- Generative cellular automaton-like learning environments for RL.β20Updated last year
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - β ICLR 2025β81Updated 11 months ago
- Implementations of Curious Replay for model-based adaptation.β43Updated 2 years ago
- β16Updated last year
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"β31Updated 2 years ago
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Modelsβ66Updated 11 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGOβ66Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024β116Updated last year