thanks for the work, this looks amazing. Have you considered trying numba for acceleration? It's great for number crunching.