Sticking to NVIDIA-specific software has some downsides, however: It won't run on modern Mac laptops. Testing in CI is more difficult: you need custom runners that have NVIDIA GPUs. You can't use any other GPUs you might have access to, like AMD GPUs.
wgpu-py is a Python wrapper for wgpu, a Rust library that re-purposes the WebGPU API for environments outside the browser.
[
Collection
]
[
|
...
]