o 2h@sdZddlZddlZddlZddlTddlmZe#ejdee ddej de d ej j j d ed d Wdn1sBwYej j ej e <[[[dS) zIntegration with JAX.N)*)__all__ignoreF)categorymoduleappendzThe zH module is deprecated and will be removed in version 0.18.0. Please use z instead.) stacklevel)__doc__syswarningsoptree.integrations.jaxoptreercatch_warningsfilterwarnings FutureWarning__name__warn integrationsjaxmodulesrrQ/var/www/html/chatgem/venv/lib/python3.10/site-packages/optree/integration/jax.pys$