Now that https://github.com/pytorch/vision/pull/1314 has been merged, there is no more need to use C++ extensions based on pybind11, we can use TorchScript-compatible extensions instead.