We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Physics > Computational Physics

Title: 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 this https URL
Subjects: Computational Physics (physics.comp-ph); Machine Learning (cs.LG); Fluid Dynamics (physics.flu-dyn)
Journal reference: In International Conference on High Performance Computing (pp. 40-55). Springer, Cham (2021)
DOI: 10.1007/978-3-030-90539-2_3
Cite as: arXiv:2106.12929 [physics.comp-ph]
  (or arXiv:2106.12929v2 [physics.comp-ph] for this version)

Submission history

From: Mario Christopher Bedrunka [view email]
[v1] Thu, 24 Jun 2021 11:44:21 GMT (7273kb,D)
[v2] Wed, 17 Nov 2021 09:48:18 GMT (7273kb,D)

Link back to: arXiv, form interface, contact.