Some possibilities include:
- Doing operations directly on
numpy.ndarray objects rather than astropy.units.Quantity objects, since doing stuff with units can slow things down.
Additionally, it would be helpful to have the NEI time advance put into a separate function that deals directly with numpy.ndarray objects. That function could then be decorated with the appropriate Numba decorator, or alternatively implemented in Cython.