nsidn98 / PyThor
Template for projects in PyTorch powered with PyTorch Lightning, Telegrad and MLflow. Get updates on mobile and streamline PyTorch code for research.
☆9Updated last year
Related projects ⓘ
Alternatives and complementary repositories for PyThor
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆47Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year
- Deep Q-Network (DQN) and Fitted Q-Iteration (FQI) tutorial for RL Summer School 2023☆51Updated 2 weeks ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆63Updated 3 months ago
- PyTorch Package For Quasimetric Learning☆42Updated 3 weeks ago
- Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M …☆43Updated 2 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆12Updated 2 years ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆33Updated 8 months ago
- Benchmarks for Multi-Objective Multi-Agent Decision Making☆72Updated 3 weeks ago
- Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…☆70Updated 11 months ago
- Fully differentiable RL environments, written in Ivy.☆63Updated last year
- Contains implementation of the FILTER algorithm for exponentially faster inverse reinforcement learning.☆46Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆89Updated 11 months ago
- Representing robots as graphs for reinforcement-learning in PyBullet locomotion environments.☆28Updated 3 years ago
- ☆63Updated 3 months ago
- ☆55Updated last month
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆12Updated 5 months ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- ☆56Updated 2 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆41Updated last year
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆22Updated 4 months ago
- ☆149Updated this week
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆37Updated 4 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆95Updated 3 months ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆49Updated last year
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆26Updated 9 months ago