The first part of the simplification is to utilise the excellent NUMBA python JIT compiler to allow easy-to-understand code to be deployed as GPU machine code. Working on these puzzles is even ...
If you want to fully control your GPU on Linux you may want to check out the latest update to Linux GPU Configuration Tool (LACT), which brings in various new features.