codingfisch / flashrlLinks
Fast reinforcement learning π¨
β28Updated 4 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
- Synchronized Curriculum Learning for RL Agentsβ116Updated 3 weeks ago
- Implementation of ReWiND, "Language-Guided Rewards Teach Robot Policies without New Demonstrations", from USC / Amazon Roboticsβ35Updated 3 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasksβ36Updated last year
- A collection of matrix games in JAXβ12Updated last year
- Minimal RLHF implementation built on top of minGPT.β30Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ117Updated last year
- Implementation of Soft Actor Critic and some of its improvements in Pytorchβ60Updated 9 months ago
- Clean RL implementation using MLXβ33Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.β32Updated 5 months ago
- β15Updated last year
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."β35Updated 4 months ago
- Efficient baselines for autocurricula in JAX.β201Updated last year
- Repo to reproduce the First-Explore paper resultsβ38Updated 11 months ago
- A simple, performant and scalable JAX-based world modeling codebaseβ113Updated last month
- GPT implementation in Flaxβ18Updated 3 years ago
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observβ¦β34Updated 5 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024β114Updated last year
- β35Updated last year
- β87Updated last year
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Modelsβ65Updated 9 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGOβ64Updated last year
- β23Updated last year
- β54Updated last year
- β87Updated 2 months ago
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).β71Updated 11 months ago
- Simple repository for training small reasoning modelsβ46Updated 9 months ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)β20Updated last year
- fast + parallel AlphaZero in JAXβ106Updated 11 months ago