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
jnibauer/galdynamix
FardalDF Particle spray implemented in Jax. For best performance, use a GPU!
Jupyter Notebook