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:
- 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
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated 2 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆48Updated 2 years ago
- Contact-Aware Symplectic Integrator Network☆15Updated 2 years ago
- This repository contains the source code to perform Geometry-aware Bayesian Optimization (GaBO) on Riemannian manifolds.☆54Updated 4 years ago
- ☆21Updated 3 years ago
- ☆112Updated 4 years ago
- Neural Optimal Transport with Lagrangian Costs☆60Updated 7 months ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆44Updated last year
- a starter-kit for jaynes, the cloud-agnostic launch library☆17Updated last year
- Tutorial on amortized optimization for learning to optimize over continuous domains☆249Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 5 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated last month
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆81Updated 8 months ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Updated 3 years ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆58Updated 4 years ago
- A python implementation of differentiable quality diversity.☆51Updated 4 years ago
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆29Updated 10 months ago
- ☆58Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆20Updated 5 years ago
- Pytrees + dataclasses ❤️☆69Updated last week
- Code for the paper Task Agnostic Morphology Evolution.☆20Updated 4 years ago
- The Differentiable Cross-Entropy Method☆124Updated 5 years ago