Will be interesting to compare the API to what Modular has [1] with Mojo.

1. https://docs.modular.com/mojo/stdlib/gpu/