Numba Simple
from numba import njit, prange
import numpy as np
@njit(parallel=True)
def add(A):
s = 0
# Without "parallel=True" in the jit-decorator
# the prange statement is equivalent to range
for i in prange(A.shape[0]):
s += A[i]
return s
abc = add(np.array([1, 2, 3]))
abc
6
More on Numba
http://numba.pydata.org/numba-doc/latest/user/parallel.html#numba-parallel
http://numba.pydata.org/numba-doc/latest/index.html