Since the code is written in `Numpy` and `Numba` supports `Numpy` how fast will it take the code to run one epoch?