
    WhH'              	       V   d dl Zej                            dd           [d dlmZ d dlmZ d dlmZ	 	  e	             n(# e
$ r Zd dlmZ  eded	           [Y dZ[ndZ[ww xY w[	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/m0Z0m1Z1 d dl2m3Z3 d dl4m5Z5 d dl6m7Z8 e8j9        Z9[8d 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lBmCZC d dlBmDZD d dlBmEZE d dlBmFZF d dlGmHZH d dlGmIZI d dl:mJZJ d dl:mKZK d dl:mLZL d dl:mMZM d d lGmNZN d d!l:mOZO d d"l:mPZP d d#lQmRZR d d$l:mSZS d d%l:mTZT d d&lGmUZU d d'lGmVZV d d(lGmWZW d d)l:mXZX d d*l:mYZY d d+l:mZZZ d d,l:m[Z[ d d-l:m\Z\ d d.lGm]Z] d d/lGm^Z^ d d0l:m_Z_ d d1l:m`Z` d d2l:maZa d d3l:mbZb d d4l:mcZc d d5l:mdZd d d6l:meZe d d7lGmfZf d d8lGmgZg d d9lGmhZh d d:limjZj d dl=m>Zk d d;l:mlZl d d<l:mmZm d d=l:mnZn d d>l:moZo d d?lpmqZq d d@lpmrZr d dAlsmtZtmuZumvZvmwZwmxZxmyZy [t[u[v[w[x[yd dBlzm{Z{m|Z|m}Z} d dCl~mZmZmZmZmZmZmZ d dDlmZ d dElmZ d dFlmZ 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 dZlzmZ d dlmc mc mZ [d[efd\efd]efd^efd_efd`efdaefdbdcZd dlZej        r+d ddl~mZ d del~mZ d dfl~mZ d dgl~mZ d dhl~mZ d dil~mZ d djl~mZ nd dklmZ  eee          Z[[d dlZdS )l    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	spmd_mode)ensure_compile_time_eval)print_environment_info)
xla_client)effects_barrier)block_until_ready)checkpoint_wrapper)checkpoint_policies)clear_caches)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)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)adbatchingmlirpartial_evalpxlaxla)$make_array_from_single_device_arraysmake_array_from_callback"make_array_from_process_local_data)tree_maptreedef_is_leaftree_flattentree_leavestree_structuretree_transposetree_unflatten)custom_derivatives)custom_batching)custom_transpose)api_util)distributed)debug)dlpack)dtypes)errors)image)lax)
monitoring)nn)numpy)ops)profiler)random)scipy)sharding)stages)	tree_util)util)ShardzEjax.treedef_is_leaf is deprecated: use jax.tree_util.treedef_is_leaf.z|jax.tree_flatten is deprecated: use jax.tree.flatten (jax v0.4.25 or newer) or jax.tree_util.tree_flatten (any JAX version).zyjax.tree_leaves is deprecated: use jax.tree.leaves (jax v0.4.25 or newer) or jax.tree_util.tree_leaves (any JAX version).zjax.tree_structure is deprecated: use jax.tree.structure (jax v0.4.25 or newer) or jax.tree_util.tree_structure (any JAX version).zjax.tree_transpose is deprecated: use jax.tree.transpose (jax v0.4.25 or newer) or jax.tree_util.tree_transpose (any JAX version).zjax.tree_unflatten is deprecated: use jax.tree.unflatten (jax v0.4.25 or newer) or jax.tree_util.tree_unflatten (any JAX version).zpjax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).)z-jax.clear_backends was removed in JAX v0.4.36N)rc   rd   re   rf   rg   rh   rb   clear_backends)rc   )rd   )re   )rb   )rf   )rg   )rh   )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$   r%   jax._src.corer&   jax._src.environment_infor'   jax._src.libr(   _xcDevicejax._src.apir)   r*   jax._src.ad_checkpointr+   
checkpointr,   r-   jax._src.custom_derivativesr.   r/   r0   r1   jax._src.xla_bridger2   r3   r4   r5   r6   r7   r8   r9   r:   jax._src.dtypesr;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   jax._src.callbackrR   rematrS   rT   rU   rV   jax._src.sharding_implsrW   rX   jax.interpretersrY   rZ   r[   r\   r]   r^   jax._src.arrayr_   r`   ra   jax._src.tree_utilrb   _deprecated_tree_maprc   _deprecated_treedef_is_leafrd   _deprecated_tree_flattenre   _deprecated_tree_leavesrf   _deprecated_tree_structurerg   _deprecated_tree_transposerh   _deprecated_tree_unflattenri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   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     R/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/jax/__init__.py<module>r      s	          -s 3 3 3 3 2 2 2 2 2 < < < < < < F E E E E E/   $$$$$$% 	<# 	< 	< 	< = = =eeeee       	
 . - - - - -                                                                         8 O N N N N N V V V V V V * * * * * *	 ; ; ; ; ; ; ? ? ? ? ? ? C C C C C C M M M M M M 5 5 5 5 5 5 J J J J J J J J J J J J @ @ @ @ @ @ @ @ @ @ @ @ B B B B B B < < < < < < 1 1 1 1 1 1 1 1 1 1 1 1 A A A A A A G G G G G G 2 2 2 2 2 2 3 3 3 3 3 3 1 1 1 1 1 1 , , , , , , % % % % % % + + + + + + 8 8 8 8 8 8 2 2 2 2 2 2 4 4 4 4 4 4 - - - - - - ) ) ) ) ) ) ) ) ) ) ) ) # # # # # # # # # # # # H H H H H H > > > > > > / / / / / / = = = = = = 3 3 3 3 3 3 1 1 1 1 1 1 1 1 1 1 1 1 3 3 3 3 3 3 % % % % % % > > > > > > > > > > > > B B B B B B < < < < < < > > > > > > = = = = = = 9 9 9 9 9 9 # # # # # # % % % % % % B B B B B B : : : : : : I H H H H H H H H H H H H H H H$dC                           9 8 8 8 8 8 2 2 2 2 2 2 4 4 4 4 4 4 $ $ $ $ $ $ * * * * * *                                                       ( ( ( ( ( (                   $ $ $ $ $ $                   $ $ $ $ $ $             & & & & & &       * ) ) ) ) ) F F F F F F F F F F F F
 L
769993M* *X    
 CCCCCC======;;;;;;555555AAAAAAAAAAAAAAAAAAA POOOOO$$X}==+ s   
? A$AA$