基于JAX的高性能离散事件仿真方法及系统

本发明提供了一种基于JAX的高性能离散事件仿真方法及系统，包括：首先通过不可变数据结构完成环境配置与初始化，生成固化静态参数的EnvParams对象；其次采用纯函数step_env处理事件逻辑，结合JAX条件控制原语与lax.scan迭代循环实现无副作用的函数式事件处理与状态迭代；再通过作业封装生成批处理任务，借助jax.jit编译与jax.vmap向量化，在多GPU/TPU设备上实现作业级并行加速；最后封装标准化AI交互接口，支持外部算法调用并完成结果后处理。本发明通过函数式编程与JAX硬件加速结合，提升仿真速度的同时保障事件动态准确性，可无缝支持强化学习大规模采样，适用于运筹学领域AI应用场景。