A time-stepping deep gradient flow method for option pricing in (rough) diffusion models