CUDA kernels for generalized matrix-multiplication in PyTorch
☆85Oct 11, 2021Updated 4 years ago
Alternatives and similar repositories for genbmm
Users that are interested in genbmm are comparing it to the libraries listed below
Sorting:
- Generic PyTorch implementation of einsum that supports different semirings☆50Dec 4, 2025Updated 3 months ago
- Python package to heuristically solve optimization problems on quasi-2d lattices using approximate tensor network contractions.☆12Sep 24, 2021Updated 4 years ago
- A tutorial for Yao.jl☆12Oct 9, 2023Updated 2 years ago
- Dump a quantum circuit in Yao to a tensor network graphical model☆11Dec 11, 2020Updated 5 years ago
- ☆12Jan 29, 2021Updated 5 years ago
- Can you solve all the puzzles in this repo? Master Julia in an easy way!☆22Sep 20, 2022Updated 3 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated last year
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Apr 28, 2020Updated 5 years ago
- ☆14Feb 1, 2024Updated 2 years ago
- ☆12Dec 13, 2022Updated 3 years ago
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22May 5, 2019Updated 6 years ago
- Autoregressive model for solving statistical mechanics problems☆11Mar 19, 2021Updated 4 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- ☆23Jul 23, 2021Updated 4 years ago
- ☆17Dec 19, 2024Updated last year
- Initial attempts at a package for defining many body systems in both quantum and statistical physics☆13May 31, 2021Updated 4 years ago
- Variable-order CRFs with structure learning☆17Aug 1, 2024Updated last year
- ☆12Jan 9, 2019Updated 7 years ago
- Legendre decomposition for tensors☆13Oct 29, 2018Updated 7 years ago
- Expanding linear RNN state-transition matrix eigenvalues to include negatives improves state-tracking tasks and language modeling without…☆21Mar 15, 2025Updated 11 months ago
- Code for the paper "The Surprising Computational Power of Nondeterministic Stack RNNs" (DuSell and Chiang, 2023)☆19Mar 21, 2024Updated last year
- ☆20Nov 19, 2023Updated 2 years ago
- The Return of Lexical Dependencies: Neural Lexicalized PCFGs (TACL)☆33Sep 22, 2025Updated 5 months ago
- Code for "Variational Sequential Labelers for Semi-Supervised Learning" (EMNLP 2018)☆34Oct 4, 2018Updated 7 years ago
- Python implementation of projection losses.☆27Nov 18, 2019Updated 6 years ago
- Code Repository for "Please Mind the Root: Decoding Arborescences for Dependency Parsing" and "On Finding the K-best Non-projective Depen…☆20Dec 12, 2022Updated 3 years ago
- ☆15Dec 28, 2020Updated 5 years ago
- Macros to define custom adjoints for TensorOperations.jl☆14Aug 17, 2021Updated 4 years ago
- ☆53Aug 22, 2023Updated 2 years ago
- Implementation of a fast exponential matrix for large matrices (full and sparse)☆32Mar 8, 2022Updated 3 years ago
- Continuous Matrix Product States in Julia☆30Sep 24, 2025Updated 5 months ago
- cuASR: CUDA Algebra for Semirings☆45Aug 22, 2022Updated 3 years ago
- Official Repository for Efficient Linear-Time Attention Transformers.☆18Jun 2, 2024Updated last year
- fermionic Projected Entangled Pairs States (fPEPS)☆16Oct 25, 2018Updated 7 years ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- ☆133Sep 17, 2023Updated 2 years ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 5 years ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆17Oct 11, 2021Updated 4 years ago
- A tool for weighted model counting through tensor network contraction☆22Nov 13, 2021Updated 4 years ago