JAX-BTE: A GPU-Accelerated Differentiable Solver for Phonon Boltzmann Transport Equations

Wenjie Shang, Jiahang Zhou, J. P. Panda +5

This paper introduces JAX-BTE, a GPU-accelerated, differentiable solver for the phonon Boltzmann Transport Equation (BTE) based on differentiable programming. JAX-BTE enables accur…