XLB:Python 中的可分化大規模平行網格博爾茨曼庫

https://arxiv.org/abs/2311.16080

XLB 是一個基於 Python 的 LBM(lattice Boltzmann method)程式庫,採用 JAX 平台,為可微分的並行運算工具。該程式庫具備擴展性、效能和存取性,能有效運行於 CPU、TPU、多 GPU 和分佈式多 GPU 或 TPU 系統上,並支援機器學習生態系統,應用於物理學機器學習、優化和反問題等領域。符合 Apache-2.0 授權,並在 GitHub 上公開。

via physics updates on arXiv.org

April 3, 2024 at 12:45PM

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *