Ë
    ñukiÏ  ã                   ó"  — d dl mc 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mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z-m.Z.m/Z/m0Z0m1Z2m3Z4m5Z6m7Z7 dejp                  fdejr                  fdejt                  fdejv                  fdejx                  fdejz                  fd	ej|                  fd
ej~                  fdej€                  fdej‚                  fdœ
ZBd dlCZDeDjŠ                  r{ejp                  Z8ejr                  Z9ejt                  Z:ejx                  Z<ejv                  Z;ejz                  Z=ej|                  Z>ej~                  Z?ej€                  Z@ej‚                  ZA[D[yd dlFmGZH  eHeIeB«      ZJ[H[D[y)é    N)0ÚAbstractValueÚAtomÚCallPrimitiveÚ	DebugInfoÚDropVarÚEffectÚEffectsÚInconclusiveDimensionOperationÚJaxprPpContextÚJaxprPpSettingsÚJaxprTypeErrorÚ
OutputTypeÚ	ParamDictÚShapedArrayÚTraceÚTracerÚValueÚabstract_tokenÚaval_mapping_handlersÚcallÚcheck_jaxprÚconcrete_or_errorÚconcretization_function_errorÚcustom_typechecksÚensure_compile_time_evalÚeval_contextÚ
eval_jaxprÚfind_top_traceÚgensymÚget_opaque_trace_stateÚis_concreteÚis_constant_dimÚis_constant_shapeÚjaxprs_in_paramsÚliteralable_typesÚmax_dimÚmin_dimÚnew_jaxpr_eqnÚno_axis_nameÚ
no_effectsÚnonempty_axis_envÚprimal_dtype_to_tangent_dtypeÚpytype_aval_mappingsÚ	trace_ctxÚunsafe_am_i_under_a_jitÚunsafe_am_i_under_a_vmapÚunsafe_get_axis_namesÚvalid_jaxtypez!jax.core.call_impl is deprecated.z8jax.core.get_aval is deprecated; use jax.typeof instead.z#jax.core.mapped_aval is deprecated.z)jax.core.set_current_trace is deprecated.z!jax.core.subjaxprs is deprecated.z*jax.core.take_current_trace is deprecated.z-jax.core.traverse_jaxpr_params is deprecated.z%jax.core.unmapped_aval is deprecated.z%jax.core.AbstractToken is deprecated.z jax.core.TraceTag is deprecated.)
Ú	call_implÚget_avalÚmapped_avalÚset_current_traceÚ	subjaxprsÚtake_current_traceÚtraverse_jaxpr_paramsÚunmapped_avalÚAbstractTokenÚTraceTag)Údeprecation_getattr)KÚjax._src.coreÚ_srcÚcoreÚ	_src_corer   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   Únonempty_axis_env_DO_NOT_USEr,   r-   r.   r/   Ú"unsafe_am_i_under_a_jit_DO_NOT_USEr0   Ú#unsafe_am_i_under_a_vmap_DO_NOT_USEr1   Ú unsafe_get_axis_names_DO_NOT_USEr2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   Ú_deprecationsÚtypingÚ_typingÚTYPE_CHECKINGÚjax._src.deprecationsr=   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úC/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/core.pyú<module>rQ      s¸  ð÷$ "Ð !÷1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1÷ 1ó 1ðl (Ø×Ñðð
 ?Ø×Ñðð
 *Ø×Ñðð
 0Ø×Ñðð
 (Ø×Ñðð
 1Ø× Ñ ðð
 4Ø×#Ñ#ðð
 ,Ø×Ñðð
 ,Ø×Ñðð
 'Ø×ÑðñM*€óX Ø
×ÒØ×!Ñ!€)Ø×Ñ€(Ø×%Ñ%€+Ø×!Ñ!€)Ø×1Ñ1ÐØ ×3Ñ3ÐØ#×9Ñ9ÐØ×)Ñ)€-Ø×)Ñ)€-Ø×Ñ€(ð
 Ùõ	 PÙ$ X¨}Ó=€+ØØÙrO   