Fig. presents the computational cost of 2DES simulation at various levels of vectorization for different laser pulses as a function of the number of time steps for a single laser pulse. The red curve is the computational cost FIG. 2. Computational cost scaling as a function of the number of time steps (T ) of each laser pulse delay at different levels of vectorisation of Eq. 39. The red curve denotes the computational cost of serial computation for a single trajectory of three laser pulses. The yellow curve denotes the computational cost when one of the lasers is vectorised. The green curve is the case of vectorisation along two laser pulses and the blue curve denotes vectorization along all the laser delays. Each curve involves propagating a wavefunction in the 2D excitation manifold of ≈ 50 molecules with around 2500 total quantum states. The above calculations were performed on NVIDIA Hopper H100 GPUs using pytorch with device "meta". Each of the curve above is averaged over 10 6 instances of Eq. 39 operations.