JESTER Documentation#

Jax-based Eo**S** and Tov solvER

JESTER is a scientific computing library for solving the Tolman-Oppenheimer-Volkoff (TOV) equations for neutron star physics. It provides hardware-accelerated computations via JAX with automatic differentiation capabilities.

arXiv Paper Code Coverage

Key Features#

  • JAX-Accelerated: Hardware acceleration with GPU/TPU support

  • Automatic Differentiation: Built-in gradients for parameter studies

  • Neutron Star Physics: Complete TOV equation solver with realistic equations of state

  • Extensible: Modular design for custom equation of state models

  • Well-Tested: Comprehensive test suite with 95+ tests

Quick Start#

Install JESTER:

pip install jester-tov

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

Basic usage:

import jesterTOV as jtov

# Create equation of state
eos = jtov.eos.MetaModel_EOS_model()

# Solve TOV equations
masses, radii = eos.M_R_curve()

print(f"Maximum mass: {max(masses):.2f} solar masses")

Contents#

Indices and tables#