
    vh0                     F   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_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZmZmZmZmZ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 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$ d dl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. d dl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZG d dlHmIZImJZJmKZK d dlLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZc d d	ldmeZe d d
lfmgZgmhZh d dlimjZjmkZkmlZl d dlmmnZn d dlompZp d dlqmrZr d dlqmsZs d dltmuZu d dl Zvdevj                  j                  j                  j                  fdevj                  j                  j                  j                  fdevj                  j                  j                  j                  fdevj                  j                  j                  j                  fdZyd dlzZ{e{j                  revj                  j                  j                  j                  Zbevj                  j                  j                  j                  Zcevj                  j                  j                  j                  Zevj                  j                  j                  j                  Z[{yd dl}m~Z  eeey      Z[[{y)    )DotDimensionNumbersRaggedDotDimensionNumbers	PrecisionPrecisionLikeDotAlgorithmDotAlgorithmPresetRandomAlgorithmRoundingMethodabsabs_pacosacos_pacoshacosh_paddadd_p	after_allafter_all_pand_pargmaxargmax_pargminargmin_pasinasin_pasinhasinh_patanatan_patan2atan2_patanhatanh_pbatch_matmulbitcast_convert_typebitcast_convert_type_pbitwise_andbitwise_not
bitwise_orbitwise_xor	broadcastbroadcast_in_dimbroadcast_in_dim_pbroadcast_shapesbroadcast_to_rankbroadcasted_iotacbrtcbrt_pceilceil_pclampclamp_pclzclz_pcollapsecomplex	complex_p	compositeconcatenateconcatenate_pconjconj_pconvert_element_typeconvert_element_type_pcopy_pcoscos_pcoshcosh_pcreate_tokencreate_token_pdivdiv_pdotdot_generaldot_general_pdtypeeqeq_peq_to_pexpexp_pexp2exp2_pexpand_dimsexpm1expm1_pfloorfloor_pfull	full_likegege_pgtgt_pimagimag_pinfeedinfeed_pinteger_powinteger_pow_piotaiota_p	is_finiteis_finite_plele_ple_to_ploglog1plog1p_plog_plogistic
logistic_pltlt_plt_to_pmaxmax_pminmin_pmulmul_pnene_pnegneg_p	nextafternextafter_pnot_poptimization_barrieroptimization_barrier_por_poutfeed	outfeed_ppadpad_ppadtype_to_padspopulation_countpopulation_count_ppowpow_p
ragged_dotragged_dot_generalrealreal_p
reciprocalreduce
reduce_andreduce_and_p
reduce_maxreduce_max_p
reduce_minreduce_min_p	reduce_orreduce_or_preduce_preduce_precisionreduce_precision_preduce_prodreduce_prod_p
reduce_sumreduce_sum_p
reduce_xorreduce_xor_premrem_preshape	reshape_prevrev_prng_bit_generatorrng_bit_generator_prng_uniformrng_uniform_proundround_prsqrtrsqrt_pselectselect_n
select_n_pshape_as_value
shift_leftshift_left_pshift_right_arithmeticshift_right_arithmetic_pshift_right_logicalshift_right_logical_psignsign_psinsin_psinhsinh_psortsort_key_valsort_psplitsplit_psqrtsqrt_psquaresquare_psqueeze	squeeze_pstop_gradientsubsub_ptantan_ptanhtanh_ptop_ktop_k_p	transposetranspose_pxor_pzeros_like_array)
bessel_i0ebessel_i0e_p
bessel_i1ebessel_i1e_pbetaincdigamma	digamma_perferfcerfc_perf_inv	erf_inv_perf_pigammaigammac	igammac_pigamma_grad_aigamma_grad_a_pigamma_plgammalgamma_p	polygammapolygamma_prandom_gamma_gradregularized_incomplete_beta_pzetazeta_p)GatherDimensionNumbersGatherScatterModeScatterDimensionNumbersdynamic_index_in_dimdynamic_slicedynamic_slice_in_dimdynamic_slice_pdynamic_update_index_in_dimdynamic_update_slicedynamic_update_slice_in_dimdynamic_update_slice_pgathergather_pindex_in_dim
index_takescatterscatter_applyscatter_addscatter_add_pscatter_maxscatter_max_pscatter_minscatter_min_pscatter_mulscatter_mul_p	scatter_pscatter_subscatter_sub_psliceslice_in_dimslice_p)ConvDimensionNumbers"ConvGeneralDilatedDimensionNumbersconvconv_dimension_numbersconv_general_dilatedconv_general_dilated_pconv_general_permutationsconv_general_shape_tupleconv_shape_tupleconv_transposeconv_transpose_shape_tupleconv_with_general_padding)	reduce_windowreduce_window_max_preduce_window_min_preduce_window_preduce_window_shape_tuplereduce_window_sum_pselect_and_gather_add_pselect_and_scatter_pselect_and_scatter_add_p)associative_scancondcond_pcumlogsumexpcumlogsumexp_pcummaxcummax_pcummincummin_pcumprod	cumprod_pcumsumcumsum_pcustom_linear_solvecustom_root	fori_looplinear_solve_pmapscanscan_pswitch
while_loopwhile_pplatform_dependent)fftfft_pFftType)
all_gatherall_gather_invariantall_gather_p
all_to_allall_to_all_p
axis_indexaxis_index_p	axis_size
pbroadcastpmaxpmax_ppmeanpminpmin_pppermute
ppermute_ppshufflepsumpsum_ppsum_scatter	pswapaxesragged_all_to_allragged_all_to_all_p)pvary)conv_general_dilated_localconv_general_dilated_patches)approx_max_kapprox_min_kapprox_top_k_p)stop_gradient_p)linalg)with_sharding_constraint)sharding_constraint_p)device_put_pNzNjax.lax.infeed was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0.zPjax.lax.infeed_p was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0.zOjax.lax.outfeed was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0.zQjax.lax.outfeed_p was deprecated in JAX v0.6.0 and will be removed in JAX v0.7.0.)rd   re   r   r   )deprecation_getattr(  jax._src.lax.laxr   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   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   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   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   r   r   r   r   r   jax._src.lax.specialr   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   jax._src.lax.slicingr   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  jax._src.lax.convolutionr  r  r  r  r  r  r  r  r   r!  r"  r#   jax._src.lax.windowed_reductionsr$  r%  r&  r'  r(  r)  r*  r+  r,  jax._src.lax.control_flowr-  r.  r/  r0  r1  r2  r3  r4  r5  r6  r7  r8  r9  r:  r;  r<  r=  r>  r?  r@  rA  rB  rC  rD  jax._src.lax.fftrE  rF  rG  jax._src.lax.parallelrH  rI  rJ  rK  rL  rM  rN  rO  rP  rQ  rR  rS  rT  rU  rV  rW  rX  rY  rZ  r[  r\  r]  r^  jax._src.corer_  jax._src.lax.otherr`  ra  jax._src.lax.annrb  rc  rd  jax._src.ad_utilre  jax.laxrf  jax._src.pjitrg  rh  jax._src.dispatchri  jax_srclax_deprecationstyping_typingTYPE_CHECKINGjax._src.deprecationsrj  _deprecation_getattr__name____getattr__     K/opt/face_recognition/venv/lib/python3.12/site-packages/jax/lax/__init__.py<module>r     s  $\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \z       :                             B      
 
 
 
 
 
            4  
            2    
 @ ? $ $ N N H H : : 
 	 	!! 	   	""-> 
88<<""&XX\\&&(HHLL$$'hhll(() 	 PO$X}=+r  