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

开源评测基准 PowerZooJax 基于 JAX 构建，将电力系统强化学习的完整训练与评估流程保留在 GPU 上。它涵盖发电、输电、配电、分布式能源和数据中心微电网 5 个任务，并将潮流计算、经济调度及设备动态重写为 JAX 计算图。其相比 CPU 仿真实现大幅加速，支持对策略回报、安全违规和分布外压力条件的标准化评测。