Asymptotic-Preserving Neural Networks for Solving Multiscale Kinetic Equations

This repository contains the PyTorch/TensorFlow code for the following papers

Shi Jin, Zheng Ma, and Keke Wu. Asymptotic-preserving neural networks for multiscale time-dependent linear transport equations. J. Sci. Comput., 94(3):57, 2023. [PDF]

Shi Jin, Zheng Ma, and Keke Wu. Asymptotic-preserving neural networks for multiscale kinetic equations. Commun. Comput. Phys., 35(2024), pp.693-723.[PDF]

Setup

Environment:

• PyTorch
• TensorFlow