JAX-Based Batched AC Power Flow for GPU Acceleration and AI Ecosystem Integration

Yihong Zhou, Dylan Cope, Jakob Foerster +1

Coordinating growing grid flexibility under uncertainty is becoming increasingly important for efficient and reliable power-system operation. A core computational requirement is th…