Ë
    ñukim  ã                   ó,   — d Z ddlmZmZmZmZmZmZmZ y)aN  Modules for JAX extensions.

The :mod:`jax.extend` module provides modules for access to JAX
internal machinery. See
`JEP #15856 <https://docs.jax.dev/en/latest/jep/15856-jex.html>`_.

This module is not the only means by which JAX aims to be
extensible. For example, the main JAX API offers mechanisms for
`customizing derivatives
<https://docs.jax.dev/en/latest/notebooks/Custom_derivative_rules_for_Python_code.html>`_,
`registering custom pytree definitions
<https://docs.jax.dev/en/latest/custom_pytrees.html#pytrees-custom-pytree-nodes>`_,
and more.

API policy
----------

Unlike the
`public API <https://docs.jax.dev/en/latest/api_compatibility.html>`_,
this module offers **no compatibility guarantee** across releases.
Breaking changes will be announced via the
`JAX project changelog <https://docs.jax.dev/en/latest/changelog.html>`_.
é    )ÚbackendÚcoreÚlinear_utilÚmlirÚrandomÚshardingÚsource_info_utilN)	Ú__doc__Ú
jax.extendr   r   r   r   r   r   r	   © ó    úN/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/extend/__init__.pyú<module>r      s   ðñ÷0÷ ò r   