5 skills found
f-dangel / EinconvConvolutions and more as einsum for PyTorch
M-e-r-c-u-r-y / Pytorch TransformersCollection of different types of transformers for learning purposes
baahl-nyu / EinhopsEinsum Notation for FHE
ShawnZhong / CS759 Spring 2020 Final ProjectOptimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
adadima / GstacoA compiler based on the Einsum notation for sparse tensor algebra compilation