mlysy / rodeo-legacy
Probabilistic Solution of Differential Equations
☆13Updated 2 years ago
Alternatives and similar repositories for rodeo-legacy:
Users that are interested in rodeo-legacy are comparing it to the libraries listed below
- ☆18Updated 3 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- ☆10Updated 2 years ago
- ☆15Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- ☆12Updated 4 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Updated 3 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Course Website☆9Updated 3 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆9Updated 4 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- A differentiation API for PyTorch☆30Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- ☆80Updated 3 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- ☆15Updated 2 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago