lezcano / expm
Differentiable and numerically stable implementation of the matrix exponential
☆32Updated 4 years ago
Alternatives and similar repositories for expm:
Users that are interested in expm are comparing it to the libraries listed below
- Riemannian Convex Potential Maps☆67Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆115Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆101Updated last year
- Normalizing Flows using JAX☆82Updated last year
- ☆38Updated 4 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 3 years ago
- A generic interface for linear algebra backends☆71Updated 8 months ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- Kernel Stein Discrepancy Descent : a method to sample from unnormalized densities☆21Updated 10 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- ☆22Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆42Updated 4 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Updated 3 years ago
- Optimization with orthogonal constraints and on general manifolds☆127Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆48Updated 5 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆169Updated 3 years ago
- Algorithms for computations on random manifolds made easier☆87Updated last year
- ☆80Updated 3 years ago
- ☆53Updated 7 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 7 months ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Matrix-free linear algebra in JAX.☆115Updated last month
- Implementations of Normalizing Flows in Pytorch/Pyro☆19Updated 4 years ago
- Convex potential flows☆83Updated 3 years ago
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago