JAX-BTE: a GPU-accelerated differentiable solver for phonon Boltzmann transport equations

Abstract This paper introduces JAX-BTE, a GPU-accelerated, differentiable solver for the phonon Boltzmann Transport Equation (BTE) based on differentiable programming. JAX-BTE enables accurate, efficient and differentiable multiscale thermal modeling by leveraging high-performance GPU computing and...

Full description

Saved in:
Bibliographic Details
Main Authors: Wenjie Shang, Jiahang Zhou, J. P. Panda, Zhihao Xu, Yi Liu, Pan Du, Jian-Xun Wang, Tengfei Luo
Format: Article
Language:English
Published: Nature Portfolio 2025-05-01
Series:npj Computational Materials
Online Access:https://doi.org/10.1038/s41524-025-01635-0
Tags: Add Tag
No Tags, Be the first to tag this record!