Forward and backward Attention DNN operators implementationed by LibTorch, cuDNN, and Eigen.
☆31Jun 6, 2023Updated 3 years ago
Alternatives and similar repositories for eigenMHA
Users that are interested in eigenMHA are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- The only known (by 2022) open-source, easy-to-understand basic algorithm implementations in TD-CEM. (Please star and fork this project if…☆15Mar 1, 2022Updated 4 years ago
- Overlapping Schwarz Domain Decomposition Finite Element Algorithm in both Matlab and serial/parallel C++☆18Mar 1, 2022Updated 4 years ago
- CUDA 8-bit Tensor Core Matrix Multiplication based on m16n16k16 WMMA API