Lettuce: PyTorch-based Lattice Boltzmann Framework

2 code implementations • 24 Jun 2021 • Mario Christopher Bedrunka, Dominik Wilde, Martin Kliemank, Dirk Reith, Holger Foysi, Andreas Krämer

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.

BIG-bench Machine Learning