einsum-notation

There are 0 repository under einsum-notation topic.

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