Lettuce: PyTorch-based Lattice Boltzmann Framework

Abstract

The lattice Boltzmann method (LBM) is an efficient simulation technique for computational fluid mechanics and beyond. It is based on a simple stream-and-collide algorithm on Cartesian grids, which is easily compatible with modern machine learning architectures. While it is becoming increasingly clear that deep learning can provide a decisive stimulus for classical simulation techniques, recent studies have not addressed possible connections between machine learning and LBM. Here, we introduce Lettuce, a PyTorch-based LBM code with a threefold aim. Lettuce enables GPU accelerated calculations with minimal source code, facilitates rapid prototyping of LBM models, and enables integrating LBM simulations with PyTorch's deep learning and automatic differentiation facility. As a proof of concept for combining machine learning with the LBM, a neural collision model is developed, trained on a doubly periodic shear layer and then transferred to a different flow, a decaying turbulence. We also exemplify the added benefit of PyTorch's automatic differentiation framework in flow control and optimization. To this end, the spectrum of a forced isotropic turbulence is maintained without further constraining the velocity field. The source code is freely available from https://github.com/lettucecfd/lettuce.

References in corpus (10)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Machine learning accelerated computational fluid dynamics
• Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers
• waLBerla: A block-structured high-performance framework for multiphysics simulations
• Lat-Net: Compressing Lattice Boltzmann Flow Simulations using Deep Neural Networks
• Semi-Lagrangian lattice Boltzmann method for compressible flows
• Extended Lattice Boltzmann Model for Gas Dynamics
• Deep learning of the spanwise-averaged Navier-Stokes equations
• Compressible lattice Boltzmann methods with adaptive velocity stencils: An interpolation-free formulation
• Cubature rules for weakly and fully compressible off-lattice Boltzmann methods

Cited by in corpus (3)

• Towards learning Lattice Boltzmann collision operators
• Cubature rules for weakly and fully compressible off-lattice Boltzmann methods
• Perfectly Matched Layers and Characteristic Boundaries in Lattice Boltzmann: Accuracy vs Cost