jax/__init__.py,sha256=dQlccp1SwjK0S0m-3JyzU4WDe5FSwJaW8GlRS2AKHJU,10107
jax/ad_checkpoint.py,sha256=N3-hWts5BdssIjnoNm1NnGzACdpQmtjol7fd_priwKo,802
jax/api_util.py,sha256=JnZFSuqh5WBglJwerrWxcIDslBaXYZQrCv6j0jPfu7M,769
jax/cloud_tpu_init.py,sha256=nNdPdZZffdRGb38rEx1YOLXNpymSNt05ZVWSm8lhZ8c,633
jax/collect_profile.py,sha256=exKxJOhMrkMnXcseOR2VwV4m3ZNw00DsMmXAR-9Qr1w,4721
jax/config.py,sha256=xx0PZUvoNtaxkxXQImoymJlY6I7J1oSbnkgCigp1nMg,1040
jax/core.py,sha256=POA3yaxLJnbb_Oj699lSD6epH0S20VO3tSXFuqQZkf4,7340
jax/custom_batching.py,sha256=zQmkJIxOxxgAqEgby1q-vQWHYKzyqp1__n8Hqs04OV4,657
jax/custom_derivatives.py,sha256=FWBXGy0zp70zI46ErbGucn23R6axD_0BuvZoeZJaW0U,1341
jax/custom_transpose.py,sha256=BmKD3o61GFyTE68GoNX_3Z8NBJl5HItzBCi8RehA9ks,644
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=_-zv-zCv4BirRPOM8Uy6zXOCp-V4ej4zNDS2YTiCRSw,638
jax/dlpack.py,sha256=Xq3mQbR6R_ak_CD2itRttfJx11cVaXtdl_xLQL3clfY,653
jax/dtypes.py,sha256=Mp2QLfj2hwLdfJHz8Kpzg2FiX85hJMyBLo3rapL3iMo,1146
jax/errors.py,sha256=zIyWVOrlGmSSVF5-50dOxgZLfgPeZU7VNeQx6i6wFpc,1272
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/profiler.py,sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=DLfiGlzcPAOi-DCoo8wIqn39c7ZyEyXmBNgMykTs7ZU,8990
jax/sharding.py,sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94,1184
jax/stages.py,sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg,1272
jax/test_util.py,sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU,835
jax/tree.py,sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA,956
jax/tree_util.py,sha256=z9oGFs_ZJOhnsc9_ljiQ_rbkkwVTooz6XOxUunAAq80,2971
jax/typing.py,sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk,3331
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=3fDfFKASNW5eJ-EDpGV3YntNwKg9FETdthdnchVrs2s,6132
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=r5HI4QdiNeLvvysAx4HoMR7MqrEzEbypFIzc8549OKQ,2680
jax/_src/ad_checkpoint.py,sha256=FFRmP7llpRq_KBKMGhr26gsw_gUFJ9iuEvQ41T7RCSM,36532
jax/_src/ad_util.py,sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY,3773
jax/_src/api.py,sha256=HMfaxRovlmBVJftzTDAS89Yub5aj0UVnJl6-uhzJg5s,130113
jax/_src/api_util.py,sha256=y9Hk8Lto3mameVddN7iz2HdFKwYqpqIXKVcWPSTO2Pw,28383
jax/_src/array.py,sha256=muWlC9ygk-uFe_lrByHf22o6HetmPqC9Jz6pqxz1_vE,37113
jax/_src/basearray.py,sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A,4296
jax/_src/basearray.pyi,sha256=2gzeRPRuIcLJuSQZL3zOBVY1izUIkPonPhEjAma6RiI,10732
jax/_src/cache_key.py,sha256=tdlwLLhNmNl9VOeZfGAJrX6eOgHbEDMMPvfLnk_uu88,10498
jax/_src/callback.py,sha256=PI9maf7onxINRRhmPI4fxR9jLVSw-Ad0GxLRQPdqHak,18988
jax/_src/checkify.py,sha256=GKISKwXgUnAxgMO9UPrqDG80x2r83unDD2nw4tyOq4w,52918
jax/_src/cloud_tpu_init.py,sha256=0IY467hAWhiTXeL4PXM49RCrci2298bSurPiQ55paXM,2371
jax/_src/compilation_cache.py,sha256=45YR_4oGGUkvFm5-vDt-Vm1dHxAEb6R68_YwfhwsQoE,8842
jax/_src/compilation_cache_interface.py,sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c,863
jax/_src/compiler.py,sha256=p5bcWfLgMtwD---p8ndwdNtRXrRIkdtPbbgKs4PBhJU,20648
jax/_src/config.py,sha256=dtrumUo8XH7iiFDSE7iqCuVpfF9hvYkU5-wLiwiwpwU,61646
jax/_src/core.py,sha256=maebiZGHNjyuVpGiyP2voeX63gO3Jcuqxs2bl2Yms6s,118878
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=x0zm1OkE1g-bBo6w_W4ul3zGnDx7ouUMTLRLz09-6ss,10502
jax/_src/custom_derivatives.py,sha256=NDSZvq1PdL2u4nXp_Wx-jwCvfD_o2QQ14uPvdEtuykg,60484
jax/_src/custom_transpose.py,sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E,8310
jax/_src/debugging.py,sha256=oKvllSijZ6zbDpjy7d424qWEfoxEsgTIlg4xGlvR6tU,24309
jax/_src/deprecations.py,sha256=c2fclD9ZpSKNQHNTndSiV0nR4dsmZS4qG3Tfnaa4JKE,3064
jax/_src/dispatch.py,sha256=TxHT8sTirhKfBTyP1nYiNJIjciVp40W9mwSjKZt3Yds,17641
jax/_src/distributed.py,sha256=OrDKJreVZYWombPwWWlOO4f6Xm4-w_g2Ttad11t9ha0,8865
jax/_src/dlpack.py,sha256=aCfyJPRz6IsWE7Eh2kIVO9NLiyD4l1byoxZPnQXBPXE,5309
jax/_src/dtypes.py,sha256=hcLTsa5nCSyAwhA9KHABHQ4aHLTLAtLVAhNHnecfzAU,28551
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=9Gwd-KeYyMlORpCWQR6a8ruZgnJkom8qT_1RMsjBacw,2010
jax/_src/errors.py,sha256=2yIUoXYA4XLYehScnEoerQIu6geOQskNqQV-_6jerG0,24766
jax/_src/flatten_util.py,sha256=-9ozNDsKw2QRDf5J4UjIlhTPOv3pGloUMKffS9aGteY,3554
jax/_src/gfile_cache.py,sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY,1923
jax/_src/hardware_utils.py,sha256=s0UuPPWP6eB7oqQ8o02dL3RqzQTYTsMKcNCukGoO5zI,1714
jax/_src/jaxpr_util.py,sha256=4mB7jfzh0IhgqiqnStUnL4_msTyOljxgzG56pw1-x3s,6428
jax/_src/lax_reference.py,sha256=hZ9Pm3Dx5iJa5yWm0oSvkrAbyuaEartugDi8uIuW454,17330
jax/_src/layout.py,sha256=f5y8-wOswaysEWfpc_KKLdgfz1PIH-NlSFP4ZgWl2HE,1231
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/linear_util.py,sha256=9s_-v7r432MAQc6r-juktBEyIu2-tOYh0j9orLJRLs4,13946
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=2BE9tfqqdOzmOQzwzzrOEvkj_P1AZWnyPZXLfCvKiO8,86657
jax/_src/mesh.py,sha256=I2YfKPwdLK9NRunSqs6ImaAJyXS3JGvYtOFunO_bzpk,11314
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/op_shardings.py,sha256=4FR-A5ldSwoP5oLU0SW_kjmOPr3rc6usnZ56B11wl98,3897
jax/_src/partition_spec.py,sha256=apayEnjD-HJ0NykjgoUquNhQ9W7PwsEKZOPqGCx6GcA,1688
jax/_src/path.py,sha256=Zd6Q4jJEK8m4MOVUsiT2ALoUDHvoZjP37SZFYJzacVI,1121
jax/_src/pickle_util.py,sha256=UUNYUdIIcN9G8NW5EwRiVuxJo7zP8-p4v66rwq7jt9w,2302
jax/_src/pjit.py,sha256=jkbmUevVHf2NJBJMoAHIbE8tbrtlKjT9y0gzEx3bMjg,103341
jax/_src/pretty_printer.py,sha256=gy89qxIgK7ZG7P9cglSmzrtFFjPS1uXWdcUbUjH1iFo,13077
jax/_src/prng.py,sha256=fX5ba58BCyFyuEi32H22EXDby5V_H2Es8ByTivmD748,49241
jax/_src/profiler.py,sha256=twgR0_2CzcOobgt8kDAoF_Sf89h5fHNWRXbWYwoZE4I,14525
jax/_src/public_test_util.py,sha256=Z4p1XzuocBDSXwI2HBFkssqPrIGpO3R3VETa5gyi2bY,10451
jax/_src/random.py,sha256=o9E2RqCsSsgD6pu_E2V39SpVeHxT67N7zuTUY_54bt8,100218
jax/_src/shard_alike.py,sha256=sdSYS0jagWfqS8pkdLEqQg6tJBr2QoU92dEgG0Hv4nM,3978
jax/_src/sharding.py,sha256=fB6sGov0DDVy1eCAJ3MrwypVzQ7Rho85-0QNXv6kvO0,5070
jax/_src/sharding_impls.py,sha256=nxvhLRMPmH28uEE_1UAOJTbauloMb6pe5OU6Gjkzgig,50872
jax/_src/sharding_specs.py,sha256=9s9WfSdGriLgmreTp_2xlqQtkf1TmSxCWRmtT-q--1U,9207
jax/_src/source_info_util.py,sha256=AjLwjmZXBhYFM13j4kMBWKK0Ch1A9Q07aKwyt3Wcefc,9912
jax/_src/stages.py,sha256=asm4WlL5rFZ6Q4s-XYSTI4xElsGgETO1UuXy4FH4aVQ,26931
jax/_src/test_util.py,sha256=YbzZlcccPLbfIdCI5N62MIu58x53cFkfUPloqTXLIS8,62652
jax/_src/tpu_custom_call.py,sha256=heMk1vjYPSuvy_MbQBMWLGVEDroWHjcZbp6-Mi7-W5k,18012
jax/_src/traceback_util.py,sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA,9060
jax/_src/tree.py,sha256=qf_oRe7McgufUBFD7LtV9KG8XL_fJB-kn7SZDUaSka4,3492
jax/_src/tree_util.py,sha256=T9sJ-TWs4HRS7ZpTZYxTf3D3xxkGmv0LkBeEFzxqNlk,41254
jax/_src/typing.py,sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI,2892
jax/_src/util.py,sha256=EXgaKRmaRT1MSHfWDNbU5TMM7T9rBFqNfHEaqiNXLLY,19149
jax/_src/xla_bridge.py,sha256=V9ItYKtrPwWwE1XyVpsGjyvLO3YdWOFbCXnFju6aP94,42572
jax/_src/clusters/__init__.py,sha256=ZIzPN7-Awrf3mDafphxQXBXWwItcp_8axkztAA1sBYc,1095
jax/_src/clusters/cloud_tpu_cluster.py,sha256=ILjkSzl0OI8cZyUzwCTAsXDG2LcguWfr1o4B4ZzKyuM,8148
jax/_src/clusters/cluster.py,sha256=4zenRWGqAo0CHbhEq_X0lNZDvDm5WsWwNFzHYDmvtPQ,4901
jax/_src/clusters/ompi_cluster.py,sha256=EUK4RbIetmu9LL_jzxr1cT3kRtTfgiRI__VcOLGKqmA,2294
jax/_src/clusters/slurm_cluster.py,sha256=1AmqbARp7rNWe2v7UI_AD100v3qdg2o0IV66zuCf7M0,2297
jax/_src/cudnn/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=JpSdIJC2zw-dq3Kte6531W-Esh74VlUz5iEKCFGxKn0,36670
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
jax/_src/debugger/cli_debugger.py,sha256=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew,4780
jax/_src/debugger/colab_debugger.py,sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA,7823
jax/_src/debugger/colab_lib.py,sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U,4298
jax/_src/debugger/core.py,sha256=0rEPhWjcCaiMey4sAuDivG1H5RFqqlfesAxlZZ1rlRU,8369
jax/_src/debugger/web_debugger.py,sha256=WJzG5FubQxA9EiuyXzHUb_v2_2GqvovyQiy6EaBomY8,3282
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=bAtdkmxevVfY9JfcwrWX6Ah8RXMoNelXg2I644ePHCM,1276
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=sWBhsUcujSz-pDPJe-gqiK-byJk_Dd2gISoT7in8dXg,13758
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=VBDNP08YLWvuUdt1HBVx8LUcBi17jXScwuTQejB7wMk,15299
jax/_src/internal_test_util/lax_test_util.py,sha256=LBych5KKJ1fYfxHTk9YyyiOuVGljYYyngz1Ckp__BNY,12148
jax/_src/internal_test_util/test_harnesses.py,sha256=ZJBbtnWCQphK67bXX5dCus2XZMURqyAIxR-JoGJ6FtM,124694
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/ad.py,sha256=XGIiZyYNXujjGE4M4T6feZ8yStPKiXaSSBiunyn5sJs,32701
jax/_src/interpreters/batching.py,sha256=NJDTJxEy2T1BVJVXzyyyjMWcuua7lmTUou5Og2V6EjI,50430
jax/_src/interpreters/mlir.py,sha256=M_naKrJPWaPj9iMOcVswS_kLv_o-_eWU81WdzlO5eqo,117364
jax/_src/interpreters/partial_eval.py,sha256=87xtsxD5rbMZNGWnEwnXCn-9JK1tMvQS5il2TJVYG6I,125831
jax/_src/interpreters/pxla.py,sha256=DwXz98ziAunCaZQ4-UEDlFl7bfjmBEWvSTaHcLGVJEM,138402
jax/_src/interpreters/xla.py,sha256=AFLJHc6vzGVwcnD8Bm7ywuLeYraYJg8gEDXANMd_R2U,12203
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M,17623
jax/_src/lax/convolution.py,sha256=SNNtSYshLLvMAMwsrTfnDPGO-deZeqR-Y_Nu3rebJeM,44715
jax/_src/lax/eigh.py,sha256=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=izHxbi-S686onnDHQC3WRuu9PbKhDhYkZUtZPx6LxFI,10151
jax/_src/lax/lax.py,sha256=Fkv-NIpJvInniuEguhUvSYCIShhiVw2MzKyAxZ96fXs,210995
jax/_src/lax/linalg.py,sha256=pq380CYVFKEUJNLQxJTQYnPXP3qm_dkdRn841ucM0kQ,91611
jax/_src/lax/other.py,sha256=wTmrVnNdLQajm04LyGa4ioIiJseZ9dT9bWbQmCSmO-k,10539
jax/_src/lax/parallel.py,sha256=gUSrc33otbkZ8jt30h3BtkfFJDWJz5j98TSy7hadp3Q,75667
jax/_src/lax/qdwh.py,sha256=f3QJwPMDDO_qC-VFaKrzSXUJ0njCSwSEjICsA5FckxU,8028
jax/_src/lax/slicing.py,sha256=glCJwZAhftQCVnXybrH3KhxGogIqFVpVwW0glM771mg,111664
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=vSEwqDksgYifZlIxtqjDodJVul5b_C819162EZZ5zcE,11705
jax/_src/lax/utils.py,sha256=KHbNG7zWt4w8vrSxSJv5iKubesvLdZlX9HVTXLyr5Iw,4968
jax/_src/lax/windowed_reductions.py,sha256=StHNyfk40hMA9-SxVsI2_yC0w671r5tvQcQbua3A5Q8,40180
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/common.py,sha256=jE7vo0resS9tHdg9F9nleNB9Gwcv8X3VUzpMHNo0OxA,10976
jax/_src/lax/control_flow/conditionals.py,sha256=x3vSQsquCWjZ4XFydP0kuanzgdN12wH5LdMsTfntr3c,42956
jax/_src/lax/control_flow/for_loop.py,sha256=3-Cfq-xoXWruQDDtzH8vDgvtYNZvDCntU0fY3BH1ZRc,37122
jax/_src/lax/control_flow/loops.py,sha256=mb0q02qk0otlJ3p9vme4_oiWM38JwgpgttXJtsq4m8I,109671
jax/_src/lax/control_flow/solves.py,sha256=MtOEjg9pa0NTCfVkPaKlfEobIViNhOvKfvyv0CsR1g4,18874
jax/_src/lib/__init__.py,sha256=HSyqGdFIAIFHVBk45EIzsqK-M5HHqza5-5ubmXqmipY,5476
jax/_src/lib/triton.py,sha256=HbqoIneu52En_DHWH6Mph9rGtj324CRCXoT0SZA3u44,1336
jax/_src/lib/mlir/__init__.py,sha256=vv85BXVE6lq8QZTUZtOIFPrwcoIWROkeiAqVa5a8SB0,885
jax/_src/lib/mlir/dialects/__init__.py,sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs,1195
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=_pJUINxbGf0y7K5pjWIalL9Pg11eraNN6CcQfCD6LpI,19224
jax/_src/nn/initializers.py,sha256=526DihqSYNUWM4i5EET9rvZl0btVP27vdZs8Tpl-Qu0,25570
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_methods.py,sha256=ISRBbYRa3HcXLjMD5egv5Ie7khsn5224QRBw-E57Gok,32582
jax/_src/numpy/fft.py,sha256=rVEbkoRIJ_WiGPnrsl4bztstWiZ50acmrFPvl9pde58,11373
jax/_src/numpy/index_tricks.py,sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g,9888
jax/_src/numpy/lax_numpy.py,sha256=srvD6i-Tj2tmPqL6yNcmRt6efGbqMJkXHTHum001TVw,219682
jax/_src/numpy/linalg.py,sha256=rtiU5yILGJD_xcQw0Kk_xTo-XEW99Wnt0gET4UQGCpg,31450
jax/_src/numpy/polynomial.py,sha256=2uEH06Zqya_x15M2ROYEaV68fwmwCmXs-gKZe6mnXzw,12162
jax/_src/numpy/reductions.py,sha256=VaDp9mMQzwGLI37b-mc3yiEWSAGtpQ790kEZ49TnGbM,41617
jax/_src/numpy/setops.py,sha256=pdYBHaTrUHOhQtNIvH2VXy3kk2Vk39VmpM2BdUWwl0w,15407
jax/_src/numpy/ufunc_api.py,sha256=RbGKNxN0MhSzkQPQOYbEeJS7PaaEwMtyDrS4MqbQPzg,15264
jax/_src/numpy/ufuncs.py,sha256=A4kN0GkuisphqHQBk8A8nCdewKXGUqZrPyl2pxWcm84,28522
jax/_src/numpy/util.py,sha256=55wWiEaqu2SHthyoIB6SYdIWCQeUoFmehrWiP9kYwfQ,19129
jax/_src/numpy/vectorize.py,sha256=H-OiIbDQvKmjpOJlO9L7cpQOYn8cOJQod_7C6OHOrB8,13021
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=kZ-h0mJzy57grnDe9PFZdv4TN6u6_OxRauIRtCVZ214,17593
jax/_src/ops/special.py,sha256=nEM6g8Ha6-LY9Ai7gZaMAH3LkT6Xj362V17yKVoP8QA,3884
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=fJYY46uKMqa__bwaXGOFh0J2VpvT8tVAArDvlQr-m4E,12353
jax/_src/pallas/pallas_call.py,sha256=0GTbGeEH8P8QNxCTskXacW5EoR-3snd8QbnuGNlJwcw,25357
jax/_src/pallas/primitives.py,sha256=SQ5tbo_O0qyHyzOWSYKlq8UrZfSWL9PjhKp8SisX934,16743
jax/_src/pallas/utils.py,sha256=Z4RR0ZkyZWFiILDn48GIS45yui60vtDkiqPhca2JohE,5991
jax/_src/pallas/mosaic/__init__.py,sha256=mlQsjEmalHvk5cLlYbm-VCHeEDbKRasWqtDrrttNN0s,2462
jax/_src/pallas/mosaic/core.py,sha256=pIsvseu1Eu8kRKBe1ftlMVMWfvOjh9s-SPJyItYse6A,7643
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=ZD4tfVy8IfuVGG-7aOYJ5BktRfgWS8_PCtAEaTtAkNQ,75303
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=Wwio6VdKTx63oLMtn1GD5AH_4SkvGVCqMKxk_Tb8Xd4,4250
jax/_src/pallas/mosaic/pipeline.py,sha256=A7WI4ankZcs2wqaxUUDnOaihha5a3MDoliC4ixPXEFk,43393
jax/_src/pallas/mosaic/primitives.py,sha256=J6qUdR4DQvQB-9VfCBqxZVoxm5SKIcXL8R5OOtZVPDU,18126
jax/_src/pallas/triton/__init__.py,sha256=x4B61dVc6mxT_re9aORVC0QfLvoSo2g1lQfudxuIxmE,1120
jax/_src/pallas/triton/lowering.py,sha256=hKuMtK6lptc5A7Ds4ZZ7E7bzFEQP4FIuW7-Skc_hOIg,79836
jax/_src/pallas/triton/pallas_call_registration.py,sha256=XRcnTe4ylzz2AfdyNg-JoVDdu3zSsc6HY6dEtP_O-oI,9863
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=G9Zrf4TsbLQH6BZlKwcWFaDXiJcEiuL6eVG59avHXhk,6249
jax/_src/scipy/integrate.py,sha256=6I9YyM7lwGwhpPVmDzy6wk2f9jz1iqVQVfiVNiNYj7U,1036
jax/_src/scipy/linalg.py,sha256=a8twFXM19x3qZuxco2YU9-2l31_VR4Wr4cqEEf4i3wo,41777
jax/_src/scipy/ndimage.py,sha256=viYzxaVZVP3pXG1yBKMSG5wssJUKYDerBxUcXfCT2ds,5362
jax/_src/scipy/signal.py,sha256=IuhZdas7Uv-mk9u_WR3bW4aJQn_jYakhr0bwGtvEwBE,27552
jax/_src/scipy/special.py,sha256=-Hho_ccefglXuNDKh9OTNfVmY0hl-RuDWFrKQZEoKoE,63694
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=lmudnV7wOY8eN0CUT2a7GxA3mhwRGmjPa7cVF2PNB98,1779
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/_lbfgs.py,sha256=Io1xjWLc90YFTCPiGMz-fP9_FMWlpNuymTIVIKJXjZQ,7790
jax/_src/scipy/optimize/bfgs.py,sha256=fJ5GdaoI79pGHpFlX0USGF-1ooGa_-bKjMVvAaqqu1U,5582
jax/_src/scipy/optimize/line_search.py,sha256=JF16fsbL1kGJnVRRXumssJzXmsV7cQGeDFInmxYnlG0,13229
jax/_src/scipy/optimize/minimize.py,sha256=miFHZRaf8KpiDd4XsEUAKHUGrGodKYCOiKAITye-BIY,4979
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/linalg.py,sha256=VCjQWXeUaf_9xqTCTIiGTVkMjaa9IvEwKgLrhinavEw,27326
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/transform.py,sha256=VGTcZRDye7xJBLFPiIJM1DJdJTDiIPGytz7HFR0o4QU,15167
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=FLIRKSf8SY-lKXwscEFVn12ALW3iQtu4QDWH6HhqQMc,5725
jax/_src/scipy/stats/bernoulli.py,sha256=9PntVJ6XZC9btsq51pzvgi1Ndpg7IGgAZzIglEOSyL8,2270
jax/_src/scipy/stats/beta.py,sha256=YxIfH_-b5vXkqmOiC5n57J3kZMmkkf_aYEawfwTtHLw,2968
jax/_src/scipy/stats/betabinom.py,sha256=vusiYROP2Vk2PNZs0E1BCiwEpFaIlRC1hZ02v8-Ga-E,2005
jax/_src/scipy/stats/binom.py,sha256=T4wO99oUze6-nuv5Vw6_crdy1jury62cA8k29RYN9CM,1671
jax/_src/scipy/stats/cauchy.py,sha256=ejSOHHOc0MJjsRFTJstpP4l5pTYWFWsRJdczaYlgbHw,3142
jax/_src/scipy/stats/chi2.py,sha256=Q-vgl-gttIpxzlU59fSFIR-08fZkl4cT7P1MPK7cKzw,2952
jax/_src/scipy/stats/dirichlet.py,sha256=v_KIk7RegcQYeNOD5memO7kLmj043NzmggMPVf73WSw,2148
jax/_src/scipy/stats/expon.py,sha256=yVjLkyb0z9tQXtltE0Qk6WHjeQlW7IW0G78ypijU7vM,1326
jax/_src/scipy/stats/gamma.py,sha256=gOi_AqGcD-d1nys898_UFEJF8dEaKzN2F6KzcPHYxjs,2567
jax/_src/scipy/stats/gennorm.py,sha256=ESjS4pr1gpY-dKLfkBLzoBiuVmqx3xn7dZj5hHqvKCQ,1311
jax/_src/scipy/stats/geom.py,sha256=PC6utDsp2J-AAm9ZEt5HOe-C94Eqe6DiNT_HAjtxexE,1394
jax/_src/scipy/stats/kde.py,sha256=r-Ua7AKOF755cE9OPFvtodfXWZ0jo5U3DjpUoaKks3g,9876
jax/_src/scipy/stats/laplace.py,sha256=uaazCUEB7qdWY48bxV2sP0msvWb9dpAxjkNnhqQWzEU,1803
jax/_src/scipy/stats/logistic.py,sha256=3HIoYI-iN2UVGLc_UHRpaNw9a32VfPsoUObjCJWOjS0,2434
jax/_src/scipy/stats/multinomial.py,sha256=Vgndccd5iNj1sV-2q32oHOzOeGo7LNLBq52ICRwq0qQ,1687
jax/_src/scipy/stats/multivariate_normal.py,sha256=T86VsQGq7585Bte2nK9ATHCWKkkUuLk4aiuDqmw-ssU,2231
jax/_src/scipy/stats/nbinom.py,sha256=GTXNvC69AUsqgnnT2ySJcdPe0Eq1bjNdrTF5BPP8rEA,1710
jax/_src/scipy/stats/norm.py,sha256=CXkWEgZmVayncIpL2Vi5rLNFVxqlz215qedtEQm0oPI,2899
jax/_src/scipy/stats/pareto.py,sha256=ahSAdtFCCf10Idfjsb0nvecNzgVvcCm8X5JxJnRAEOk,1509
jax/_src/scipy/stats/poisson.py,sha256=Xikr-qVqCQvD8BM12XQN5QquFgI00gsriu7oCfQnuUM,1700
jax/_src/scipy/stats/t.py,sha256=vx7uvNW6xzMTCOHl6423T8ho6E4oMyRNdqIx-ODAUiE,1834
jax/_src/scipy/stats/truncnorm.py,sha256=UiA72yshSWEQFQJkfxgUO_jHQt4P7A7ZT6z56OxzKRs,3993
jax/_src/scipy/stats/uniform.py,sha256=ijGxgJabQi95MyiSex0aaBOJnBTUAnJ4KF-lhJcBl3w,2122
jax/_src/scipy/stats/vonmises.py,sha256=iCC8-yFGHGH6ky4XD_2ISRt-cjRIh2acHKYBwIPFzCQ,1284
jax/_src/scipy/stats/wrapcauchy.py,sha256=-k6Wt4lROA5sK9Juyc78a8YO3imF0HRx0iYLW6UVHzw,1422
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/discharge.py,sha256=wS-uXlK9iOM-WixcObmgO6f7gjqKgce_edxteHBXw7o,36963
jax/_src/state/indexing.py,sha256=Hykl5Z-UNtQQ5LduX_OJM3ggZha7z3nH3wjh2FxWW80,8443
jax/_src/state/primitives.py,sha256=v6IOxjWutFixvreCKU3PboIeY8NlIsL5tAKjW2QXm1g,23555
jax/_src/state/types.py,sha256=AA-MoZixDqnHWh6upqlJwtfaGlKSuJfwi8tVf3SmBbQ,6685
jax/_src/state/utils.py,sha256=AwdUqrfZto_M6x2L5L4KCoBARMsjhYwfb3jvjmawjsU,2411
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/linalg.py,sha256=HDpnzipYvKzz7-xxu_2Ft56Ty8czVtpxhmNB2mhIiVU,6231
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/betaln.py,sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=A_SKMrHHpf6AnbS1YgGxg3RPbNQo9FWgOGLLkslwHQo,5881
jax/_src/third_party/scipy/linalg.py,sha256=PyQc2ABPlYIjbQF8pH8729L7DIHNAiN6JyjyjYOjZao,2698
jax/_src/third_party/scipy/signal_helper.py,sha256=VqLSblEWCItAViXc6lIXRbF-FT83l9305aHIZP9B788,3081
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=224Ba6KPYLs3zlF3g7dlubbQkIkvoko-WGqDx4svBzE,20559
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4,877
jax/experimental/attrs.py,sha256=m_iOiq5bKo6emXg7CfpdWAvkYq56Gedhmoe4u4osWgE,9155
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/custom_partitioning.py,sha256=VT4g15KfF4BAv1pxOI61_jKdI3AHHkhQCnjQcQxWFuI,22825
jax/experimental/host_callback.py,sha256=mKudOnX4IhhbJYL10yHbCSamsArUkpGa3fS_86oE-Qc,81388
jax/experimental/jet.py,sha256=k_ZTCD4YyQ7g7bieTLP2zNnKZA5jHopI0eqKYGXwsg8,26689
jax/experimental/layout.py,sha256=tsBD272jSV5nas1rK-h27UQpAEVHRs825quM3hHLMrU,672
jax/experimental/maps.py,sha256=NmRFDPQdGIFLD9YnSOr9Ui7_kPZV3JaV3MXGcVr_2vo,1660
jax/experimental/mesh_utils.py,sha256=n54kJr0Q-Vtwc2w9Yi42bqFYpyz611K-XcqdZS2Bh9M,27505
jax/experimental/multihost_utils.py,sha256=xbm-MycVR98sd95Tcstq-UYewQ8JRpvSwxagmrhLokQ,18355
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pjit.py,sha256=Yq-bII0QH18i7VYcphiiQmvx3xRcr8Ktsr1ijPhVepg,808
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=HL4GGWdBbp5T-4KH6Jwq_FCqtbslppjzf8eCbl85cGU,18684
jax/experimental/serialize_executable.py,sha256=-ytQl3LPU31nJnxHC-eZEJcHx724XiCm9-ThYj2qaUk,3041
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=noCZW0x5wCwnTz2cD5T-FTFHGtBsLlKyGxkFRln47Ig,81891
jax/experimental/topologies.py,sha256=DxNh3ErC2h9eUrG8UzN007aE-r_EiZtU7R2NeyWkbus,2140
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/array_api/__init__.py,sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA,5364
jax/experimental/array_api/_array_methods.py,sha256=6rP6HnjVF1CBdSpwf0cEkj1wLL4rbUTKkiuIBaWueb4,1759
jax/experimental/array_api/_constants.py,sha256=Q8eoyHE5GDoeBUgbur2mF3_RzUW6OO1PhzneUzmnH_c,669
jax/experimental/array_api/_creation_functions.py,sha256=RdV4316PQaXg5sf4wVrWWw_JQ0rf6akXgYj26XC0Kdk,2496
jax/experimental/array_api/_data_type_functions.py,sha256=v3ZQsTh_9qLqdA6zKMGpfbYhRn_OVCueGM3YwqNXtCs,5391
jax/experimental/array_api/_dtypes.py,sha256=4itd8mk4z4SpohYVHfELDpGQF87fu2ScT61nBD3tjG4,968
jax/experimental/array_api/_elementwise_functions.py,sha256=hPsPNdyFwEg8UGjz8xJnLqDlH_vrLT2_44PIf9HWFqw,14835
jax/experimental/array_api/_fft_functions.py,sha256=okg_ZQy62JTj2_CVfOkvrwn6XyCBWwX7ScBdLDnVWNg,3096
jax/experimental/array_api/_indexing_functions.py,sha256=gemN8vpsGfjy1GDilIQbtes-NHIvE9A0x18mqxZR5mk,675
jax/experimental/array_api/_linear_algebra_functions.py,sha256=mdCDSzLE7XMpn73zOe1uq1yyQ9JKrBo72iygziG-ZQA,4775
jax/experimental/array_api/_manipulation_functions.py,sha256=egQnTwdYoV8c65D0rh94zUG4F6x_zggjsNiCGUO6j_I,2826
jax/experimental/array_api/_searching_functions.py,sha256=VhWWD3rH1lQi4EfoGzKqC74bnNmCB4pmByk0WqTG3ek,1483
jax/experimental/array_api/_set_functions.py,sha256=Kv-4PcMtsszLMEgG_qK84kt-2qJGai0KSa2ArPr22W4,1375
jax/experimental/array_api/_sorting_functions.py,sha256=XMYMwfgPC8AMC3NIlhuBGe3hGx22WqTmSn65CIegKzA,1118
jax/experimental/array_api/_statistical_functions.py,sha256=0xcB70zRpzfhTqOZyKEkfvFmXYSB8gBQPInuX_ai5T4,2003
jax/experimental/array_api/_utility_functions.py,sha256=Nx0zBTWs_uXSL65mo3rqnoQHAbFatpfkQCsv4UsFCuQ,975
jax/experimental/array_api/_version.py,sha256=fVbhvRAhSrwJXVDwqRQEt5vZWd-PGzRQkxJ1zmfiMFA,616
jax/experimental/array_api/fft.py,sha256=1N17z4OpG086sZne714AEJrfmZ3OnagMyMRYcLQ48aA,906
jax/experimental/array_api/linalg.py,sha256=OEDELaMerquTrn2FSXDImzsrawzBUdvWXG-da3MkixM,1153
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=r5UnmZ0o1X8F2FCrXIHzCG0KiKZqK71IKIP8sJyRmA0,23136
jax/experimental/array_serialization/serialization_test.py,sha256=INQMiRaWx7tKYwEwonhReD-BEIYa22XEgcKYjl551Sc,15781
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/export/__init__.py,sha256=dk1YZG7FSmrB47imONOWDKSEsX26XyyjI5WiHVULU8k,1268
jax/experimental/export/_export.py,sha256=suWuyMuhx6-IHpcOtgtkjkWhXopiUKZmZpPBOBqYDpA,58950
jax/experimental/export/_serialization.py,sha256=f3Bu0wxMA0DvJ8Fw1pMyd7W_xi4k5P_Npv9k3HXddR4,17212
jax/experimental/export/_shape_poly.py,sha256=gElBg9TnYmy_gpxDGsLAwEd3Wa4ZFqa7Dx9HbL_Eo14,85388
jax/experimental/export/_shape_poly_decision.py,sha256=x9AADPHzkyQSjJEtxbIX0rOsWBqT2T2_TJ1oXsgTmN0,20711
jax/experimental/export/serialization_generated.py,sha256=wc5JKltgI-SJx6g361gxpfY4OK2DvsaVARhrm5neJzU,26195
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=lZ1TLh0iFSYWRMCcmKkfym_DLLGCUFS8ggqvA6Vy1-s,27665
jax/experimental/jax2tf/impl_no_xla.py,sha256=SIvqeN30ld0PqoGduuUC3JwMxvCPvOsqUZnWxF_M9W8,54476
jax/experimental/jax2tf/jax2tf.py,sha256=j3QphITfS649XeDCs_X4JJXTV6p4WEzd4eceKxNxADI,143199
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is,2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=1FBpLLwCpm9v4udaL0tEN3umiqF4a1lXSr8k483R9Eg,1616
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=LftkJ-7__c0JpPxUWdx9uv3QiTGbOKRXzLhc1ixkev8,11520
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=UdSLpLnb2zABl5MWLZ7JRfGinscpWMSwXHb6a4rtFxA,7096
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=f4BcJt5KvsNvQrZsKhV4_J6QrxHkppRgu792dy9E2S0,8051
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g,2467
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8,4952
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=9FGx-2zxn89IzuQ6OYhaqA0zII0OQ6L6jktQvHzSqY0,5412
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=3bt6V_bcE223ShF5pezs6aoMpVvCf5ouZ1XtBuYSaiU,64801
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=cM5bEK9TiF_V7FDMA0Ix8NUZhuNAgdb8K6qBYFrb9fc,10127
jax/experimental/jax2tf/tests/converters.py,sha256=eqmQzNLSxZ3fhRIjJ5VpuRUH84VfnIHqju_na1OFFb8,3528
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=mnuni8_Kwueia-gIq_sbqi1l1OHeHscO5rXWNgzT9iI,7421
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=EdkD4mRtzfAcIqpnrAagOR9YvzBcFJlMAGyNkqCPqp4,59208
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=MdIt1fAu5R6WxjtRO7hc2Lmw4Lre5dzj9HVJeTUlzPw,68314
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=F_tkcePs1M1XDh83wlgFEu8b8iPZMJwkvPu1yWYI3cs,6710
jax/experimental/jax2tf/tests/model_harness.py,sha256=1lzDFrtZV9XmDkkIxoQO0VtNPlmzptr1RUuf15mBjQ4,13951
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=wP442L9HwVRIG2hBbUy1N1GNLfkPxCS9wIkGS05o91M,14099
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=YiGeI2270Z2NM3alt_e88SeptRd7O5Jyt500GgqzTxE,134345
jax/experimental/jax2tf/tests/sharding_test.py,sha256=a-OWyXIgrushNtZqrEyoSXCWj6h1X-1JtN9Yvq4k1os,31884
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=7L-xikndbablYLUybeHvRQptIL4vsZ-xd7Yoyh_XvAQ,20034
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=ZeHQSVN-jnvarc-Onvs8Qi_R_123kbAc4ZddL5jQ798,22197
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/pallas/__init__.py,sha256=gGfMuwK6zh0Dgg5QMYa3OJM0PEgNKG7rm3Drncb9oMA,2094
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/tpu.py,sha256=DPAT3IP_nAFNBDKDmu-1MQcyIZ0m5sDwwWuf6YR1pRE,2187
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
jax/experimental/pallas/ops/attention.py,sha256=hzjxm73o_bPA6IQV6bHkblFwYLO_ov8GXZoot57Zl6k,19059
jax/experimental/pallas/ops/layer_norm.py,sha256=69JJCFFul_osv8y5p5RDyXGObLetVH1AUVUnDwY-hNU,11099
jax/experimental/pallas/ops/rms_norm.py,sha256=qaquKh22ka6HMiMZbq2h_vZhzRyZUHjZ4kp92ePohX4,9992
jax/experimental/pallas/ops/softmax.py,sha256=M8TJA66HR9OhvdQrA0SzrTrZNyU5n-kSkYuogkcWEIg,2702
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=7QywIkmmDXtqcCCBYK2YXvvMAbrI15AM9LKDX1jcTUQ,5536
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=nX5EZ_x3VmIvX-hZ6rgUa88Nq2_APD9ZRFqc109hHNE,49725
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=Qutkom55JBjOTRrTwsmPrjkZPUx3om4iqXxp5IUZ5c8,643
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=3LIXN9rn6UGN2Dn2oEgiI26c8ydgqc6NJirQWOBaZO0,27242
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=UMptQqdj5o3jgSM32_1zupFNHgD4ft7MQ1zfPbb5xZI,2944
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=bCNpNWR-hJeynDYamiq0BfIsRtvvBoGu2SopmYG48UY,681
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=lbDwEj9iGL2bW7tBBuTmUjY5OjcaMRvTB7NsLTDoz7M,14410
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8,2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=-5YjykKjvFwHvuk2NKd5L1lEwH5aaKbr3RfjHvNlQwU,73093
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=HddoyCimPGN66heae28qj7JnUOo5d2l-cK3bSlslhro,14542
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=lLtXMkHOllIL2OVH0HewOo-ySqAzTsqOkHlDy71k2LA,33390
jax/experimental/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=NW3WFfrvsf5O9sxB4vN6fMj6b7F0NxtQEuj3L1oO-NA,7661
jax/experimental/sparse/ad.py,sha256=TFSakBd97x3rP0i9KemXKlcIl8vfijmpxdvOggiT6mU,7588
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=pWVUYzg9AVseWedJhavgT4k2DDN_N3mWhY7m2iQv9_Q,128304
jax/experimental/sparse/bcsr.py,sha256=OUfQcM1pEzJLdlh_TMOz6tzcsdiYLmQzXFdRV9bwoG8,38224
jax/experimental/sparse/coo.py,sha256=MKH3YpJvSroCUItkDlUwHZxkJnyG1AXPlLsO47_ooGY,24122
jax/experimental/sparse/csr.py,sha256=5XdYEe3aA2W00_TK8pjZ30J0lnWD0dfwIk3YO0_epSk,24414
jax/experimental/sparse/linalg.py,sha256=fh3KocDf6MVEx_n1gVt3l15mAMfEJrZefFFtptoUAoM,23291
jax/experimental/sparse/random.py,sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA,3882
jax/experimental/sparse/test_util.py,sha256=41Hvh5w84LiGxrZ1TPZWTQs1yiTSTCrhv5LP0E-89Kc,9597
jax/experimental/sparse/transform.py,sha256=53Js6AbAApsDnItGALSqkw8MORqT5GLDEjImiTQkqCY,36957
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/extend/__init__.py,sha256=cDhCUK2LM4fNu0JYq8amg4G9dHflTjkrSyHSgmSEyMY,1249
jax/extend/backend.py,sha256=zDbInIE5czEJduevvmZjKRB_9Avtnvdkf7fHFYsQPrg,776
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/random.py,sha256=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU,1083
jax/extend/source_info_util.py,sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI,1215
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/arith.py,sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E,643
jax/extend/mlir/dialects/builtin.py,sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ,645
jax/extend/mlir/dialects/chlo.py,sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A,642
jax/extend/mlir/dialects/func.py,sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk,642
jax/extend/mlir/dialects/math.py,sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ,642
jax/extend/mlir/dialects/memref.py,sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w,644
jax/extend/mlir/dialects/mhlo.py,sha256=KkubU5XifH1vf8E9AY_aDM5nbDDafsnRfraerqfP4kA,844
jax/extend/mlir/dialects/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,641
jax/extend/mlir/dialects/sparse_tensor.py,sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE,651
jax/extend/mlir/dialects/stablehlo.py,sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE,647
jax/extend/mlir/dialects/vector.py,sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI,644
jax/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=BjuN2XLu4xEetW4JuqsC_FC8oiYRjHDqa4-p6kcv7AQ,3302
jax/interpreters/batching.py,sha256=jjy5WCy9pS2mUzdd0fjKViSEPO6L0e5zL5gdkgFR-eY,2582
jax/interpreters/mlir.py,sha256=dQAg30IEPU5_3NypHtc9IoTySAM9x_y5aSDivt-99zI,2800
jax/interpreters/partial_eval.py,sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E,4537
jax/interpreters/pxla.py,sha256=BBH-kEticPc4e06G9kkNL_60BKlhkySRUNl3GPvZHoY,1850
jax/interpreters/xla.py,sha256=w0snrW9QYScupJNeFkjKM6s0kvLMMavMkBeWDRYKb-U,3454
jax/lax/__init__.py,sha256=snyYN8ZtNsZnY_ZEC4zKNLGslOkbMD-kVt9LxOlSLCE,10943
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA,767
jax/lib/xla_bridge.py,sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw,840
jax/nn/__init__.py,sha256=uf005wuZkgtF0Tv-1Ex7er0JOHF4wYs655UnyOrcbj4,1896
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=bkomDQW_gDqER6PMsmJWrhFwPda8MhBifik2Ysfhz3c,11580
jax/numpy/__init__.pyi,sha256=eSqGVP0ELMNsGVt2OqNde0ldXvvQ4VN9vlKv2BT6kHY,38038
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=olFy0Mtm-dBsv_HhWgPki_J6jBACg0lzdFSO5M37Xc0,1451
jax/ops/__init__.py,sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ,870
jax/scipy/__init__.py,sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c,1474
jax/scipy/fft.py,sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE,809
jax/scipy/integrate.py,sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ,777
jax/scipy/linalg.py,sha256=yqsNTdJ19X6BxTrZfi8RFJZ_vSjebVkIZnuuKFPCshY,1350
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/special.py,sha256=jdGeirmuJrppj9EFta_eeXa_YEPK8Ma5OGNZMNVlY6E,1568
jax/scipy/cluster/__init__.py,sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI,768
jax/scipy/cluster/vq.py,sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM,776
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/optimize/__init__.py,sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg,822
jax/scipy/sparse/__init__.py,sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4,757
jax/scipy/sparse/linalg.py,sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/transform.py,sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro,802
jax/scipy/stats/__init__.py,sha256=7D_LWgxpwfFnL46KaPYEc4IVTmDj7K5ynv6fRuP_VUU,1995
jax/scipy/stats/bernoulli.py,sha256=V_oR9XYdYZBKZ3Nh5Y5nH_6HicaJtsqoaRFg4GYeujs,819
jax/scipy/stats/beta.py,sha256=rEx9J6IrNfs2Ln0Q1XwKl6TkbNqD5nIu8pdsQe5I-Po,851
jax/scipy/stats/betabinom.py,sha256=-B4rZ-30Nvr4Lxg9D2DXcqxKdBI3pEItbQ-8ncNJWs0,792
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
jax/scipy/stats/cauchy.py,sha256=mZmRyAN4umg3GfohvaArByDHuXK54FUHSTkN9ESk2xs,881
jax/scipy/stats/chi2.py,sha256=tO4FT3DBLI-jJTfiH1_f5SqEvnRadBHNQUj1vVqm-Xc,851
jax/scipy/stats/dirichlet.py,sha256=KOGl-X9h2kKS0YlfWzsbMT5xEg3ONDgn-MWvQMBmD_A,792
jax/scipy/stats/expon.py,sha256=WS-OP1ng1jUlodys956ykXfnvs18u7hfGzQv7PnKLDQ,788
jax/scipy/stats/gamma.py,sha256=x98RFaDH4iSYxLwgmqKHsIVtEQBLSGGVZvjogfF4Ssc,852
jax/scipy/stats/gennorm.py,sha256=_NdhXS5lWu6cDZgyizZfDlm1rFVDgXGAEOYIk3Cp7FU,804
jax/scipy/stats/geom.py,sha256=uyOJ8a88yGVTe5YTjg_JO0sLZzBi9OdHMAEOYxtpaaQ,787
jax/scipy/stats/laplace.py,sha256=Oi_fMUikhKflSBJCu1R7UjGkgJLij45M-_9Q5VG0yuw,804
jax/scipy/stats/logistic.py,sha256=c-_nDR9e1a8yk7Ngv7Og5G1vfqjhFLzdTMxDE5UwzUc,845
jax/scipy/stats/multinomial.py,sha256=mysALVSGzMZahxx69AeoGbS-p1Vod2RkGdHZfKJAtRw,794
jax/scipy/stats/multivariate_normal.py,sha256=OL3t4XR7qmqBuw7Ojl3Chvsbwf_UsqtkiTL-9gHQ8JU,802
jax/scipy/stats/nbinom.py,sha256=eMTqIxq9njSnXtEInWHGJBfINEo6N4yVrwji4jsyc58,643
jax/scipy/stats/norm.py,sha256=boHN9rGUO9hx51JuRTxMep-on2HmsNxIbM3YoN-c-HM,879
jax/scipy/stats/pareto.py,sha256=gzEhI-vI4gllIHFFSe1VaTvGbf6GYzPG4siMkxOKRsg,789
jax/scipy/stats/poisson.py,sha256=hMSNeBSToe3D9W6-isGDSZDB-Qyk8GBUtdT8J2PHCMU,804
jax/scipy/stats/t.py,sha256=RO4Zd0Lm2creTHIsFR2tVerxe5TdzRWMOlSSyaH3s1w,784
jax/scipy/stats/truncnorm.py,sha256=PnB_oNVAW0trLzh8ua_T9V0vPnrlBmeBpA1FkPDPoXE,855
jax/scipy/stats/uniform.py,sha256=CfbYJ8U5fEmNa15Rgeqq-X3nEVY1P3GtVDKWj8a5gy4,818
jax/scipy/stats/vonmises.py,sha256=nQvzYinMztVFMPZm0fd14ZmHxxoi1CZC_fvRgmBwAMo,791
jax/scipy/stats/wrapcauchy.py,sha256=aWOW5KZnRAFa2rgHxwTybH1c-gbzYE0KhGVdy6BNDbY,793
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/build_utils.py,sha256=31JhjUYidLlxyogfXJ3KvjiKSpKg9nb_XF8qLFu8Rrc,3410
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax/tools/pgo_nsys_converter.py,sha256=udKwSZG-RDhlvA_Z0vGkW92nR1BLXGVBdxTU0xLJhVE,2716
jax-0.4.26.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.26.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.26.dist-info/METADATA,sha256=iAhoIbsqnaLrNzO0mCwOzxcRvR_NTRMlHyyBQ1uKb28,23551
jax-0.4.26.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
jax-0.4.26.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.26.dist-info/RECORD,,
