If we import the math functions from numpy instead would that make it work with parakeet as well? I don't think python and numba performance is significantly impacted by the change.