Jax-based tOv SolvEr (JOSE) consists of a set of tools for solving the TOV equation with a given equation-of-state (EOS).
Since it is based on JAX, JOSE makes use of hardware acceleration on GPUs and TPUs. It can also be easily used together with other JAX-based software, e.g., Jim.
You may install the bleeding edge version by cloning this repo, or doing
pip install git+https://github.com/tsunhopang/jose