pytorch-custom-c-cuda-operations. Example of easily implementing custom tensor operations in C and CUDA.

github.com/davidstutz/pytorch-custom-c-cuda-operations

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.