f-dangel / einconv Star 14 Code Issues Pull requests Convolutions and more as einsum for PyTorch fisher pytorch convolution ntk transpose-convolution im2col kfac einsum tensornetwork einsum-notation vjp jvp ggn Updated Jun 6, 2024 Python
M-e-r-c-u-r-y / pytorch-transformers Star 13 Code Issues Pull requests Collection of different types of transformers for learning purposes transformers pytorch multi-head-attention einsum-notation multi-query-attention Updated Jan 30, 2020 Jupyter Notebook
ShawnZhong / CS759-Spring-2020-Final-Project Star 7 Code Issues Pull requests Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction benchmarking performance pytorch dynamic-programming profiling performance-optimization einsum cutensor einsum-notation Updated Sep 25, 2020 Python