PowerZooJax：基于 JAX 的电力系统强化学习基准

PowerZooJax 是一个基于 JAX 的电力系统强化学习基准套件，涵盖发电、输电、配电、分布式能源和数据中心微电网五类约束马尔可夫决策任务。通过将潮流、经济调度、市场出清和设备动力学重写为 JAX 计算图，整个训练与评估循环保持在 GPU 上，实验显示相比 CPU 仿真有显著加速，并实现了策略回报、安全违规和分布外压力条件的标准化评估。该开源基准已发布于 arXiv。