In this repository you can find examples showing how to extend PyTorch with
custom C code. To use the ffi you need to install the cffi package from pip.
Currently there are two examples:
package- a pip distributable packagescript- compiles the code into a local module, that can be later imported from other files
You may also want to look at pytorch/audio for an example of a type-generic C FFI extension.