google-deepmind / dm_hamiltonian_dynamics_suiteLinks
☆36Updated 4 years ago
Alternatives and similar repositories for dm_hamiltonian_dynamics_suite
Users that are interested in dm_hamiltonian_dynamics_suite are comparing it to the libraries listed below
Sorting:
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆37Updated 2 years ago
- Re-implementation of Hamiltonian Generative Networks paper☆33Updated 3 years ago
- This repository contains the source code to perform Geometry-aware Bayesian Optimization (GaBO) on Riemannian manifolds.☆54Updated 4 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- ☆21Updated 2 years ago
- Tutorial on amortized optimization for learning to optimize over continuous domains☆249Updated 2 months ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- ☆112Updated 4 years ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated 2 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Updated 3 years ago
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆29Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costs☆60Updated 6 months ago
- Contact-Aware Symplectic Integrator Network☆15Updated 2 years ago
- Pytrees + dataclasses ❤️☆69Updated 7 months ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆58Updated 4 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆48Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 4 months ago
- Official implementation for our paper "Unsupervised Learning of Lagrangian Dynamics from Images for Prediction and Control"☆20Updated 3 years ago
- ☆58Updated 3 years ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆41Updated 6 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 7 months ago
- Baselines for Model-Based Optimization☆54Updated 3 years ago
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆55Updated 4 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- ☆53Updated 3 years ago
- Code for the paper Task Agnostic Morphology Evolution.☆20Updated 4 years ago
- Fully differentiable RL environments, written in Ivy.☆66Updated 2 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆40Updated last year