pytorch / extension-cppLinks
C++ extensions in PyTorch
☆1,092Updated 4 months ago
Alternatives and similar repositories for extension-cpp
Users that are interested in extension-cpp are comparing it to the libraries listed below
Sorting:
- ATen: A TENsor library for C++11☆694Updated 5 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,342Updated this week
- Tutorial for building a custom CUDA function for Pytorch☆515Updated 6 years ago
- PyTorch C++ API Documentation☆228Updated last week
- A GPU performance profiling tool for PyTorch models☆503Updated 3 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,040Updated last year
- common in-memory tensor structure