Doing it via a physics-informed neural network could work if the stiffness is sufficiently low. GPU is fine though: Float32 is a lot easier to handle than BFloat32.
That does seem like something to try a PINN on, and it’s something we’re looking into.