JAX
===

.. py:module:: jeanspy.model_jax

.. currentmodule:: jeanspy.model_jax

See :doc:`../guides/contracts` for units, shapes, domains and backend
boundaries, and :doc:`../quickstart` for executable minimal examples.

.. autosummary::

   AxisymmetricDSphModel
   Model
   PlummerModel
   ZhaoModel
   NFWModel
   ConstantAnisotropyModel
   BaesAnisotropyModel
   OsipkovMerrittModel
   DSphModel
   configure_runtime
   get_runtime_config
   AxisymmetricPlummerModel
   AxisymmetricZhaoModel
   AxisymmetricConstantAnisotropyModel

Aliases
-------

* ``jeanspy.model_jax.AxisymmetricStellarModel`` is :py:obj:`jeanspy.model.AxisymmetricStellarModel`.
* ``jeanspy.model_jax.AxisymmetricDMModel`` is :py:obj:`jeanspy.model.AxisymmetricDMModel`.
* ``jeanspy.model_jax.AxisymmetricAnisotropyModel`` is :py:obj:`jeanspy.model.AxisymmetricAnisotropyModel`.