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:
- Bregman ADMM for mass transportation on GPU☆17Dec 4, 2014Updated 11 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 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.☆102Oct 22, 2023Updated 2 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆83May 13, 2024Updated 2 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago
- ☆18Jun 24, 2026Updated last week
- Curated list of JAX Resources and Packages☆45Jun 7, 2026Updated 3 weeks 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)]☆116Dec 5, 2023Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Mar 1, 2021Updated 5 years ago
- ☆37Apr 27, 2023Updated 3 years ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 3 years ago
- A reinforcement learning algorithm for the 2048 game☆20Mar 25, 2014Updated 12 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 last year
- Vectorization techniques for fast population-based training.☆57Apr 26, 2026Updated 2 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆17Oct 23, 2022Updated 3 years ago
- An implementation of MuZero in JAX.☆58Nov 8, 2022Updated 3 years ago
- Explorer is a PyTorch reinforcement learning framework for exploring new ideas.☆98Jun 11, 2026Updated 3 weeks 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 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Code of the paper "Universal Morphology Control via Contextual Modulation" at ICML 2023☆15Aug 3, 2023Updated 2 years ago
- [ACL 2025 Findings] Text2World: Benchmarking Large Language Models for Symbolic World Model Generation☆29Feb 25, 2025Updated last year
- Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024☆24Apr 8, 2024Updated 2 years ago
- ☆60Sep 22, 2022Updated 3 years ago
- General Modules for JAX☆74Apr 7, 2026Updated 2 months ago
- RL Environments in JAX 🌍☆905Apr 2, 2026Updated 3 months ago
- ☆15Jan 26, 2023Updated 3 years ago
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated 2 years ago
- ☆19Nov 25, 2022Updated 3 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [NeurIPS 2019] LOIS: Learning to Optimize In Swarms, guided by posterior estimation☆18Aug 14, 2021Updated 4 years ago
- A benchmark library for Dynamic Algorithm Configuration.☆37May 11, 2026Updated last month
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆14Feb 3, 2023Updated 3 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- ☆28Nov 22, 2019Updated 6 years ago
- ☆21Dec 22, 2020Updated 5 years ago
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 3 months ago