Build Accelerated, Differentiable Computational Physics Code for AI with NVIDIA Warp | NVIDIA Technical Blog
…The following rk3_update() kernel computes the diffusion and the advection terms and performs a single RK3 substep update. The step() function calls this kernel three times per timestep, once for each…