Neural partial differential equations for chaotic systems