基于 JAX，Chromatix 提供了一个用于开发、共享和复用波动光学仿真的标准化框架。其功能已在多种计算显微成像应用中得到验证。