https://singularityfeed.com/speed-up-pytorch-with-custom-kernels-alex-dremov/