xwinxu / bayeSDE
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
☆169Updated 3 years ago
Alternatives and similar repositories for bayeSDE:
Users that are interested in bayeSDE are comparing it to the libraries listed below
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆275Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆103Updated 3 weeks ago
- Experiments from the paper "On Second Order Behaviour in Augmented Neural ODEs"☆57Updated 4 months ago
- Normalizing Flows using JAX☆82Updated last year
- Regularized Neural ODEs (RNODE)☆82Updated 3 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 10 months ago
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆115Updated last year
- Algorithms for computations on random manifolds made easier☆87Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆69Updated 4 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆427Updated last year
- Official code for the ICLR 2021 paper Neural ODE Processes☆70Updated 2 years ago
- This repository contains code released by DiffEqML Research☆85Updated 2 years ago
- ☆45Updated last year
- Turning SymPy expressions into PyTorch modules.☆144Updated last year
- Convex potential flows☆82Updated 3 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 8 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- ODE2VAE: Deep generative second order ODEs with Bayesian neural networks☆126Updated 5 months ago
- Stochastic Normalizing Flows☆75Updated 3 years ago
- Matrix-free linear algebra in JAX.☆114Updated last month
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆100Updated last year
- Manifold-learning flows (ℳ-flows)☆229Updated 4 years ago
- Sequential Neural Likelihood☆39Updated 5 years ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- Experiments for Neural Flows paper☆93Updated 3 years ago
- Differentiable computations for the signature-PDE-kernel on CPU and GPU.☆53Updated 9 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆111Updated this week