ΦFlow is a research-oriented, open-source fluid simulation toolkit. It is written mostly in Python and can use both NumPy and TensorFlow for execution.
Having all functionality of a fluid simulation running in TensorFlow opens up the possibility of back-propagating gradients through the simulation as well as running the simulation on GPUs.
- Support for a variety of differentiable simulation types, from Burgers over Navier-Stokes to the Schrödinger equation.
- Tight integration with TensorFlow allowing for straightforward network training with fully differentiable simulations that run on the GPU.
- Object-oriented architecture enabling concise and expressive code, designed for ease of use and extensibility.
- Reusable simulation code, independent of backend and dimensionality, i.e. the exact same code can run a 2D fluid sim using NumPy and a 3D fluid sim on the GPU using TensorFlow.
- Flexible, easy-to-use web interface featuring live visualizations and interactive controls that can affect simulations or network training on the fly.
The following commands will get you ΦFlow + browser-GUI + NumPy execution:
$ pip install phiflow[gui]
See the detailed installation instructions on how to install ΦFlow with TensorFlow support.
Documentation and Guides
|Index||Demos / Tests||Source||Fluids Tutorial / Playground|
If you would like to get right into it and have a look at some code, check out the tutorial notebook on Google Colab. It lets you run fluid simulations with ΦFlow in the browser.
The following introductory demos are also helpful to get started with writing your own app using ΦFlow:
- simpleplume.py: Runs a fluid simulation and displays it in the browser
- optimize_pressure.py: Uses TensorFlow to optimize a velocity channel. TensorBoard can be started from the GUI and displays the loss.
Writing a ΦFlow Application introduces the high-level classes and explains how to use the ΦFlow GUI for displaying a simulation.
Optimization and Learning
For training machine learning models, this document gives an introduction into writing a GUI-enabled application.
The simulation code design documentation provides a deeper look into the object-oriented code design of simulations.
The software architecture documentation shows the building blocks of ΦFlow and the module dependencies.
The Version history lists all major changes since release.
TensorBoard: Live supervision does not work when running a local app that writes to a remote directory.
Resampling / Advection: NumPy interpolation handles the boundaries slightly differently than TensorFlow.
Contributions are welcome! Check out this document for some guidelines.
This work is supported by the ERC Starting Grant realFlow (StG-2015-637014) and the Intel Intelligent Systems Lab.