Optimize array stacking in the output by using NumPy rather than JAX.#586

Open

copybara-service[bot] wants to merge 1 commit intomainfrom test_701895415

+11-4

Commits on Dec 20, 2024

Optimize array stacking in the output by using NumPy rather than JAX.

sbodenstein
authored and
Torax team
committed