الملخص
Power system operation is a safety-critical sequential decision-making problem, making it a natural testbed for reinforcement learning (RL). However, existing RL environments for power systems are often narrow in scope and computationally limited by CPU-based simulation workflows, making large-scale evaluation difficult. We introduce PowerZooJax, a JAX-based benchmark suite for RL in power system operation. It provides five constrained Markov decision process tasks spanning generation, transmission, distribution, distributed energy resources, and data center microgrid. By rewriting power flow, economic dispatch, market clearing, and device dynamics as JAX computation graphs, PowerZooJax keeps the entire training and evaluation loop on the GPU. Experiments show substantial speedups over CPU-based simulations and demonstrate standardized evaluation of policy returns, safety violations, and out-of-distribution stress conditions. Our open-source benchmark is available at: https://github.com/powerzoojax/PowerZooJax.
الكلمات المفتاحية
الموضوع
بيانات النشر
- المجلة
- غير متاح
- وصول مفتوح
- وصول مفتوح أخضر
اقتبس هذه المقالة
APA 7
Pan, Z., Liu, X., Cao, Z., Wang, J., & Qiu, D. (2026). PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning. https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning
MLA 9
Pan, Zhanhua, et al. "PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning." https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning.
شيكاغو (المؤلف–التاريخ)
Pan, Zhanhua, Xiao Liu, Zhilong Cao, Jianhong Wang, and Dawei Qiu. 2026. "PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning." https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning.
هارفارد
Pan, Z., Liu, X., Cao, Z., Wang, J. and Qiu, D. (2026) 'PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning', Available at: https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning.
فانكوفر
Pan Z, Liu X, Cao Z, Wang J, Qiu D. PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning. https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning
IEEE
Z. Pan, X. Liu, Z. Cao, J. Wang, and D. Qiu, "PowerZooJax: A JAX-based Power System Benchmark for Reinforcement Learning," https://omanscience.com/ar/articles/powerzoojax-a-jax-based-power-system-benchmark-for-reinforcement-learning.