https://github.com/fluiddyn/transonic can accelerate Python/Numpy code and run on accelerators