3Ki{ SSKrSSKr\RS\4Sj5r\RS\4Sj5r\RS S\\\\4S\\\\44Sjj5r\RS\4Sj5r \RS\4Sj5r \RS\4S j5r \RS\4S j5r \RS\4S j5r \RS\4S j5rg)Nreturnc.SSKng![a gf=f)zCheck if JAX is installed.rNTF)jax ImportError)rs U/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/torch/utils/_pallas.pyhas_jax_packagers s  cR[5(dgSSKJn g![a gf=f)z0Check if Pallas (JAX experimental) is available.Fr)pallasT)rjax.experimentalr r)pls rhas_pallas_packager s1     s  &&fallbackcSSKnURRS5nSUSS5up4nX4U4$![[[ 4a Us$f=f)z/Get JAX version as (major, minor, patch) tuple.rN.c38# UHn[U5v M g7f)N)int).0vs r "get_jax_version..'sA/@!s1vv/@s)r __version__splitr ValueErrorAttributeError)rr version_partsmajorminorpatchs rget_jax_versionr sZ--c2 A}Ra/@Aee$$ ^ 4s14A Ac[5(dgSSKnURS5n[U5S:$![a gf=f)z&Check if JAX has CUDA backend support.FrNgpurrdeviceslen Exceptionrr$s rhas_jax_cuda_backendr(-I   ++e$7|a "5 AAc[5(dgSSKnURS5n[U5S:$![a gf=f)z%Check if JAX has TPU backend support.FrNtpur#r's rhas_jax_tpu_backendr-<r)r*c[5$)z,Checks for a full Pallas-on-CPU environment.)r rhas_cpu_pallasr1Ks   r0c[5=(a/ [RR5=(a [ 5$)z-Checks for a full Pallas-on-CUDA environment.)r torchcuda is_availabler(r/r0rhas_cuda_pallasr6Qs(   XEJJ$;$;$= XBVBXXr0c8[5=(a [5$)z,Checks for a full Pallas-on-TPU environment.)r r-r/r0rhas_tpu_pallasr8Ws   9$7$99r0cZ[5=(d [5=(d [5$)z Check if Pallas backend is fully available for use. Requirements: - JAX package installed - Pallas (jax.experimental.pallas) available - A compatible backend (CUDA or TPU) is available in both PyTorch and JAX. )r1r6r8r/r0r has_pallasr:]s   D0 DN4DDr0))rrr) functoolsr3cacheboolrr tuplerr r(r-r1r6r8r:r/r0rr?sF    D    eCcM2 5cSVCW    d    T       YYY  :::   ED E Er0