PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning

arXiv:2609.36052 · cs.AI, cs.LG, cs.SY, eess.SY · Submitted 2026-09-28 · Read on arXiv

cs.AI, cs.LG, cs.SY, eess.SY

Submitted: 2026-09-28

Updated: 2026-09-28

Code: https://github.com/powerzoojax/PowerZooJax

Terminology

• Benchmark — 58× in this paper · explained in 2 episodes
• Reinforcement Learning — 17× in this paper · explained in 4 episodes
• Action Space — 15× in this paper · explained in 1 episode
• Trace — 13× in this paper · explained in 2 episodes
• Actor — 8× in this paper · explained in 3 episodes
• Benchmarking — 5× in this paper · explained in 3 episodes
• Adam — 4× in this paper · explained in 2 episodes
• Fine-Tuning — 4× in this paper · explained in 5 episodes
• Distribution System Operator — 3× in this paper
• National Energy System Operator — 3× in this paper
• Reward Function — 3× in this paper · explained in 2 episodes
• Aggregation — 2× in this paper · explained in 1 episode
• Combinatorial Optimization — 2× in this paper · explained in 2 episodes
• Credit assignment — 2× in this paper · explained in 3 episodes
• Great Britain — 2× in this paper
• Reinforcement Learning (RL) — 2× in this paper · explained in 25 episodes
• Transmission System Operator — 2× in this paper
• Artificial Intelligence (AI) — 1× in this paper · explained in 2 episodes
• Communication Overhead — 1× in this paper · explained in 2 episodes
• Core — 1× in this paper · explained in 2 episodes
• Deep Reinforcement Learning — 1× in this paper · explained in 2 episodes
• GB Market Index Data — 1× in this paper
• Generalization — 1× in this paper · explained in 13 episodes
• Nanyang Technological University — 1× in this paper
• Physical Constraints — 1× in this paper · explained in 7 episodes
• Transformer — 1× in this paper · explained in 2 episodes
• Verification — 1× in this paper · explained in 2 episodes

Related papers

• MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
• Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
• The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
• MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
• Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems
• VSAL: A Vision Solver with Adaptive Layouts for Graph Property Detection