Solutions to Sasha Rushes GPU puzzles. Implemented with NUMBA in Python 3.10+. Try running this in Google Collab and connecting your GPU to the "Runtime" by following the steps to setup a local runtime.
As you run this notebook and change things, the graphs in the notebook will update. Try with CUDA.jl instead next?