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:
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.β32Updated 8 months ago
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasksβ36Updated last year
- Clean RL implementation using MLXβ34Updated last year
- 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
- GPT implementation in Flaxβ18Updated 4 years 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
- Minimal RLHF implementation built on top of minGPT.β32Updated last year
- Efficient baselines for autocurricula in JAX.β206Updated last year
- β91Updated 2 weeks ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ122Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.β129Updated 2 months ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).β73Updated last year
- β90Updated last year
- Synchronized Curriculum Learning for RL Agentsβ121Updated this week
- A simple, performant and scalable JAX-based world modeling codebase.β129Updated 3 weeks ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)β20Updated 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
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Modelsβ66Updated 11 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024β116Updated last year
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3β43Updated last week
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGOβ66Updated last year
- Simple repository for training small reasoning modelsβ48Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorchβ153Updated 9 months ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."β36Updated 6 months ago
- β24Updated last year
- Intrinsic Motivation from Artificial Intelligence Feedbackβ134Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β59Updated 3 years ago