Brian DuSell

Generic PyTorch implementation of einsum that supports different semirings

CUDA kernels for generalized matrix-multiplication in PyTorch

Code for the paper "The Surprising Computational Power of Nondeterministic Stack RNNs" (DuSell and Chiang, 2023)

Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"