pytorch / extension-cppLinks
C++ extensions in PyTorch
☆1,152Updated 3 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:
- Tutorial for building a custom CUDA function for Pytorch☆518Updated 6 years ago
- ATen: A TENsor library for C++11☆704Updated 5 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆988Updated this week
- TVM integration into PyTorch☆454Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆508Updated 4 years ago
- PyTorch layer-by-layer model profiler☆606Updated 4 years ago
- A GPipe implementation in PyTorch☆857Updated last year
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,062Updated 2 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,459Updated this week
- PyTorch C++ API Documentation