XLB 是一款 基于 Python 的可微分 LBM 库，它利用了 JAX 框架的功能。XLB 的架构旨在确保可访问性、可扩展性和计算性能，从而能够跨 CPU、多 GPU 和分布式多 GPU 系统进行有效扩展。该框架可以轻松地通过新颖的边界条件、碰撞模型或模拟功能进行增强。XLB 具有与 JAX 广泛的机器学习回声系统集成的独特优势，并且能够利用自动微分来解决基于物理的机器学习、优化和逆问题。