HEmile / storchastic
Stochastic Automatic Differentiation library for PyTorch.
☆197Updated 5 months ago
Alternatives and similar repositories for storchastic:
Users that are interested in storchastic are comparing it to the libraries listed below
- ☆98Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 6 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆112Updated last year
- Hessian spectral density estimation in TF and Jax☆121Updated 4 years ago
- Differentiable Sorting Networks☆110Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆83Updated 2 years ago
- Normalizing Flows using JAX☆82Updated last year
- Differentiable Optimizers with Perturbations in Pytorch☆65Updated 3 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆183Updated 2 months ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆49Updated last year
- ☆50Updated last year
- ☆80Updated 3 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆31Updated this week
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 7 months ago
- ☆49Updated 4 years ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆142Updated 4 months ago
- ☆236Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 2 years ago
- ☆53Updated 6 months ago
- ☆164Updated 6 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆104Updated 4 years ago
- Convex potential flows☆82Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Updated 5 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated 2 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆107Updated 2 years ago