Towards Understanding the Unreasonable Effectiveness of Learning AC-OPF Solutions. - My H. Dinh; Ferdinando Fioretto; Mostafa Mohammadian; Kyri Baker (CoRR, 2021)

Gradient-Enhanced Physics-Informed Neural Networks for Power Systems Operational Support. - Mostafa Mohammadian; Kyri Baker; Ferdinando Fioretto (CoRR, 2022)

Learning to Optimize meets Neural-ODE: Real-Time, Stability-Constrained AC OPF. - Vincenzo Di Vito; Mostafa Mohammadian; Kyri Baker; Ferdinando Fioretto (CoRR, 2024)

Learning to Solve Differential Equation Constrained Optimization Problems. - Vincenzo Di Vito Francesco; Mostafa Mohammadian; Kyri Baker; Ferdinando Fioretto (ICLR, 2025)

Restoring Feasibility in Power Grid Optimization: A Counterfactual ML Approach. - Mostafa Mohammadian; Anna Van Boven; Kyri Baker (CoRR, 2025)