You beat me to making this point. I don't see anything in the article's python code that the numba's jit decorator cannot handle. When numba works (it's rapidl

You beat me to making this point. I don't see anything in the article's python code that the numba's jit decorator cannot handle. When numba works (it's rapidly improving), it's seriously impressive. For this particular case, you should be able to get really good performance without sacrificing readability. Also Jax - https://github.com/google/jax https://github.com/google/jax