In [6]:

    

import numba
import numpy as np

def sqrt(x):
    return x**0.5

sqrt_jitted = numba.njit(sqrt)

@numba.njit
def sqrt_jitted_as_decorator(x):
    return x**0.5

x = np.arange(100000)


    

In [7]:

    

%timeit sqrt(x)


    

    




49.4 ms ± 8.77 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

In [8]:

    

%timeit sqrt_jitted(x)


    

    




4.66 ms ± 70.3 µs per loop (mean ± std. dev. of 7 runs, 1 loop each)

In [9]:

    

%timeit sqrt_jitted_as_decorator(x)


    

    




4.38 ms ± 50.8 µs per loop (mean ± std. dev. of 7 runs, 1 loop each)