google-deepmind / dm_hamiltonian_dynamics_suite
☆34Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for dm_hamiltonian_dynamics_suite
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆31Updated last year
- ☆18Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆45Updated 2 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- [ICLR 2022] Path integral sampler☆43Updated last year
- Re-implementation of Hamiltonian Generative Networks paper☆33Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆27Updated 3 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- Official implementation for our paper "Unsupervised Learning of Lagrangian Dynamics from Images for Prediction and Control"☆17Updated 2 years ago
- ☆49Updated last year
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆61Updated last year
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- A PyTorch library for all things nonlinear control and reinforcement learning.☆44Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- ☆102Updated 3 years ago
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- Augmenting Physical Models with Deep Networks for Complex Dynamics Forecasting☆39Updated 11 months ago
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- ☆25Updated 2 months ago
- Refining continuous-in-depth neural networks☆39Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆51Updated 8 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆20Updated 2 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆66Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆31Updated last year
- Code for the paper: "Independent mechanism analysis, a new concept?"☆25Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆38Updated last week
- Neural ODE tutorial☆9Updated 2 years ago
- Neural Optimal Transport with Lagrangian Costs☆49Updated 4 months ago