
    vhK$                        d dl Zej                  j                  dd       [d dlmZ d dlmZ d dlmZ	 	  e	        [	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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/m0Z0 d dl1m2Z2 d dl3m4Z4 d dl5m6Z7 e7jp                  Z8[7d dl1m9Z9 d dl:m;Z; d dl:m<Z< d dl=m>Z? d dl=m@Z@ d dl:mAZA d dl:mBZB d dlCmDZD d dlCmEZE d dlCmFZF d dlCmGZG d dlHmIZI d dlHmJZJ d dl:mKZK d dl:mLZL d d l:mMZM d d!l:mNZN d d"lHmOZO d d#l:mPZP d d$l:mQZQ d d%lRmSZS d d&l:mTZT d d'l:mUZU d d(l:mVZV d d)lHmWZW d d*lHmXZX d d+lHmYZY d d,l:mZZZ d d-l:m[Z[ d d.l:m\Z\ d d/l:m]Z] d d0l:m^Z^ d d1lHm_Z_ d d2lHm`Z` d d3l:maZa d d4l:mbZb d d5l:mcZc d d6l:mdZd d d7l:meZe d d8l:mfZf d d9l:mgZg d d:lHmhZh d d;lHmiZi d d<lHmjZj d d=lkmlZl d dl=m>Zm d d>l:mnZn d d?l:moZo d d@l:mpZp d dAl:mqZq d dBlrmsZs d dClrmtZt d dDlumvZv d dElwmxZxmyZymzZzm{Z{m|Z|m}Z} [x[y[z[{[|[}d dFl~mZmZmZ d dGlmZ d dHlmZ d dIlmZ d dJlmZ d dKlmZ d dLlmZ d dMlmZ d dNlmZ d dOlmZ d dPlmZ d dQlmZ d dRlmZ d dSlmZ d dTlmZ d dUlmZ d dVlmZ d dWlmZ d dXlmZ d dYlmZ d dZlmZ d d[lmZ d d\lmZ d d]lmZ d d^l~mZ d dlmc mc mZ [d_d`dadbdcdddedfZd dlZej@                  rnd dglmZ  eee      Z[[d dlZy# e
$ rZd dlmZ  eded	       [Y dZ[dZ[ww xY w)h    NTF_CPP_MIN_LOG_LEVEL1)__version__)__version_info__)cloud_tpu_init)warnzcloud_tpu_init failed: zP
 This a JAX bug; please report an issue at https://github.com/jax-ml/jax/issues)Array)tree)typing)configenable_checksdebug_key_reusecheck_tracer_leakschecking_leaksenable_custom_prngsoftmax_custom_jvp%enable_custom_vjp_by_custom_transpose
debug_nans
debug_infslog_compiles
no_tracingexplain_cache_missesdefault_devicedefault_matmul_precisiondefault_prng_implnumpy_dtype_promotionnumpy_rank_promotion"jax2tf_associative_scan_reductionslegacy_prng_keythreefry_partitionabletransfer_guardtransfer_guard_host_to_devicetransfer_guard_device_to_devicetransfer_guard_device_to_host)ensure_compile_time_eval)print_environment_info)
xla_client)typeof)effects_barrier)block_until_ready)checkpoint_wrapper)checkpoint_policies)clear_caches)copy_to_host_async)closure_convert)custom_gradient)
custom_jvp)
custom_vjp)default_backend)device_count)
device_get)
device_put)device_put_sharded)device_put_replicated)devices)disable_jit)
eval_shape)float0)fwd_and_bwd)grad)hessian)
host_count)host_id)host_ids)jacobian)jacfwd)jacrev)jit)jvp)local_device_count)local_devices)	linearize)linear_transpose)live_arrays)
make_jaxpr)
named_call)named_scope)pmap)process_count)process_index)process_indices)pure_callback)ShapeDtypeStruct)value_and_grad)vjp)vmap)NamedSharding)	make_mesh)	shard_map)adbatchingmlirpartial_evalpxlaxla)$make_array_from_single_device_arraysmake_array_from_callback"make_array_from_process_local_data)custom_derivatives)custom_batching)custom_transpose)api_util)distributed)debug)dlpack)dtypes)errors)ffi)image)lax)
monitoring)nn)numpy)ops)profiler)random)scipy)sharding)stages)	tree_util)util)Shard)zQjax.treedef_is_leaf was removed in JAX v0.6.0: use jax.tree_util.treedef_is_leaf.N)zjax.tree_flatten was removed in JAX v0.6.0: use jax.tree.flatten (jax v0.4.25 or newer) or jax.tree_util.tree_flatten (any JAX version).N)zjax.tree_leaves was removed in JAX v0.6.0: use jax.tree.leaves (jax v0.4.25 or newer) or jax.tree_util.tree_leaves (any JAX version).N)zjax.tree_structure was removed in JAX v0.6.0: use jax.tree.structure (jax v0.4.25 or newer) or jax.tree_util.tree_structure (any JAX version).N)zjax.tree_transpose was removed in JAX v0.6.0: use jax.tree.transpose (jax v0.4.25 or newer) or jax.tree_util.tree_transpose (any JAX version).N)zjax.tree_unflatten was removed in JAX v0.6.0: use jax.tree.unflatten (jax v0.4.25 or newer) or jax.tree_util.tree_unflatten (any JAX version).N)z|jax.tree_map was removed in JAX v0.6.0: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).N)treedef_is_leaftree_flattentree_leavestree_structuretree_transposetree_unflattentree_map)deprecation_getattr)os_osenviron
setdefaultjax.versionr   r   jax._src.cloud_tpu_initr   _cloud_tpu_init	Exceptionexcwarningsr   _warnjax.corecore_corejax._src.basearrayr	   jaxr
   r   jax._src.configr   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.corer%   jax._src.environment_infor&   jax._src.libr'   _xcDevicer(   jax._src.apir)   r*   jax._src.ad_checkpointr+   
checkpointr,   r-   r.   jax._src.custom_derivativesr/   r0   r1   r2   jax._src.xla_bridger3   r4   r5   r6   r7   r8   r9   r:   r;   jax._src.dtypesr<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   jax._src.callbackrT   rematrU   rV   rW   rX   jax._src.sharding_implsrY   rZ   jax._src.shard_mapr[   jax.interpretersr\   r]   r^   r_   r`   ra   jax._src.arrayrb   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|   4jax.experimental.compilation_cache.compilation_cacheexperimentalcompilation_cache_ccache_deprecations_typingTYPE_CHECKINGjax._src.deprecationsr   _deprecation_getattr__name____getattr__jax.lib     G/opt/face_recognition/venv/lib/python3.12/site-packages/jax/__init__.py<module>r      s(       -s 3 3 < F  	
 .         6 O V *	 * ; ? C M 5 A J J @ @ B < 1 1 A G 2 3 1 , 3 % + 8 2 4 - ) ) # # H > / = 3 1 1 3 % > > B < > = 9 # % B : 5 I H$dC  9 2 4 $ *           (    $    $   &  * F F




?$L 
O$X}=+ K  $
!# ); 	< =s   K K;K66K;