AxisymmetricZhaoModel

jeanspy.model_jax.AxisymmetricZhaoModel

See methods and properties for individual lookup pages, or the alphabetical API dictionary to search all classes. The full class contract and existing member anchors are retained below.

class jeanspy.model_jax.AxisymmetricZhaoModel

Bases: jeanspy.axisymmetric.AxisymmetricDMModel

Functional JAX spheroidal Zhao halo.

params requires rs_pc and rhos_Msunpc3, with optional Q, alpha, beta, gamma and r_t_pc. The physical domain and units match the NumPy component. Dynamic invalid proposals yield NaN; malformed schemas raise ValueError. Forces and mass share the same optional ellipsoidal cutoff. J/D factors use the NumPy component as posterior postprocessing.

potential_gradient(R_pc, z_pc, *, params, n_force=96)

Return (dPhi/dR,dPhi/dz) in (km/s)^2/pc; n_force is static under JIT.

mass_density_3d(R_pc, z_pc, *, params)

Return halo density in Msun/pc^3 with the broadcast coordinate shape.

enclosed_mass(m_pc, *, params, n_steps=128)

Return mass in Msun inside spheroidal radius m_pc; n_steps is static.