JESTER

JAX-accelerated nuclear equation of state code and TOV solver -- with support for automatic differentiation!

Installation

pip install jesterTOV

Or install from source:

pip install git+https://github.com/nuclear-multimessenger-astronomy/jester

For GPU support:

pip install "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

Documentation

📚 Read the full documentation →

Example notebooks

• examples/eos_tov.ipynb: Basic EOS and TOV solving
• examples/automatic_differentiation.ipynb: Gradient-based optimization

Acknowledgements

If you use jester in your work, please cite our paper!

@article{Wouters:2025zju,
    author = "Wouters, Thibeau and Pang, Peter T. H. and Koehn, Hauke and Rose, Henrik and Somasundaram, Rahul and Tews, Ingo and Dietrich, Tim and Van Den Broeck, Chris",
    title = "{Leveraging differentiable programming in the inverse problem of neutron stars}",
    eprint = "2504.15893",
    archivePrefix = "arXiv",
    primaryClass = "astro-ph.HE",
    reportNumber = "LA-UR-25-23486",
    month = "4",
    year = "2025"
}