HEmile / storchastic
Stochastic Automatic Differentiation library for PyTorch.
☆197Updated 6 months ago
Alternatives and similar repositories for storchastic:
Users that are interested in storchastic are comparing it to the libraries listed below
- ☆99Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 8 months ago
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- Differentiable Sorting Networks☆114Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- Differentiable Algorithms and Algorithmic Supervision.☆114Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆105Updated 4 years ago
- ☆52Updated last year
- ☆49Updated 4 years ago
- Discrete Normalizing Flows implemented in PyTorch☆111Updated 3 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆573Updated 2 months ago
- Riemannian Convex Potential Maps☆67Updated last year
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆141Updated last year
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆145Updated last year
- ☆192Updated 10 months ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- ☆80Updated 3 years ago
- Convex potential flows☆83Updated 3 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated 2 weeks ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆34Updated 3 weeks ago
- Normalizing Flows using JAX☆83Updated last year
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆48Updated 5 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆130Updated 5 years ago
- ☆240Updated 2 years ago