HEmile / storchastic
Stochastic Automatic Differentiation library for PyTorch.
☆180Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for storchastic
- ☆97Updated 2 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆69Updated 3 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆105Updated last year
- ☆50Updated last year
- Riemannian Convex Potential Maps☆68Updated last year
- Discrete Normalizing Flows implemented in PyTorch☆107Updated 3 years ago
- Differentiable Sorting Networks☆107Updated last year
- Hessian spectral density estimation in TF and Jax☆115Updated 4 years ago
- ☆49Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆39Updated 4 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆140Updated last year
- ☆159Updated 3 months ago
- ☆99Updated 4 months ago
- Optimization with orthogonal constraints and on general manifolds☆126Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆68Updated 2 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆170Updated 2 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆107Updated this week
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Sketched matrix decompositions for PyTorch☆67Updated 6 months ago
- Official repository for "Categorical Normalizing Flows via Continuous Transformations"☆56Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆127Updated last month
- Convex potential flows☆78Updated 3 years ago
- ☆213Updated 2 years ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆48Updated 9 months ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆128Updated 5 years ago