tachukao / idoc
Implicit Differentiable Optimal Control (IDOC) with JAX
☆12Updated 2 years ago
Alternatives and similar repositories for idoc:
Users that are interested in idoc are comparing it to the libraries listed below
- Code for "Learning Control-Oriented Dynamical Structure from Data" by Spencer M. Richards, Jean-Jacques Slotine, Navid Azizan, and Marco …☆14Updated last year
- Performant, differentiable reinforcement learning☆25Updated last year
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆15Updated 2 years ago
- A toolbox for inference of mixture models☆16Updated last year
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆31Updated last year
- Online variational GPs☆32Updated last year
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Updated 4 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆24Updated 2 years ago
- a High-Performance Distributed Solver for Large-Scale Markov Decision Processes (MDP) relying on Inexact Policy Iteration; for Python and…☆23Updated last week
- Variational Reinforcement Learning☆16Updated 8 months ago
- ☆13Updated last year
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated 9 months ago
- Tree-structured recurrent switching linear dynamical systems☆36Updated 4 years ago
- Input Inference for Control (i2c), a control-as-inference framework for optimal control☆23Updated last year
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆26Updated 7 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆66Updated last year
- GPT implementation in Flax☆18Updated 3 years ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆21Updated 10 months ago
- Stabilizable Nonlinear Dynamics Learning☆21Updated 5 years ago
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆37Updated 2 years ago
- Google AI Princeton control framework☆38Updated 4 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆32Updated 3 weeks ago
- Python bindings to some optimization benchmarks (robotics problems), in order to constrained optimization solvers. Includes also an inter…☆21Updated 2 years ago
- A brief JAX tutorial with examples from control theory☆12Updated 2 years ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆34Updated 2 years ago
- [CoRL 2020] COG: Connecting New Skills to Past Experience with Offline Reinforcement Learning☆32Updated 4 years ago
- Pytorch implementation of Stable Vector Fields on Lie Groups through Diffeomorphism☆18Updated 4 months ago
- ☆34Updated 3 years ago