ML for optimal power flow

I’d like to use neural networks to estimate some variable on an optimal power flow problem. For example, I’d like to estimate the power flow on transmission lines, given the system topology (generation, demand, line parameters). What is the best Julia package to do that?