
    ukiA                     X    d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
mZmZ ddddedefdZy)	    )Callable)
StoreException
WrappedFuncachemerge_linear_auxtransformationtransformation_with_auxtransformation2transformation_with_aux2	wrap_init_missing_debug_infoN
debug_infofreturnc                <    |xs t        d      }t        | ||      S )Nzlinear_util.wrap_initr   )r   
_wrap_init)r   paramsr   s      Q/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/extend/linear_util.pyr   r   %   s"    I01HI*	Av*	55    )N)collections.abcr   jax._src.linear_utilr   r   r   r   r   r	   r
   r   r   r   r    r   r   <module>r      s3   $ %  "6d 6 6z 6r   