Comment by FattiMei

A research project on converting a python application from loops to numpy vectorization to jax and pytorch compilation (the latter is fast!). I'm exploring the performance portability of such frameworks across different devices and backends (also gpu). I have also C++ hand optimized implementations that so far are unbeaten