JAX-BTE: a GPU-accelerated differentiable solver for phonon Boltzmann transport equations
Wenjie Shang, Jiahang Zhou, J. Panda, Zhihao Xu, Yi Liu, Pan Du, Jian-Xun Wang, Tengfei Luo
npj Computational Materials