https://pytorch.org/blog/compiling-numpy-code/
Compiling NumPy code into C++ or CUDA via torch.compile
by Evgeni Burovski, Ralf Gommers and Mario Lezcano
October 17, 2023
“This feature leverages PyTorch’s compiler to generate efficient fused vectorized code without having to modify your original NumPy code. Even more, it also allows for executing NumPy code on CUDA just by running it through torch.compile under torch.device(“cuda”)!”