
    ukil                          e        Zd  Zd Zy)c                 0    t         j                  |        | S )N)_custom_wrapper_typesadd)clss    S/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/custom_api_util.pyregister_custom_decorator_typer      s    C 	*    c                     |j                  d      r1t        | j                        t        v rt	        | j                  |      S t
        )Ndef)
startswithtypefunr   getattrAttributeError)self_names     r   forward_attrr      s5    	__UUYY3H H599d##
r   N)setr   r   r    r   r   <module>r      s      r   