An existing 2D solver (CPU and GPU) for the Navier-Stokes equations in a supersonic regime is taken, and the most computationally expensive part is replaced by a neural network.