v_stability_sweep vectorized, but that means that cells that aren't doing anything useful continue to get computed until all cells in the vector are finished
stability_sweep computes each cell individually, so it doesn't make good use of the GPU but does end simulations as soon as they explode/vanish or reach max_t
(Both JAX versions)
v_stability_sweep
vectorized, but that means that cells that aren't doing anything useful continue to get computed until all cells in the vector are finishedstability_sweep
computes each cell individually, so it doesn't make good use of the GPU but does end simulations as soon as they explode/vanish or reachmax_t