
    Vpfl                        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+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z| d dl}m~Z i ddej        fddej        fddej        fd	d
ej        fddej        fddej        fddej        fddej        fddej        fddej        fddej        fddej        fdddddd d!d"d#d$d%d&e`fd'Zd d(lZej        rWe`Z_ej        Zej        Zej        Zej        Zej        Zej        Zej        Zej        Zej        Zej        Zej        Zej        Znd d)lmZ  eee          Z[[[d(S )*    ){AbstractTokenAbstractValueAtomAxisSizeCallPrimitiveClosedJaxprConcreteArrayConcretizationTypeErrorDShapedArrayDropVarEffectEffects	EvalTraceInDBIdxInconclusiveDimensionOperation	InputTypeJaxprJaxprDebugInfoJaxprEqnJaxprPpContextJaxprPpSettingsJaxprTypeErrorLiteral	MainTraceMapPrimitiveNameGatheringSubstOutDBIdx
OutputType	ParamDict	PrimitiveShapedArraySublevelTRACER_LEAK_DEBUGGER_WARNINGThreadLocalStateTokenTrace
TraceStack
TraceStateTracerUnshapedArrayValueVarabstract_tokenapply_todosaval_mapping_handlers
axis_framecallcall_bind_with_continuation	call_implcall_p	check_eqncheck_jaxpr
check_typecheck_valid_jaxtypeclosed_call_pconcrete_avalconcrete_or_errorconcretization_function_errorcur_sublevelcustom_typechecksdedup_referentsdo_subst_axis_names_jaxprensure_compile_time_evalescaped_tracer_erroreval_context
eval_jaxprextend_axis_envextend_axis_env_ndfind_top_trace
full_lowergensymget_avalget_referentis_constant_dimis_constant_shapejaxpr_as_funjaxpr_uses_outfeedjaxprs_in_paramsjoin_effectslattice_joinleaked_tracer_errorliteralable_typesmap_bindmap_bind_with_continuationmapped_avalmaybe_find_leaked_tracersmax_dimmin_dimnew_base_mainnew_jaxpr_eqnnew_mainnew_sublevelno_axis_name
no_effectsnon_negative_dimoutfeed_primitivesprimal_dtype_to_tangent_dtypeprimitive_uses_outfeedprocess_env_traces_callprocess_env_traces_mappytype_aval_mappingsraise_as_much_as_possibleraise_to_shapedraise_to_shaped_mappingsreset_trace_statestash_axis_envstr_eqn_compact	subjaxprssubst_axis_namessubst_axis_names_eqnsubst_axis_names_jaxprsubst_axis_names_varsubstitute_vars_in_output_tythread_local_statetrace_state_cleantraverse_jaxpr_params	typecheck
typecompat	typematchunmapped_avalused_axis_namesused_axis_names_jaxprvalid_jaxtype)corepp_avalzjax.core.pp_aval is deprecated.pp_eqnzjax.core.pp_eqn is deprecated.pp_eqn_rulesz$jax.core.pp_eqn_rules is deprecated.pp_eqnszjax.core.pp_eqns is deprecated.pp_jaxprz jax.core.pp_jaxpr is deprecated.pp_jaxpr_eqn_rangez*jax.core.pp_jaxpr_eqn_range is deprecated.pp_jaxpr_skeletonz)jax.core.pp_jaxpr_skeleton is deprecated.	pp_jaxprsz!jax.core.pp_jaxprs is deprecated.
pp_kv_pairz"jax.core.pp_kv_pair is deprecated.pp_kv_pairsz#jax.core.pp_kv_pairs is deprecated.pp_varzjax.core.pp_var is deprecated.pp_varszjax.core.pp_vars is deprecated.DimSize)z8jax.core.DimSize is deprecated. Use DimSize = int | Any.NShape)z>jax.core.Shape is deprecated. Use Shape = Sequence[int | Any].Ncanonicalize_shape)z*jax.core.canonicalize_shape is deprecated.Ndimension_as_value)z9jax.core.dimension_as_value is deprecated. Use jnp.array.Ndefinitely_equal)z0jax.core.definitely_equal is deprecated. Use ==.N)z2jax.core.symbolic_equal_dim is deprecated. Use ==.Nz=jax.core.non_negative_dim is deprecated. Use max_dim(..., 0).)symbolic_equal_dimra   N)deprecation_getattr)jax._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+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   _deprecated_non_negative_dimrb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   jax._srcr~   	_src_corer   r   r   r   r   r   r   r   r   r   r   r   _deprecationstypingTYPE_CHECKINGjax._src.deprecationsr   _deprecation_getattr__name____getattr__     H/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/core.py<module>r      s  $| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | |~ ' & & & & &(193DE( /1AB( ;Y=ST	(
 193DE( 3Y5GH( GIef( EyGbc( 5y7JK( 79MN( 99;PQ( /1AB( 193DE(  (&  '(0  1(6  7(<  =(B
 FGcK( ( (T 	 1'&','( 31!)#*%+&''OOOOOO$$X}==+
IIr   