jnibauer / galdynamix

FardalDF Particle spray implemented in Jax. For best performance, use a GPU!

JaxStreams

Implementation of Fardal particle spray in jax, using diffrax. Inspired by https://ui.adsabs.harvard.edu/abs/2023ascl.soft06008A/abstract Demo notebook in JaxStreamsDemo_diffrax.ipynb