#

einsum-notation

Here are 3 public repositories matching this topic...

M-e-r-c-u-r-y / pytorch-transformers

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

f-dangel / einconv

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

ShawnZhong / CS759-Spring-2020-Final-Project

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

Improve this page

Add a description, image, and links to the einsum-notation topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the einsum-notation topic, visit your repo's landing page and select "manage topics."