Equivariant Neural Prediction of the Stokes Resistance Tensors for Arbitrary Microparticle Shapes
CFD
An SO(3)-equivariant neural network predicts the full 6×6 grand resistance matrix for microparticles in Stokes flow, outperforming traditional scalar drag correlations by preserving orientation dependence and rotational response.
See LICENSE file Stokes flowresistance tensorequivariant neural network