For people looking for a version that is easier to work with out of the box, please see the inspiration for this work, by Oliver Hennigh:
https://github.com/loliverhennigh/Steady-State-Flow-With-Neural-Nets
For people looking for a version that is easier to work with out of the box, please see the inspiration for this work, by Oliver Hennigh:
https://github.com/loliverhennigh/Steady-State-Flow-With-Neural-Nets