Take a look into numba, you add a decorator to your function and it turns it into compiled code.