einsum-notation

There are 3 repositories under einsum-notation topic.

f-dangel/einconv
Convolutions and more as einsum for PyTorch
Language:Python13 3 20
M-e-r-c-u-r-y/pytorch-transformers
Collection of different types of transformers for learning purposes
Language:Jupyter Notebook13 1 20
ShawnZhong/CS759-Spring-2020-Final-Project
Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
Language:Python7 1 00