Open andrekv17 opened 2 years ago
Thank you for sharing! What speedup do you get with numba?
The numba code runs roughly 3 times faster on google colab compared to numpy. Everything is already vectorized, so:
OK, that's already nice! I will add a link to this issue in the README.
Numba support decrease the computation time up to nearly 5 times. No much changes are required
Works for numba==0.53.1 and numpy==1.21.1