Optim4RL is a Jax framework of learning to optimize for reinforcement learning.
☆28Nov 27, 2024Updated last year
Alternatives and similar repositories for optim4rl
Users that are interested in optim4rl are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 years ago
- Bregman ADMM for mass transportation on GPU☆17Dec 4, 2014Updated 11 years ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- an environment based on XLA for deep learning compiler optimization research.☆24Mar 7, 2023Updated 3 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆83May 13, 2024Updated last year
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago
- Curated list of JAX Resources and Packages☆35Mar 2, 2026Updated 3 weeks ago
- ☆18Dec 30, 2025Updated 2 months ago
- Sandbox environment for generalizable agent research☆27Aug 19, 2022Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Mar 1, 2021Updated 5 years ago
- ☆37Apr 27, 2023Updated 2 years ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 4 years ago
- A reinforcement learning algorithm for the 2048 game☆20Mar 25, 2014Updated 11 years ago
- Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)☆42Jan 13, 2024Updated 2 years ago
- Exchange correlation functionals translated from libxc to jax☆52Mar 24, 2025Updated 11 months ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆17Oct 23, 2022Updated 3 years ago
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- Code of the paper "Universal Morphology Control via Contextual Modulation" at ICML 2023☆13Aug 3, 2023Updated 2 years ago
- Explorer is a PyTorch reinforcement learning framework for exploring new ideas.☆97Jun 19, 2025Updated 9 months ago
- Learning from preferences is a common paradigm for fine-tuning language models. Yet, many algorithmic design decisions come into play. Ou…☆32Apr 20, 2024Updated last year
- [ACL 2025 Findings] Text2World: Benchmarking Large Language Models for Symbolic World Model Generation☆28Feb 25, 2025Updated last year
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆24Apr 8, 2024Updated last year
- ☆59Sep 22, 2022Updated 3 years ago
- General Modules for JAX☆72Feb 21, 2026Updated last month
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- ☆15Jan 26, 2023Updated 3 years ago
- Proof of concept example for the idea of using distributed model predictive control as a function approximator in distributed reinforceme…☆22Mar 17, 2025Updated last year
- ☆19Nov 25, 2022Updated 3 years ago
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated last year
- [NeurIPS 2019] LOIS: Learning to Optimize In Swarms, guided by posterior estimation☆18Aug 14, 2021Updated 4 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆14Feb 3, 2023Updated 3 years ago
- A benchmark library for Dynamic Algorithm Configuration.☆35Feb 23, 2026Updated last month
- ☆28Nov 22, 2019Updated 6 years ago
- ☆21Dec 22, 2020Updated 5 years ago