lasgroup / opax
☆18Updated last month
Alternatives and similar repositories for opax:
Users that are interested in opax are comparing it to the libraries listed below
- ☆30Updated last year
- This repository contains the code for RL for POMDPs through learning an Approximate Information State.☆20Updated 3 years ago
- MBRL library in JAX☆10Updated 2 years ago
- Experiment code for "Continuous-Time Model-Based Reinforcement Learning"☆50Updated last year
- Formulating Model-based RL Dynamics as a continuous rather then one step prediction☆35Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆73Updated last year
- Implementation of Robust Reinforcement Learning using Offline Data [NeurIPS'22]☆23Updated 3 months ago
- improved Cross Entropy Method for trajectory optimization☆73Updated 3 years ago
- ☆15Updated last year
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆34Updated 2 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆24Updated last year
- IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL☆37Updated 5 months ago
- Offline Contextual Bayesian Optimization☆14Updated last year
- ☆48Updated 2 years ago
- An MPC algorithm which supports polytopic state and action constraints, using CEM optimisation.☆14Updated 5 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆66Updated 9 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- A framework for Reinforcement Learning research.☆137Updated last month
- ☆35Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.☆144Updated 3 weeks ago
- ☆34Updated 5 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆134Updated 2 months ago
- Simple maze environments using mujoco-py☆54Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated last year
- Companion code to "Learning Stable Deep Dynamics Models" (Manek and Kolter, 2019)☆32Updated 5 years ago
- ☆19Updated last year
- Stochastic Optimal Control and Reachability Toolbox Written in Python☆13Updated last year
- Goal-Conditioned Reinforcement Learning with JAX☆123Updated this week