CUDA is basically a C++ extension for NVIDIA GPUs. Basically what PyTorch uses under the hood.

→ View original post on X — @rasbt