phiflow 3.1.0

Differentiable PDE solving framework for machine learning

PhiFlow

Homepage     Documentation     API     Demos     Fluids Tutorial     Playground

PhiFlow is an open-source simulation toolkit built for optimization and machine learning applications. It is written mostly in Python and can be used with NumPy, TensorFlow, Jax or PyTorch. The close integration with machine learning frameworks allows it to leverage their automatic differentiation functionality, making it easy to build end-to-end differentiable functions involving both learning models and physics simulations.

See the installation Instructions on how to compile the optional custom CUDA operations.