torrewk / Python-Jax-Fast-Stokesian-Dynamics

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.
3 stars 0 forks source link