JAX

See Units, shapes and model contracts for units, shapes, domains and backend boundaries, and Quickstart for executable minimal examples.

AxisymmetricDSphModel([n_force, n_vertical, ...])	Compose functional JAX axisymmetric stellar, halo and anisotropy components.
Model([submodels])	Functional JAX model container with explicit physical-parameter mappings.
PlummerModel([submodels])	Plummer surface/volume density normalized so that int 2πR Σ(R) dR = 1.
ZhaoModel([submodels])	Functional JAX Zhao halo with transition, outer and inner slopes.
NFWModel([submodels])	Functional JAX NFW density and mass with a hard cutoff at r_t_pc.
ConstantAnisotropyModel([submodels])	JAX constant spherical anisotropy, beta(r) = params["beta_ani"].
BaesAnisotropyModel([submodels])	JAX Baes--van Hese anisotropy with a numerical LOS kernel.
OsipkovMerrittModel([submodels])	JAX Osipkov--Merritt anisotropy with an analytic LOS kernel.
DSphModel([submodels])	Spherical JAX Jeans model for LOS variance on fixed quadrature grids.
configure_runtime(*[, jax_enable_x64])	Update global runtime knobs used by model_jax.
get_runtime_config()	Return the current model_jax runtime configuration.
AxisymmetricPlummerModel()	Functional JAX Plummer tracer with explicit re_pc and q/q_projected.
AxisymmetricZhaoModel()	Functional JAX spheroidal Zhao halo.
AxisymmetricConstantAnisotropyModel()	Functional constant cylindrical beta_z (default 0), finite and <1.

Aliases

• jeanspy.model_jax.AxisymmetricStellarModel is jeanspy.model.AxisymmetricStellarModel.

• jeanspy.model_jax.AxisymmetricDMModel is jeanspy.model.AxisymmetricDMModel.

• jeanspy.model_jax.AxisymmetricAnisotropyModel is jeanspy.model.AxisymmetricAnisotropyModel.