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

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
FurtherTuned_jaxvgala_mockstream_Samemasses.pdf		FurtherTuned_jaxvgala_mockstream_Samemasses.pdf
JaxStreams.py		JaxStreams.py
JaxStreamsDemo.ipynb		JaxStreamsDemo.ipynb
JaxStreamsDemo_diffrax.ipynb		JaxStreamsDemo_diffrax.ipynb
JaxStreams_diffrax.py		JaxStreams_diffrax.py
JaxStreams_diffrax_refactored.py		JaxStreams_diffrax_refactored.py
Pal5Diffrax.ipynb		Pal5Diffrax.ipynb
README.md		README.md
TaylorExpansion_PotentialFinalSnap.pdf		TaylorExpansion_PotentialFinalSnap.pdf
TaylorExpansion_PotentialFinalSnap_MASS.pdf		TaylorExpansion_PotentialFinalSnap_MASS.pdf
Untitled.ipynb		Untitled.ipynb
Untitled1.ipynb		Untitled1.ipynb
Untitled2.ipynb		Untitled2.ipynb
custom_integrator.ipynb		custom_integrator.ipynb
jaxvgala_mockstream_Samemasses.pdf		jaxvgala_mockstream_Samemasses.pdf
jaxvgala_mockstream_differingmasses.pdf		jaxvgala_mockstream_differingmasses.pdf
sample_stream.pdf		sample_stream.pdf
stream_integrator_static.ipynb		stream_integrator_static.ipynb
stream_integrator_static_time_dep.ipynb		stream_integrator_static_time_dep.ipynb
stream_integrator_static_time_dep_first_deriv_tests.ipynb		stream_integrator_static_time_dep_first_deriv_tests.ipynb
subhalo_constraint.pdf		subhalo_constraint.pdf
try_train.ipynb		try_train.ipynb

jnibauer/galdynamix