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
Sorting:
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22May 5, 2019Updated 6 years ago
- fermionic Projected Entangled Pairs States (fPEPS)☆16Oct 25, 2018Updated 7 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago
- Infinite order automatic differentiation for Monte Carlo with unnormalized probability distribution☆24Mar 24, 2023Updated 2 years ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆50Dec 4, 2025Updated 3 months ago
- Official Repository for Efficient Linear-Time Attention Transformers.☆18Jun 2, 2024Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated last year
- A tool for weighted model counting through tensor network contraction☆22Nov 13, 2021Updated 4 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
- Tutorials for the paper: "Neural network in quantum many-body physics: a hands-on tutorial"☆53Feb 7, 2022Updated 4 years ago
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆14Dec 11, 2021Updated 4 years ago
- [AAAI 2020 Oral] Low-variance Black-box Gradient Estimates for the Plackett-Luce Distribution☆39Jan 12, 2021Updated 5 years ago
- Arbitrary order exchange-correlation functional derivatives using JAX.☆22Jul 12, 2020Updated 5 years ago
- Variable-order CRFs with structure learning☆17Aug 1, 2024Updated last year
- Estimating Gradients for Discrete Random Variables by Sampling without Replacement☆40Feb 14, 2020Updated 6 years ago
- Sparse Circuits on the GPU (ICLR2025)☆21Dec 25, 2025Updated 2 months ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Jan 12, 2026Updated last month
- Building blocks for simple and advanced particle filtering in Gen.☆27Mar 19, 2024Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19May 8, 2025Updated 9 months ago
- Riemannian optimization for quantum technologies☆65Mar 25, 2023Updated 2 years ago
- ☆29Oct 3, 2022Updated 3 years ago
- Generic API for dispatch to Pyro backends.☆16Feb 13, 2022Updated 4 years ago
- Macros to define custom adjoints for TensorOperations.jl☆14Aug 17, 2021Updated 4 years ago
- ☆15Dec 28, 2020Updated 5 years ago
- Library for normalizing flows and neural flows.☆26May 26, 2022Updated 3 years ago
- Functional tensors for probabilistic programming☆245Aug 31, 2023Updated 2 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32May 25, 2024Updated last year
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated 3 weeks ago
- Neural Monte Carlo Renormalization Group☆15Oct 12, 2020Updated 5 years ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)☆10Feb 21, 2023Updated 3 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 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 2 years ago
- Official Implementation of ACL2023: Don't Parse, Choose Spans! Continuous and Discontinuous Constituency Parsing via Autoregressive Span …☆14Aug 25, 2023Updated 2 years ago
- Code in support of the paper Continuous Mixtures of Tractable Probabilistic Models☆12Oct 12, 2024Updated last year
- ☆12Jan 29, 2021Updated 5 years ago
- Official implementation of the paper "GRAPES: Learning to Sample Graphs for Scalable Graph Neural Networks".☆13Apr 1, 2025Updated 11 months ago