Python. That's really all you need to know. 5 lines of Python.
psi = psi * np.exp(-1j * V * dt / 2) psi = np.fft.fft2(psi) psi = psi * np.exp(-1j * (np.fft.fftfreq(Nx, dx)**2 + np.fft.fftfreq(Ny, dy)**2) * dt) psi = np.fft.ifft2(psi) psi = psi * np.exp(-1j * V * dt / 2)
And that's all the thinking here. The rest is theory and complex plots.
Docs (built with pdoc)