PyTorchHSE

Toy problem to learn a hydrostatic equilibirum (HSE) state as a function of height and potential temperature. The learned model will then be integrated in the ERF code base as a proof of concept.