This port is a good pedagogical tool to learn how to implement traditional numerical solvers using modern deep learning software interfaces. The main changes are documented in the torch_cfd directory.

Small. With bf16/fp16 (supported by native pytorch), our baseline could be trained with only 2GB GPU memory. Friendly. You may use the off-the-shelf options to apply many state-of-the-art tricks in ...