The source code for our latent space physics paper is online now:
https://github.com/wiewel/LatentSpacePhysics

It contains both the Navier-Stokes solver for data generation (based on mantaflow http://mantaflow.com), and the keras code (for tensorflow https://www.tensorflow.org) for training the autoencoder and LSTM networks.

The preprint can be found here: https://arxiv.org/pdf/1802.10123

Paper Abstract

Our work explores methods for the data-driven inference of temporal evolutions of physical functions with deep learning techniques. More specifically, we target fluid flow problems, and we propose a novel network architecture to predict the changes of the pressure field over time. The central challenge in this context is the high dimensionality of Eulerian space-time data sets. Key for arriving at a feasible algorithm is a technique for dimensionality reduction based on convolutional neural networks, as well as a special architecture for temporal prediction. We demonstrate that dense 3D+time functions of physics system can be predicted with neural networks, and we arrive at a neural-network based simulation algorithm with practical speed-ups. We demonstrate the capabilities of our method with a series of complex liquid simulations, and with a set of single-phase simulations. Our method predicts pressure fields very efficiently. It is more than two orders of magnitudes faster than a regular solver. Additionally, we present and discuss a series of detailed evaluations for the different components of our algorithm.