Stochastic Automatic Differentiation library for PyTorch.
☆209Aug 30, 2024Updated last year
Alternatives and similar repositories for storchastic
Users that are interested in storchastic are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22May 5, 2019Updated 6 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Feb 14, 2020Updated 6 years ago
- Sparse Circuits on the GPU (ICLR2025)☆21Mar 17, 2026Updated last week
- fermionic Projected Entangled Pairs States (fPEPS)☆15Oct 25, 2018Updated 7 years ago
- Infinite order automatic differentiation for Monte Carlo with unnormalized probability distribution☆24Mar 24, 2023Updated 3 years ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- Arbitrary order exchange-correlation functional derivatives using JAX.☆22Jul 12, 2020Updated 5 years ago
- Official implementation of the paper "GRAPES: Learning to Sample Graphs for Scalable Graph Neural Networks".☆13Apr 1, 2025Updated 11 months ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆78Jul 25, 2024Updated last year
- A list of resources dedicated to compositionality☆14Feb 21, 2019Updated 7 years ago
- Variable-order CRFs with structure learning☆17Aug 1, 2024Updated last year
- Generic PyTorch implementation of einsum that supports different semirings☆50Dec 4, 2025Updated 3 months ago
- ☆30Oct 26, 2020Updated 5 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆38Jan 12, 2021Updated 5 years ago
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆290Feb 1, 2021Updated 5 years ago
- A tool for weighted model counting through tensor network contraction☆21Nov 13, 2021Updated 4 years ago
- Code in support of the paper Continuous Mixtures of Tractable Probabilistic Models☆12Oct 12, 2024Updated last year
- Building blocks for simple and advanced particle filtering in Gen.☆27Mar 19, 2024Updated 2 years ago
- ☆14Feb 1, 2024Updated 2 years ago
- Official Repository for Efficient Linear-Time Attention Transformers.☆18Jun 2, 2024Updated last year
- Probabilistic inference using contraction of tensor networks☆25Sep 9, 2025Updated 6 months ago
- Tutorials for the paper: "Neural network in quantum many-body physics: a hands-on tutorial"☆53Feb 7, 2022Updated 4 years ago
- ☆15Dec 28, 2020Updated 5 years ago
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆14Dec 11, 2021Updated 4 years ago
- Riemannian optimization for quantum technologies☆66Mar 25, 2023Updated 3 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Jul 1, 2020Updated 5 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Oct 7, 2024Updated last year
- ☆11Oct 11, 2023Updated 2 years ago
- PyPSDD porting to Python 3 + PyTorch equivalent tree construction.☆16Jun 7, 2023Updated 2 years ago
- Finite-temperature variational Monte Carlo calculation of uniform electron gas using neural canonical transformation.☆16Jun 17, 2023Updated 2 years ago
- A method for evaluating the high-level coherence of machine-generated texts. Identifies high-level coherence issues in transformer-based …☆11Mar 18, 2023Updated 3 years ago
- Repository for the NeurIPS 2023 paper "Beyond Confidence: Reliable Models Should Also Consider Atypicality"☆13Apr 21, 2024Updated last year
- Source code of ACL2022 "Headed-Span-Based Projective Dependency Parsing" and "Combining (second-order) graph-based and headed-span-based …☆16Jan 12, 2023Updated 3 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆14Nov 26, 2019Updated 6 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,163Updated this week
- Functional tensors for probabilistic programming☆245Aug 31, 2023Updated 2 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago