Ë
    ñuki  ã                   ót   — d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ dd	lmZ dd
lmZ y)zPublic API for fuser.é    )Úget_fusion_values)Úmake_scalar_prefetch_handler)Úpull_block_spec)Úpush_block_spec)Úcustom_fusion)Úevaluate)Úfusible)ÚFusion)ÚfuseN)Ú__doc__Ú jax._src.pallas.fuser.block_specr   r   r   r   Ú'jax._src.pallas.fuser.custom_fusion_libr   Ú%jax._src.pallas.fuser.custom_evaluater   Újax._src.pallas.fuser.fusibler	   Újax._src.pallas.fuser.fusionr
   Ú"jax._src.pallas.fuser.jaxpr_fusionr   © ó    úX/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/experimental/pallas/fuser.pyú<module>r      s%   ðñ å SÝ iÝ OÝ OÝ RÝ FÝ <Ý 9Þ ;r   