GitHub repository: torrewk/Python-Jax-Fast-Stokesian-Dynamics
Primary language: Python
Description: A python implementation of the Fast Stokesian Dynamics methods (A. Fiore, J. Swan, 2019) which leverage the library JAX to just-in-time compile and run on GPU the simulation.