Or you use numba [1]. Then you can use a subset of plain Python.

[1] https://numba.pydata.org/