You might want to take a look at https://discuss.python.org/t/what-to-do-about-gpus-and-the-built-distributions-that-support-them/7125 https://discuss.python.or

You might want to take a look at https://discuss.python.org/t/what-to-do-about-gpus-and-the-built-distributions-that-support-them/7125 https://discuss.python.org/t/what-to-do-about-gpus-and-the-b... if you haven't seen it - there's a practical problem that hosting built GPU code on PyPI is very difficult.