You can use Numba which is for generic accelerated work, or really any ML library will have math primatives you can use just fine for non ML purposes. That said

You can use Numba which is for generic accelerated work, or really any ML library will have math primatives you can use just fine for non ML purposes. That said there are overhead considerations and restrictions compared to normal Python programming for any of these solutions so make sure you really need the GPU.