jacobjinkelly / easy-neural-odeLinks
Code for the paper "Learning Differential Equations that are Easy to Solve"
☆282Updated 3 years ago
Alternatives and similar repositories for easy-neural-ode
Users that are interested in easy-neural-ode are comparing it to the libraries listed below
Sorting:
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- Turning SymPy expressions into PyTorch modules.☆152Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 6 months ago
- Tensorflow implementation of Ordinary Differential Equation Solvers with full GPU support☆223Updated 5 years ago
- This repository contains code released by DiffEqML Research☆90Updated 3 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆457Updated last year
- Manifold-learning flows (ℳ-flows)☆230Updated 4 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆448Updated last week
- ☆110Updated 4 years ago
- Code for our paper "Hamiltonian Neural Networks"☆480Updated 4 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆127Updated 10 months ago
- ODE2VAE: Deep generative second order ODEs with Bayesian neural networks☆129Updated last year
- Pytorch implementation of Augmented Neural ODEs☆547Updated 2 years ago
- ☆214Updated 3 years ago
- Lagrangian Neural Networks☆511Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆308Updated 2 months ago
- A library for programmatically generating equivariant layers through constraint solving☆273Updated 2 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 4 years ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆236Updated last year
- Regularized Neural ODEs (RNODE)☆85Updated 4 years ago
- Algorithms for computations on random manifolds made easier☆92Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 2 years ago
- Nonparametric Differential Equation Modeling☆54Updated last year
- Extensible Tensorflow library for differentiable particle filtering. ICML 2021.☆42Updated 2 years ago
- Normalizing Flows using JAX☆84Updated last year
- Convex potential flows☆84Updated 3 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆120Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated last year
- Gaussian processes in JAX and Flax.☆534Updated this week