2 ms·

Numba would probably do the job too