jax/__init__.py,sha256=p4Cn1Um8oD6skJF8sPRlCkA_O_OGi_q5eBgOw3j8x8g,9576
jax/ad_checkpoint.py,sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U,932
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/cloud_tpu_init.py,sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs,651
jax/collect_profile.py,sha256=nCmC3aNbZgJ-djAkRdRSG10JInh6tLZX5dHCjkVn0VI,4860
jax/core.py,sha256=H40vR3n3MY6_FH1Qr85qH-DoiAZtk9B02LfSfe7oJtc,4796
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=CVhRiRgnWk5hBsc3ZX7mFD0Yst2S9VDAirtwF8rSZcY,1967
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE,710
jax/dlpack.py,sha256=7YqIO8PDBqaW5vUtr0FYgpt6LO3IR8uDAeG5vyg0nqM,1776
jax/dtypes.py,sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I,1188
jax/errors.py,sha256=LCZJqNAUhGcqPQYd1yA600R97ayA7lunuB_lu4zf0iY,1794
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/ffi.py,sha256=S4EjUxeohrl-Kmmnu8rykPCTxMl2LAxGdaerWLhd5ac,1052
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/monitoring.py,sha256=xWZF_CyFg7Vv7MYmIyb3JWNA-AwsvIm9oc-MbGX4Bbw,1415
jax/profiler.py,sha256=_aALYj1kkwBxGCHmiIzcdafpIBoJV2M5Kqw7VBNXQME,1387
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=-1oFQ0v1HFzwvFTipdl1JQVqsk8dXRJe2ZH4xTyU51g,9434
jax/sharding.py,sha256=gEdFNr-gSJdaYEn8BMBPrMtRQvIQQmYBFJe9CpBIyMU,1898
jax/stages.py,sha256=KOQqCPRCnLKryQUrTRPVRiNuZqyV4chGi_oY2t3HpQ4,1286
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tree.py,sha256=UMACxhh3xfcgrO6h9QE00cqAl4gMzyRkweFfBxPnlAQ,1152
jax/tree_util.py,sha256=iRmZsGa44S8xWIzqv8KXvdUfX04FgnU2rBV5SnQ6pac,3661
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/util.py,sha256=HGjj7010lZQHbIpx-qvh4RnJFXgH_Ds3PF-h2jVtyt8,3221
jax/version.py,sha256=r5-Qose9Wu1aHA8_SYapxwCKAkAFjrtxJj80IYZWQcI,6725
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=1kxQA1qnwmri9_GeZHerQmghGZ-zHOy-etSyGHIhTIE,2889
jax/_src/ad_checkpoint.py,sha256=gA34Ejz1cJLWedUg0V14tuEMvedMK8Mi6pyMZpKsPhg,37948
jax/_src/ad_util.py,sha256=i1Oj_yGWZXcPb33GVukSYkLskVN3L4Kt0AUZ2F-i8wc,4446
jax/_src/api.py,sha256=E3fp7QkNJoLU2mxH1zIdoXG8F4_VH3qb3KRA1N6jUIY,130758
jax/_src/api_util.py,sha256=AnOPgETxh47DgwF1_8IDt6ODA2bucs5yPnP5lyCTdpY,29979
jax/_src/array.py,sha256=ig7B7haTAvrowY5fdyESewplIaYfpAdIH_8ZMfpAr6U,49857
jax/_src/basearray.py,sha256=FQOcRwUhh1w4byHijEzCZAVqa8Tukr3bSr4fEqZz1q4,6904
jax/_src/basearray.pyi,sha256=IG5GkrrMFSIphXGJCngtgP_gCHf6boXBUIoZy5rDTxo,13980
jax/_src/blocked_sampler.py,sha256=mxHv2Eec-AXzPuFe4YK7EN0AVTzmNrWFSD4Ph01ocUs,5890
jax/_src/buffer_callback.py,sha256=1qpEUw7Lku1JrNIU-vKW36oYO47aPkAxmJPyt7YmJog,10533
jax/_src/cache_key.py,sha256=Yon7NspNxWKIm0yBJso5j6I4Z6mnKFLmAIg_g0rc228,12923
jax/_src/callback.py,sha256=0f0GGg9_Cf3HC3cEaPYJCPMluZt7UPvs_YnsUQ8XntI,33433
jax/_src/checkify.py,sha256=-7U9mXOm0b5fi6ca2JEp-wal4JZ2qp-cut96JFFwiUI,57943
jax/_src/cloud_tpu_init.py,sha256=Ks9-0g8s45dOGPs3cHVa7Ddjy-Kqlbag9JAFBWyPlr0,4875
jax/_src/compilation_cache.py,sha256=WfrNBOZSpIyxrKFIzdgq7pkzekYy_Nxa_eFY6CttL5w,12006
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=JgUM0uoDCIcJBsgnPXLjMkgM5iBKKmPoj1JcpGbhmrw,32817
jax/_src/compute_on.py,sha256=0b74TSV1R31kJTR53thafdtB0sYiFGqmRoAbZwAFcps,1911
jax/_src/config.py,sha256=6N4FQKP29J7XKzysyGS-o34lpFBsAX9txXeipXrqPnU,72499
jax/_src/core.py,sha256=Wulzmz-664pGyUFcyn6vmSzbay3Q9U1qQ2pFwCyLb40,139300
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=KkZKiuhmWPm_5VoTb6MawxP0wJHA3d3ftoRRSDkLgls,15236
jax/_src/custom_dce.py,sha256=UHUTZ9UZM4EyeTBrHDw3xp7RgPqeeoSb5d1lRLKIoUY,17470
jax/_src/custom_derivatives.py,sha256=Cjq20O1uC6XHxNDxAuGLznQ4UkWYgcCEO3ZHHy9gkWw,80122
jax/_src/custom_partitioning.py,sha256=hEZutF-EN8tA6jwXATMosJg12duXiBQThteGzTWAM-E,26178
jax/_src/custom_partitioning_sharding_rule.py,sha256=qijmSIw4pKpyqyTb_mzTMYCnDHV3eMb-yzixrLUi7XY,18026
jax/_src/custom_transpose.py,sha256=sg_Rg3-cY6yj5Wj7fwrFxlLJPHerH7x950A8yCaJurE,9202
jax/_src/debugging.py,sha256=k52KwyXFtd7WQ1ZHELXxfzZL9Btqlb4nUf95n9vqFDk,29496
jax/_src/deprecations.py,sha256=vBQeRe1h_4Okm3lJV5Xpu5yFSb9Z-nLX8Ieh5c7Yypg,4795
jax/_src/dispatch.py,sha256=sA_HYgleVrolZObDwlLzY9PyDRSOnEqT5FMRjeDVcKc,27043
jax/_src/distributed.py,sha256=NNv_Di9-xUalZES0rJm-ctwLL79A2Yg6ozFn_y00ch0,14110
jax/_src/dlpack.py,sha256=bxRkfVUf6ppc4TULw6SorGDvPVnaeRhrkxd6pXKjEp8,11978
jax/_src/dtypes.py,sha256=UbhVjYTdMf0iqIbMUqGH2ylCQ2ukC577gTcS-WdsmsI,35661
jax/_src/earray.py,sha256=6KsYjNLzWaT1S0IKDwmZMAsa94E8yg9TU2Yb3-2QN70,4412
jax/_src/effects.py,sha256=qQCTznjR8_76HVCp_MAAKKXbs1HvIspt1Z3w3UYS5CA,5041
jax/_src/environment_info.py,sha256=XWio-o7xHNIOFxQu4dVNPJ4BDWoerRcBUVAnU8C7XgY,1996
jax/_src/error_check.py,sha256=Heko9U_uF8hiYndm8yu4P24Qvu4i4LRhqTdqO4IoSBE,12977
jax/_src/errors.py,sha256=Vp2gp5kQxtoss9aby-vpGwO5wBxRndvddVD_3-UWyZ4,24710
jax/_src/ffi.py,sha256=YkzA8FEsj1iE61k8bq4x7xAlLnhl9m3aLBoNjUAAdXE,29275
jax/_src/flatten_util.py,sha256=48xt4AJcMToiJVEoe2OzY_yxmLG-IayxEttdAkxp0Mk,3482
jax/_src/frozen_dict.py,sha256=GKsfqbGoJ46Lakb3gFGKl1sIGycFVA6RuvNh__Pv4hI,1369
jax/_src/hardware_utils.py,sha256=YQcgkyPgQN94grS6VvBIVVFL2R-LDCsdY8Tqj43jnVA,2349
jax/_src/hashable_array.py,sha256=4ff9eVDYR7DFK3l9nNF0btGecguXonn1AzeWviiJQcQ,1036
jax/_src/jaxpr_util.py,sha256=TOdYMOrOMa5Ei4WnawyfuH7nDdZ25Dt6TkpBSzUIXOI,10475
jax/_src/lax_reference.py,sha256=nBwzDRkyR5C6W83o27qxBrdpLT3rHtPB5olm_smwN78,19279
jax/_src/layout.py,sha256=zCZmpCr9euOjcjkGK7G-Yq7fiKqau9tbbBTwT_HDNfE,5006
jax/_src/lazy_loader.py,sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw,1842
jax/_src/linear_util.py,sha256=36c5Fk-oVwTFxFqgBjbNc0aNUGIZs3cc1ZzH5hroT5c,18971
jax/_src/logging_config.py,sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk,4452
jax/_src/lru_cache.py,sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs,6741
jax/_src/mesh.py,sha256=syDV-8uWx5NK1jpQRLd31D6ZUKkU_J4QJfV_NFBjdOE,19329
jax/_src/mesh_utils.py,sha256=xO5jPwRWuPUYp3jjOq78p1RU_o9V6FLu0B4H3J0Nq0Q,32170
jax/_src/monitoring.py,sha256=PSOb1js-21Hb4INsIdiJpm0s483xeljnoRr0Lha4nzM,6241
jax/_src/named_sharding.py,sha256=u6-vNjyeY-VEXOWvY67x6OeKgWZhVc93_K4LEjBqBMk,21121
jax/_src/op_shardings.py,sha256=nRnGgfaKz7mIB7sHxh_QkiCa1oPMA9VsDpPyQkivtlU,3857
jax/_src/partition_spec.py,sha256=7dBKnj-ONLW7kVIWjnMag-_8fIUIXPUGZ4pnF8f_Slk,6220
jax/_src/path.py,sha256=d9R4bFgk3cHS37ZUO8up_dHW10Vz3oR2NZTaRM28gdc,2162
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=9RYWF7nPV26hiiOmW1reyom7HEdTp2El2SGVlcX7L6A,140702
jax/_src/pretty_printer.py,sha256=1tSAtbJj6N9a0ZjwqphDD-kPPolsYxRp8JM_WYkrUXI,17578
jax/_src/prng.py,sha256=6oTZmOsqID24YcciRdgz10ZVwVuCY7mJM4cIQsKysCs,46718
jax/_src/profiler.py,sha256=zPgtTULdZBFGT629usNmQq82mtAfRjd77Hssq2Xsf3Q,17282
jax/_src/public_test_util.py,sha256=iuO-3tHEx9M5XEfULXiMdy1KM8ro31Idz96YSS3EQnY,13087
jax/_src/random.py,sha256=br71zawuS0FH2eoE9bS4YRzSPBLhg64urjUy38OLltQ,109248
jax/_src/shard_alike.py,sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg,3989
jax/_src/shard_map.py,sha256=zzoDFUA6K4lTMHs6qe_ctUzXUPB16zd89Lzpb6TWMd8,87673
jax/_src/sharding.py,sha256=mtNHNhrrbQdmHEARTzFBtdeE0WXESc-4Ru0WcXXxPXE,7843
jax/_src/sharding_impls.py,sha256=K2pHGag7_LYdJBS8z8D1a7gDvF_ZWgFfCP_xY86PVZk,47947
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=RLrfyRjQFHSya3X72qQf-19Ies8i5Fk58ekWR8nCrgc,10471
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=aqRTSM-RYZjFPvPHPMl7AeL-lD7gjmomRhp99kafmx8,32549
jax/_src/test_loader.py,sha256=l7LDHdnqD5g44nUo4Hdwtc8IplXjCcn-L9-RO4Q_qhs,7260
jax/_src/test_multiprocess.py,sha256=xB8pU-uGDFFXd-cWEXxAgtsL6uCer-UDxgYn6XHW9Lc,9279
jax/_src/test_util.py,sha256=xv855qH1fsxwGyDiTHv_awrV_DlEnirRCVErLrdAa-A,81579
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
jax/_src/tpu_custom_call.py,sha256=ulCXI6_1h0FQ7viaV_xGdHdtnZBFb1x8AulLVNAmUaM,26188
jax/_src/traceback_util.py,sha256=ywejARaYxtWnK-Oxsi-MdgZiUYflvU3UtujpBq8MGbo,8135
jax/_src/tree.py,sha256=XjxdG9NXDl3MQB6AcBSfYyybrIOxZMJMRtsKIF_h_TE,15319
jax/_src/tree_util.py,sha256=L4RKGQ-hogn1K3THsHY_loBYgH_-RaPPe3THLLF7AhQ,48088
jax/_src/typing.py,sha256=Ed5ujNHsKktTk0Z2nhZojS_77K1INnsPvj4S7o0KQ5Q,3529
jax/_src/util.py,sha256=GJ33wirP3uljts2x1FP6H0LLXhyavJ0r0pV8YsAhizM,21132
jax/_src/xla_bridge.py,sha256=py9pdESthog4LYs3dWWIkZiZGL97XpnP2sHRoC3O-Lw,45822
jax/_src/xla_metadata.py,sha256=48mF2jQYDW4lf5muzE3he5y-mM1sV24HQ0Kq9O143wc,3834
jax/_src/xla_metadata_lib.py,sha256=gyRBCCpMfUG-o51PZQ9dO_rafReE7Py_yZOs6cHSY_Q,1474
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
jax/_src/clusters/cloud_tpu_cluster.py,sha256=mlbAUcQjIdZ54du59mMq7kLDTcAOfW_CO56za0zXJNc,8488
jax/_src/clusters/cluster.py,sha256=raW-A1TKgHDyqZeMfv3lgdm4ImDHxrzoMnFAOfgiK4M,5858
jax/_src/clusters/k8s_cluster.py,sha256=uo4QCCwpqixEB3Mo_28fPmPpCkze-fQb1PK8I_WIGy4,9279
jax/_src/clusters/mpi4py_cluster.py,sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg,2608
jax/_src/clusters/ompi_cluster.py,sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ,2317
jax/_src/clusters/slurm_cluster.py,sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0,2321
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=KrOqE_igOtrfuUtssB8OrBrjnyV8xaY56jr3Rz7gtpE,82122
jax/_src/cudnn/fusion.py,sha256=Aob-4WLRssNMbS3vdDJuhDjKPDshEEfu5VAYjL6b97w,2877
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=w4lv-XsJGRiO3DiNxw1I9SgA6fNk_x9dKZlSF6W65Rk,25236
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/cli_debugger.py,sha256=bzn-5Rd4C9WwGsVret8gbFNaLpXAf0SzvgltXNuZrcU,4778
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=sRN6dv-Vhuz6j5eenDq8s1ZTUbxYiOv-JxDWx9519ew,8682
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/_export.py,sha256=cYewhNM8jfbRZOlNiaXfOd5OXwxDQ9urjK3LfSNBSoE,72318
jax/_src/export/serialization.py,sha256=0fYNO50OjU8TWRC1_wHJm_z1pFbLnaGQai3Tif5KluU,19466
jax/_src/export/serialization_generated.py,sha256=Bl505_cl_U9oGOUNN0rF6sL7hlZduTzY_nc9-ekd0J0,27781
jax/_src/export/shape_poly.py,sha256=X5d2DINA9cvefSrWmpOkc9EeSQaWfP8GYZz3Zhvfpdg,85962
jax/_src/export/shape_poly_decision.py,sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc,20476
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw,1258
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=C1q4vSC4jODLT44jl6xJcbvAeTY4wDMGwkNeTE_2rAE,13880
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=ZAKRxmSNHO4SOnBfAN5aV6Fz4pSFqyrgRw2tchOd0mY,15680
jax/_src/internal_test_util/lax_test_util.py,sha256=_83LJKAcbyDkSrT2D2k4x1waaBHa37JMNISX6veyGHU,12998
jax/_src/internal_test_util/test_harnesses.py,sha256=UEhlzGVCyxT5DFKr2aQAeEgM0D5F8RIBnrXxlAFVafU,126312
jax/_src/internal_test_util/export_back_compat_test_data/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/_src/internal_test_util/export_back_compat_test_data/annotate_data_placement.py,sha256=EpoTASRU-bjg8-0oLRNZ_Ir80p2frmvTFY1tKodKMsY,27102
jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.py,sha256=liVPn-gYqIyzFzi4y3yAeURsUH4l71cC4Ci7rs2EKwU,45965
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.py,sha256=6SvL_1_UGaUkKcqXX5O5EU-D_KJxckHz5q_5sFmLDAk,35800
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.py,sha256=4OHoOpk2znnttnKIwfdVVnPXloC0jdJcDDZRzzkjFKE,58741
jax/_src/internal_test_util/export_back_compat_test_data/cpu_hessenberg_lapack_gehrd.py,sha256=cMbJDAZ_vdGK5g4hgdfBBmyxf3Dzk4jyOnHZGC0lZ1E,37276
jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.py,sha256=W7sTmxVhYK3bPUY2OQVyz03O-5tNKztupdaCPz0GGcs,80518
jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.py,sha256=gCjyU4yeampoNYPTwvXLBeTaK99gkCNaj4tIfOC5X8w,28588
jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.py,sha256=yQYdzlYdLC4saeV4pPspeyLI91YKzbhFWG83s2txLEA,25050
jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.py,sha256=oFLAqsyRQQ5dYvn3qYSWAYkr7sVdxaGUtV21x5SSUMI,44276
jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.py,sha256=LvDY70AYYAWa4ZlFkv1M3vLBVcrb5vHzh-Hw9JaSlYg,19086
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_lapack_sytrd_hetrd.py,sha256=m-93hKd0NonVaA-0fomLu5kiWsbC_Ijfu9K1PKh5dxU,58875
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_solve_lapack_gtsv.py,sha256=oQI9RA2tVCe6iKkF-HwQvgK1i2Z7vJdtdRneDfxcYlw,31786
jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.py,sha256=i37_O3sNPXdJhfziBGIXSefjxGY8O49BBgUsriD8l4c,43043
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_cusolver_getrf.py,sha256=rTViiEOXgqRovwdxMUNTIMvrC4VnEJuBsFqnBf34boc,28728
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_pivots_to_permutation.py,sha256=80gav7QLSABoktOIImerdCeOfMx6H4PQnnkMrEqFovg,4159
jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.py,sha256=MxhvzJkTFtRxSiSexJ37XOqdypQsCa4s2WVIasNrGR8,31584
jax/_src/internal_test_util/export_back_compat_test_data/cuda_svd_cusolver_gesvd.py,sha256=uowHFqyj-fXAfNcJXz9TvUyXHXr8IZDLHJv-SffSfSE,80048
jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.py,sha256=jlXolWJmf_pjGdr8M_N6q6D3aL2NL12x4huNqCYyiyM,14920
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_cusolver_sytrd.py,sha256=0nv_mDOGjjbwhrQUxeCMTsjLn5pAzPuyYOwSRQPZrz8,39049
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_solve.py,sha256=jO7yNzENuubOaabB0441suDSOeU7pLKQmynfmpJ1ueQ,7756
jax/_src/internal_test_util/export_back_compat_test_data/rocm_eigh_hipsolver_syev.py,sha256=Ei1LQm0kkFU_zbsN_lsHr65zv7k62PsEWXYSQKq7Xf4,115014
jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.py,sha256=mKPsDNmGRMCdbkeMQRo59H6aDbBPtmugY1KJiFl7aU8,22351
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_approx_top_k.py,sha256=2IgM2d_n2RRgg7dfUmhryKScHmUPArE05atL9_HKw0E,13062
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.py,sha256=gtPimKyp7-w-_Zt2xxKbPMy7j4ZFfoRERF22PSphU7k,11422
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.py,sha256=Zac1QqMUUsnUHgWTqD3a2Djoh4-oRA5H6o0AbQU69A0,14167
jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.py,sha256=MzL1cZYrKGODHjmPRYlc5iKfa52Z0pLsxSOl5jtTVQ4,12559
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.py,sha256=1D7yFSMn4-_OiO7RE0bHjSBQ8_0uGBmc49NLqy4rlbk,10146
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.py,sha256=hPRVhBpIWrqs4lWdG-5Jhax021wrA3QlcOv0_Mwh9HA,6185
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.py,sha256=L4zVrUCoSpOEqYo21litnykOdd07tbBA4EpfrSV_mdE,10228
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.py,sha256=8o3QlFx2o1ynfxI_v0BvNOCzPk6nXXB5apQlR4bc9ok,20023
jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.py,sha256=r3tKmpt6kpO0N5xJ0iDS61JiAFn5NV6KXXCTz8HTVIw,22996
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.py,sha256=gFgTn32nAZGHaatj606pjLgIv5i6p9vn9yqQkSunf70,102068
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_matmul.py,sha256=3zMrunLj4r2sskF-UJCezccCMR-j8vrfGeg-aFzlsXQ,31203
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_semaphore_dma.py,sha256=7U13cnQrTs33Bc0TA-I5yRKlHXYN8CEkEedBBe4w0ms,16908
jax/_src/internal_test_util/export_back_compat_test_data/pallas/triton_add_one.py,sha256=sgk-uKJO5c2-2KO_mO5mrZHWzBhCruM5MIRbbnhva-w,10043
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=-tINa92fy8diQ_G-EoD-lTJmLde0gwKSLme0ixMOlgM,60477
jax/_src/interpreters/batching.py,sha256=ABj9xoJq74TE9EC3fa0ioPh2lTod_jrUbdygn273eaA,52544
jax/_src/interpreters/mlir.py,sha256=uOzyDiDD-zVmjStE9o6l_3fq74zzVIUJODQQtOJN3DQ,133652
jax/_src/interpreters/partial_eval.py,sha256=QED0dtfezApjlQ6WvK2-9_5N4XhDHQ2VRg7y6vwIpW8,121752
jax/_src/interpreters/pxla.py,sha256=De07_gZ5asxq7nXoEelbxHPSEAhUnfdhvIO8iE5m-tY,137457
jax/_src/interpreters/xla.py,sha256=Me2RiFbqW_hF-RUOBNxQEGWkPhqYfYjH-12eAQbJHb0,4981
jax/_src/lax/__init__.py,sha256=SViklzZ49a1SKhmqn0FdgTOQvT8Svari9dx0FW-H0uk,7448
jax/_src/lax/ann.py,sha256=drKiFUIxy8wodSZxGqpYd91Lq_M-WFyqM4NSSn1DqGc,16786
jax/_src/lax/convolution.py,sha256=oWHCmaWAMiMnuIGSSVm9-tJlmV6ulG9qBcNAQTOLcNE,45955
jax/_src/lax/fft.py,sha256=3jloTWfLPoPvHbUfHde9Dqt-E96D5UXDmqgBxPEAsOg,7277
jax/_src/lax/lax.py,sha256=yKQtd2BhuNlAOXOyMJrfuKmQCIOwZ55Blga1GTQrEmc,359341
jax/_src/lax/linalg.py,sha256=M1DsFUnChrZLch3F34gfaqnGIDayes8vvz4Sxnm8TL0,101729
jax/_src/lax/other.py,sha256=B11uj0fddkstQoi6lA3-gK2k7y4Bf01r3BOkZRB_eps,13874
jax/_src/lax/parallel.py,sha256=sGoIBGu1HxkXSI5NF7du6loKPFEHpuQtM1AcL24nPKc,98872
jax/_src/lax/slicing.py,sha256=n2yKU3TQjsQn4tzcrNHfyU7x5UjFzRuTCAL31cDTEJE,142979
jax/_src/lax/special.py,sha256=UJMNxFGBsYN0v8nmSTEuDewye5WCewahr4mSzv5Zbz8,28862
jax/_src/lax/utils.py,sha256=WY8fd7RcOAv0BVpqHOe9KeaRBxQyBSgVt9L7zy8NHy8,8430
jax/_src/lax/windowed_reductions.py,sha256=U8NXJyfWYxXktFDSAfWp0MH37x8WVXWxAUGm0BJOr_U,47069
jax/_src/lax/control_flow/__init__.py,sha256=WRnqzy4ec591tqipY4GcyXFzniF0njoIVkP0HhKGbiI,2244
jax/_src/lax/control_flow/common.py,sha256=WjX6gD_Rv76SGVz2yHNEvUWO2teYeSImk4wJBiX92BY,7427
jax/_src/lax/control_flow/conditionals.py,sha256=3TOgNSTgKKBK74BLki7-9XwDYl3QxGwhEclQPJ_9iVg,50966
jax/_src/lax/control_flow/for_loop.py,sha256=wHObwBG9sJsmnKkXmyB-EW5tyR92zTnww9-_fdF5NJY,36638
jax/_src/lax/control_flow/loops.py,sha256=mcgrRLnDGA-OwmQP8RVRcEU95bZXZDqsVRrMfVb242I,130591
jax/_src/lax/control_flow/solves.py,sha256=uReVXR_NMwrKhjnoBK8CkQ0NhVR8ZmH4l55sxRvWdsE,20489
jax/_src/lib/__init__.py,sha256=9SQrQyIM-S7iHFQU_qGgj9rs5I5xmtQeWfMRTuldkd8,7448
jax/_src/lib/mosaic_gpu.py,sha256=andYvo1TC8Eru9Pd_rM76L66Ozami3Q5awjdTkdIE10,1017
jax/_src/lib/triton.py,sha256=lMwAguUTQxYs9HkdLqeracg8sz5RL7gm1W6PpzSrzdk,2202
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
jax/_src/lib/mlir/dialects/__init__.py,sha256=AmPoWPMMC7cOKxh68U9nEoIbL7FuiDh-5jOsf6_w57o,1965
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=ewBi8d9_o8S4c5JsIyWnmHbQ-L_dN8WJ7n5pHamEhsc,47036
jax/_src/nn/initializers.py,sha256=ATJjVrlb0afYqUe0T2SysJWN6JWUYFMCRGDezNsIMJg,26802
jax/_src/numpy/__init__.py,sha256=l65yH0eVP2FRBXmP0IYlxZ8BG3HiSHLfL4HNevyXJN8,3978
jax/_src/numpy/array_api_metadata.py,sha256=MVvoctyTPUK3Ac66PmSMWkcHZI3Km7LsAL-2sC6N3tc,3686
jax/_src/numpy/array_constructors.py,sha256=JXpPAGsiObYpb1wTWZbl_Sd-SPvxCFVR3PK5bgklp9M,14823
jax/_src/numpy/array_creation.py,sha256=G39BOP6BjSiCUC_gKTxq_1HIGBSgEZRcKaRX6dNBY6E,30527
jax/_src/numpy/array_methods.py,sha256=W9FFpifRwQyTTsZFHnJtzKnAnD62XtYLVjZrV4PVrZQ,50078
jax/_src/numpy/einsum.py,sha256=oISTX7NnVc0Yg4zN2JwhNJV4H92vfjmmMAt0Vtudrow,23705
jax/_src/numpy/error.py,sha256=WwhoOFBsaCETiEzJClflyD-MIm0edzQnPUbW_iDecgI,6392
jax/_src/numpy/fft.py,sha256=OBVai6FqtarGy_n2M82cbD-gRQE52O17dHH0lpYdQjE,52345
jax/_src/numpy/index_tricks.py,sha256=ko4vrQ8VxIm6QP97ppiGxKF4rsxq3DCh703a0DMa-yo,10101
jax/_src/numpy/indexing.py,sha256=o7dKbEwZIzYnkDGWupioC_w1El5_HzEr3rLJZc0o1oM,52221
jax/_src/numpy/lax_numpy.py,sha256=bcnSBottlVVp-v_XGOhhqLHiBAlpel30NAKE0iireBQ,336281
jax/_src/numpy/linalg.py,sha256=XS4z_mDXK0nZwtlXITfjXA2o8fCW2oiBrQtDoOquFyk,79425
jax/_src/numpy/polynomial.py,sha256=0ksPTmOu63XtFTJKRHg2XdT5BKD0KkpuchTUbgtK9ho,28500
jax/_src/numpy/reductions.py,sha256=oTmpPXJBw2Qio4Q6aiUxil6KLWJC4hJVTKKirUw3lec,109357
jax/_src/numpy/scalar_types.py,sha256=92ODM21nG1eeTvmWpMikTU1dCFC-DTcV_gZfPMX-b04,3836
jax/_src/numpy/setops.py,sha256=SIh5WY6o6AaKB2VarCu_LudqT1tDwjzzmSe3ztbvjTM,47926
jax/_src/numpy/sorting.py,sha256=7TwE5OFdllw5A9CszQ8NYLq5tsv4zz5CdNXr7hY1mgE,16249
jax/_src/numpy/tensor_contractions.py,sha256=ErJWjXRGlTGTNGGHukfAwszA27UEYCKVlwpJCczXEUg,24522
jax/_src/numpy/ufunc_api.py,sha256=Fxq_tkENp7gyKCxla7Kapp-SGYX9Qdf20tJFGRi1rnY,24376
jax/_src/numpy/ufuncs.py,sha256=L4nWL2e3f7kYWautK29Q6LHRw4WGN9hLq-kbiioqTgA,118733
jax/_src/numpy/util.py,sha256=5Pt8rcYXDPtCsOp6R3O-e1RCFK-7wH85RGSoMZnsN_0,18157
jax/_src/numpy/vectorize.py,sha256=thLB2fBHVQBas7oaVWSsaOhtLgVWRwQekq_L5F7IjAA,14077
jax/_src/numpy/window_functions.py,sha256=-AlJAgIqxJXMiXbIupk9XFC3DJljUF_BYVYIygR0k9I,5533
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=oclihVexy0bHYsDPshbACsCVBVikYvIVPuJI3AFDd88,18424
jax/_src/ops/special.py,sha256=tre_zo2RVwTeXpmR1VZ2zQT2sizCiIY0CaSQ4vi-t8c,4337
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=P9WLDLqkWSf1fYyDSgYRtT-U3njRrrD-jcMFWD-geTs,49453
jax/_src/pallas/cost_estimate.py,sha256=DJnl0aNbAtHErpMRcyojeHsCxlu5DthB33bah6T3BI0,8338
jax/_src/pallas/helpers.py,sha256=0_QjMseMMs5cegiV5U8ucQ0LDAt7dLMwRfqfVo5vkTI,3550
jax/_src/pallas/hlo_interpreter.py,sha256=B71ICSEt2dpqA5uMMJvHmBpPAeN_VeeneiIRFdHVH18,19185
jax/_src/pallas/pallas_call.py,sha256=BRIkRMw2FLokeQPQrPPMfkvxSFxM4vbzPeTzjtqLNoQ,70179
jax/_src/pallas/primitives.py,sha256=-gaTQn1iF0nbvskWw8QCO7o_N7ZAHOG18_MPeT_fpeE,43806
jax/_src/pallas/utils.py,sha256=lZUH6NbcPvS3Hsr_n_ZK_ao5Jn2Sqa84pd4llzcD6sQ,13896
jax/_src/pallas/fuser/__init__.py,sha256=cHEWioEa-43e-714tP3fFN9hdyZg_ebBFbFJur7bXwg,1182
jax/_src/pallas/fuser/block_spec.py,sha256=uH5NOCYLQdaFlBQlbUI-FqBsA2aYpIUl6_f-f6Zk844,70147
jax/_src/pallas/fuser/custom_evaluate.py,sha256=jv5JjL5BlUUK89ZKx9sufff0bVyqJA3s5ag6Bpus5ys,2823
jax/_src/pallas/fuser/fuser_utils.py,sha256=2crsWld47NLWESj0R4sQnxjTB6yaYO-fCfamR8NEbCE,1270
jax/_src/pallas/fuser/fusible.py,sha256=k0CaPyV321MFuP8St6H1bB-fOiZvemByzEXbDD5lbi4,2689
jax/_src/pallas/fuser/fusible_dtype.py,sha256=GiXYYkFPDnxHYCdvhrLkMcG0N0Q39SmRFI03n7QOuyQ,16440
jax/_src/pallas/fuser/fusion.py,sha256=AvfY5PSPzoAT5Vu6NpMHPqobSgNfvv_600VbWIlcslI,1515
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=vB_rWH0XA1L-45p3Q22be2K40qiIA9jd6tQfTinA4gk,11061
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=7fR2HqEdtC8kSZ4BL9TY50i0foPwSz43g6l37Nsw3fg,10715
jax/_src/pallas/mosaic/error_handling.py,sha256=FCbB-b5Dl75erMmr-FMvc4QJwCXqn2z7ioNC2Eh0vQc,5542
jax/_src/pallas/mosaic/helpers.py,sha256=DFBL7RuFtfBQc-aUaC4bILa5Xzsbb91Wu3VCn7b3v4M,2767
jax/_src/pallas/mosaic/interpret.py,sha256=rwMJiVCQFoLkqsHbFejGRw7Aly8xgAHyYVmHJtB2pUg,95488
jax/_src/pallas/mosaic/lowering.py,sha256=JORMrV5PscimtK-rGrNyd1DzcniOlhsVje_iDwKBYtI,144189
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=V8Os_M_JrYmdo5I76hqZxpdJ6oFT7UgSi5Vz7IOhlYs,10422
jax/_src/pallas/mosaic/pipeline.py,sha256=N5ZDS0PGzyWhpcaL1M2XTOIYRisRMYCcyR3EQKhOcyQ,82700
jax/_src/pallas/mosaic/primitives.py,sha256=vGohcEnvhgPSnYz7RpLKqsxgtxxs0ryiulJjO1qHXLc,29225
jax/_src/pallas/mosaic/random.py,sha256=l402QtsGJkvPQ9UY_o9XNWsL58_YXVDBxaGxNhFTLWo,7845
jax/_src/pallas/mosaic/verification.py,sha256=3k5rJTC5n26T3EUj_R7a5A9ppgvaDKMknTDymA-tABc,24512
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/core.py,sha256=INgk9JNOJTZTwXf2TqTpxsJgzOALZUhEznCVboL4pr0,45021
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=DOkDJ7shcKrNau9KgJ8Kt_ceUGE4lkFCEEpqaZjUE-k,2854
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=Giilj3lxeIYMIr4-clomVOYQ7mJ7XUNbEG6ho7SPeec,122988
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=b--wvQy0o5EQ1pBogoObOFCIgi_bxbXVIvG4QbWlt5U,5776
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=nLZbJgb6W1ytfj8mPJdCa9AyMG0gfgL5oyPO93AhLwY,29303
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=kklXXaS-p863q-5W8OIJUwoIlUOv6VAWdTPVg4iiVrQ,99271
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/core.py,sha256=Z4Vk2LFEpkvRzhRBfZlvEGRhFHoxnjuPxC3EfgMPONs,1222
jax/_src/pallas/triton/lowering.py,sha256=oXLHubIXzFIhK7VT8nOpLpt6QY1Xz3RSNeH-W6ctVA4,98772
jax/_src/pallas/triton/pallas_call_registration.py,sha256=SHD6MMwZ_45UwJcYqwVnYzC-Of2reeNLXVF1TxDNe2I,6892
jax/_src/pallas/triton/primitives.py,sha256=h_nAb5_WTDgtna0a5VDKql4hDx8I2gt9ryuY5LsQo9A,4235
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=HQhZQPNr1lfUbfQPA5mWnA7mQHZBeh-h2YClf0zbqJo,15357
jax/_src/scipy/integrate.py,sha256=2eo0ZUM3qdh8w9W4PJ1cKqZNCcN_fhFancTKjqu7gq8,2374
jax/_src/scipy/linalg.py,sha256=7ETV6uxshB29fculn1Wm3zygP-yJD05aUfRqDiV1e7w,79654
jax/_src/scipy/ndimage.py,sha256=fQRx29EIsv6BIRYFHh1ZoJjBYqxaYSb7FV340-9L3Wc,7161
jax/_src/scipy/signal.py,sha256=XAAx6rK9Qsv7sYt1u51kVhdH-NE7vhMfhdX14Ch4ByE,47429
jax/_src/scipy/special.py,sha256=ABFeua-VXMbwBmxTMTtZsiv0GhKMa2giY-nFsuYyUeE,90548
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=LPoQUfgmh50I-ooN43p7vBiucOgJ29FIji9ZFbFikOc,3039
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/_lbfgs.py,sha256=UlpYZmDI950InTMDGdfaPtMcplN9uaYpqmjeYaG_bA8,7874
jax/_src/scipy/optimize/bfgs.py,sha256=RmbR6n91EhFgkxJ9Oe-HB1_0RdvD5voG6ZMMOnWoFvQ,5733
jax/_src/scipy/optimize/line_search.py,sha256=1sEzt6U8YWoFeOmGy6tn4ZdpI8I-D6b42h5_0-QZHEs,13184
jax/_src/scipy/optimize/minimize.py,sha256=vYoSJVP6ijGz0nyjjuwk-dHVFMlJ6MOeDiCePaFoLig,4954
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/linalg.py,sha256=a2ENNIyq4SnTS5wkGwu4M9p-6LnEBqqeScmaw25fXs0,27379
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/transform.py,sha256=a10DkF7vLkS4BMm86swDwoqdV-xDtRjCNfho965M5fk,17634
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=ok58tDEmsRavumLXU0vRWmcwa7IPVRrBDR_nBTtOzf8,10678
jax/_src/scipy/stats/bernoulli.py,sha256=HT89cwug-OktD0KVKDWrpihz4GB_hEOU_R567Q_8whw,4578
jax/_src/scipy/stats/beta.py,sha256=jjRE4LlyBt2BO2jmmN1z5PEasjl4LykhKTNc8846etU,8026
jax/_src/scipy/stats/betabinom.py,sha256=LceEdt1jRBxYmV9UMjZBJs7gR1GAcGcU4JbOcF2iEiU,3480
jax/_src/scipy/stats/binom.py,sha256=KIydYZOcwN109i6rodNR-Lda-cog16r2jrQcDarUeeo,2859
jax/_src/scipy/stats/cauchy.py,sha256=BnKgESULqAHQZydTKMxT4oRexDWbMyHWj37CFROaar4,8901
jax/_src/scipy/stats/chi2.py,sha256=p3TVmUKKRscb-7-uBUB2aorLFdh2USag5iRt1P7yr7U,8045
jax/_src/scipy/stats/dirichlet.py,sha256=G9FAdSoODRw-lZkZlGsg1sq2bbqMLZ47wG8P4o_oaow,3237
jax/_src/scipy/stats/expon.py,sha256=NcqNEDneSnAtba3Yol0XyfNZ8_OIwcvTih2NZLE8OoY,7845
jax/_src/scipy/stats/gamma.py,sha256=876SXGzsazCTNxGXL38_PqF0MbFZ9k1XOTcXDhUJdS4,7076
jax/_src/scipy/stats/gennorm.py,sha256=s-Wo42aMKpE1EzeqC-2Vx_uH-zDg8qWjdUAe2d5azr0,2994
jax/_src/scipy/stats/geom.py,sha256=A5XokLcEPUlAjhBh-0BF9PoqGGZmE5EkkT3xT3R61Fc,2282
jax/_src/scipy/stats/gumbel_l.py,sha256=PR7XwEY_RAxKLCLJIEq3QTqzdvAv-ufQl0QgRkFh65Q,8345
jax/_src/scipy/stats/gumbel_r.py,sha256=f4pjK4sYZ7NLjgAYwgN2RjsC7eeUM-eA5V5MEAnZrfI,8746
jax/_src/scipy/stats/kde.py,sha256=wgCbK8d6FFKuN6Tk3nPu7L8PZ7EYHdQ-h860QIvqtQ8,10249
jax/_src/scipy/stats/laplace.py,sha256=KRoMp7S0Gn-xTiY9A-1_Iv1lnwswGhjfSg6WDf9Oaio,3274
jax/_src/scipy/stats/logistic.py,sha256=508gMdwcsWI6ynyrD4w7VZMw9HB8K-Xyzux6oVEdKkk,6291
jax/_src/scipy/stats/multinomial.py,sha256=W22uKDVNPoF8JkQwI0TmnXPJ_7SFa3mgzUYLTJKG2ks,2532
jax/_src/scipy/stats/multivariate_normal.py,sha256=LqDX6XJxocbMlqM3HrnUHEw2ZOGjfIvXowhJmmHD1mo,3569
jax/_src/scipy/stats/nbinom.py,sha256=pF8EOulsrjJhQmq_NjMhfeQaXf5qh7wC9A6mNk8oiUU,2637
jax/_src/scipy/stats/norm.py,sha256=8YfKWloDLhnfpfcuamJlPZuqgLtwmsNKVMYT53VKKsY,8421
jax/_src/scipy/stats/pareto.py,sha256=uN0kgylc46jwnTWbyZJf6ZWxgTEzyN-R7rA7OPiHqLQ,2630
jax/_src/scipy/stats/poisson.py,sha256=EPcPC-ThmNFxw5GUDUTybSZ2TRnEYIn7yAcGb8XXZoE,3387
jax/_src/scipy/stats/t.py,sha256=-POac0PdQa3tNpgksvZaVRHYrrbQQrlyXj6l8iP15gk,3226
jax/_src/scipy/stats/truncnorm.py,sha256=LQaZJwBZ9Y3pWOABOU0gsbHD0nq9zQEQCKQzxLJx_Aw,9270
jax/_src/scipy/stats/uniform.py,sha256=e0INco1aevZnIfWfVXVXzFbt3rEOUz0UDGscr81SU94,4372
jax/_src/scipy/stats/vonmises.py,sha256=S5GpPFeRpE5BH5-MEcyenno1zxbbvOpL2Lvn23vTG-M,2484
jax/_src/scipy/stats/wrapcauchy.py,sha256=4vJw9eZn5iMgzpR3iJmuFhFq0GTCqQm3P___4UB-czQ,2401
jax/_src/state/__init__.py,sha256=TU0qQETOXKzVAd1oVaIEtOJTC2OMj1PdFhsExMHFx2g,988
jax/_src/state/discharge.py,sha256=bkTZeTnqL79khyWq4bXKi1OdPX-QFcinkiKpxC6iNvA,50869
jax/_src/state/indexing.py,sha256=UlPKL39NIhlnsKbizF_L_8PmItR97KvfPSwjT09ArhY,12040
jax/_src/state/primitives.py,sha256=9bV7ZOZTU_4R3iTmp5s2qizwDCCI4ugTDkrqLeXBM8o,32136
jax/_src/state/types.py,sha256=j53GQ8cb4kghyOAQguNEsjAypGcE14YrigHDK5gMCYk,15173
jax/_src/state/utils.py,sha256=XzwkRN0IxoK1KFHXnftBVCyD0rqhFxg-i6JtCZ91upk,4157
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/betaln.py,sha256=EW6FFB1myCHMGwDXqQXPIRaEL-UvVZNahZu_wdsixSc,2388
jax/_src/third_party/scipy/interpolate.py,sha256=XJA3BNVCwIowXBH5g3M4dsKwHpMVyHKBxL2Oy6RkbME,6360
jax/_src/third_party/scipy/linalg.py,sha256=jgDDGByArWWjetmCdX6LCdzPT2Ao43TCe56syYpBxFw,3878
jax/_src/third_party/scipy/signal_helper.py,sha256=ihye51hrdpDnsJuZ5z9scJhl1a-pEN44rCFgLQdTM_0,3569
jax/_src/third_party/scipy/special.py,sha256=Q8kobJMwgE5SiOYDLyliaHGz8Sy7RLCzFrJoCIw9eT4,9118
jax/_src/tpu/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/tpu/linalg/__init__.py,sha256=RrUPd-WnsNpraGGzWYI-IYTpNcbtQeGqaotRh0Bwwuw,779
jax/_src/tpu/linalg/eigh.py,sha256=ixc6cNXn0T0aqcx2AGraGGgKWKfZqbwlL8L6xks2m-8,24665
jax/_src/tpu/linalg/qdwh.py,sha256=tX9TiLEe0Flr7CJiRW7UNOy1g1hGkmQIiowmzCQ2Xw8,9553
jax/_src/tpu/linalg/stack.py,sha256=XPaFJtuPGfry1ssSqTNKdHcUs7yRUyzeD6aeYz0SrU8,2570
jax/_src/tpu/linalg/svd.py,sha256=vNffW6Q6CqSvNX19WEJDuo9Gu1YNdUmgQPHH3xFwIGQ,10316
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=yoE5_c5xRagzLRgNBgo2jRWg31h8Sm-dpKW58vhgjh0,1198
jax/experimental/buffer_callback.py,sha256=ocKnbOO_9n62E2olM0c2a4ep8AHTan4I_AL-eR5VbrQ,765
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_dce.py,sha256=ORRk_kBjbseQ06T_LmLPxYCO7-Rby5U8Erro5U5Gi4A,682
jax/experimental/custom_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
jax/experimental/host_callback.py,sha256=Msk8gUwv_UtJckKwsGU0lYTYGpPG2isWCJrxJ8mzb6g,1201
jax/experimental/jet.py,sha256=wGNQmyCUOqi-4y1aObnWt0HfkOCcMnHdkmFw25fYxeQ,27605
jax/experimental/layout.py,sha256=fR0lxc08SqRZ0jv9bh6QACBFt_rFP_4naq5y43pU5GQ,738
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/multihost_utils.py,sha256=oCbJ2BfjfccFPM82eZbC58sMgw_W9zIhPYBpSG7KYdU,21886
jax/experimental/ode.py,sha256=a8ZsN4gVrHXM_LPJF3T5SeC-RHjFTGUd62eA1WDdkLM,10903
jax/experimental/pjit.py,sha256=HC4rP1t3qDoM8H6OqKrA5uZiujVaAFOtNrYU0stLpLM,729
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
jax/experimental/rnn.py,sha256=q__NeQnA8uZspzC5zFU5a-593i3VOEAbrRmsiJwYEfU,20203
jax/experimental/serialize_executable.py,sha256=dO7BA1YzdxilgiAFjqS_fCJHK-J1DQ-qR9VCK6x0bsI,4237
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=zkBPC3rKW787yjzwqz0vjc_2GHpRI18CuK_ocmo5rlg,4427
jax/experimental/topologies.py,sha256=YnSqioiXZqwsX-WzVhHVUO44lAib22-TkfIzC0KRfLA,2083
jax/experimental/transfer.py,sha256=E8dFkOjF0jnpEYk6--VEdnozMEcC-ntdSG-GkKln9eI,2327
jax/experimental/x64_context.py,sha256=XXTsGH7SavrMuT1hXzThT8f04SAFWdrhjZiyfEAaqVU,2394
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/experimental/_private_mm/__init__.py,sha256=VIjw6l_CQ5-xaicoK8-2QgM5W7GYrBvv4xASfTB0ehU,775
jax/experimental/_private_mm/mini_dime.py,sha256=Afi98radoC8RWEy6suWokFjMiszppqhjDgFcaDZxF3A,9736
jax/experimental/_private_mm/mm.py,sha256=C19wJQmkdSmUeMNpYTSDm90DRlHm41NoymTEvBX0Xno,11435
jax/experimental/_private_mm/profile_utils.py,sha256=dZ1wvrhpaRNbEyiU7FpIuIi6qUfOvBHdMOcXqsYNRrg,2169
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/pytree_serialization.py,sha256=L5zgqU3-STe4LsDJywjOCwoketfhMHn7FMdSMz6enl4,21220
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=rBZP00cuR2uWhIgdralpm4HgVrFjJ-nTMeNU2zqP7zA,2776
jax/experimental/array_serialization/serialization.py,sha256=3wiRVOMdD3odJbxFPwK3gj1ttUhkrWWTEutDF_XXPBI,13434
jax/experimental/array_serialization/serialization_test.py,sha256=tJGuysBwNY9vnSaIk1j22aBe-5DSusKefWVuioYkv7c,39422
jax/experimental/array_serialization/tensorstore_impl.py,sha256=GcPbQUUSyG32K-hVKOJJOLZeVFlrTiUst0yrPNdLz34,24680
jax/experimental/colocated_python/__init__.py,sha256=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
jax/experimental/colocated_python/api.py,sha256=ie1z3UnsCJZsMt_RDTOjSi-sUZ-HLM5jvZEctAD1C0I,4476
jax/experimental/colocated_python/func.py,sha256=ET2am9LdOVR7lRTyow4k8zu1g5sUbfY085pUVNBFVZ8,17380
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
jax/experimental/colocated_python/obj.py,sha256=EsENLLgnS8ZYR6ubgYFUGHfSa7Hhq_MTbotbaZua-wQ,5657
jax/experimental/colocated_python/obj_backend.py,sha256=4TDWWp2p_xtv02R-I0rrRsW3kl0lWdiWKbpSR18Cg1I,2310
jax/experimental/colocated_python/serialization.py,sha256=P1jvN8rRfBmihitWIk4vscL1G0YekElm6ZYaN8m85cA,9014
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=027WlrLz68JD98hvGxkHMUt3Ss8wYw3sdscEso642i8,28929
jax/experimental/jax2tf/impl_no_xla.py,sha256=zi-U4UaZ7xVJ_Y9gmhHn8evKRX4_6SKJM33INEDucLY,54479
jax/experimental/jax2tf/jax2tf.py,sha256=7uLd1mzdjROr2FSM8s0OJ55Z3ezFc3x4NLbKkgzpNaE,146664
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=AN1R_wzL4NyCgMYEb6ab8fos2vruCL0PGj_GYAEWNEs,2936
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=0nnk1mDlssNRhOIG4u1_WSx6KXF9C7F5Eih6f5hFWjw,1670
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=c-Bno2RVxfY4aVgMRK9sjrCInqWVpTIBIQH3C9nPyIQ,11512
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8,7031
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U,8003
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=GrjvEI2NjbzFDNXmJSJKrN7bEby2RvQUgGuKdL15JGY,2405
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=De9TWLSyC5EbkQRg00S9bvVOo17HJuS_M4cztRFjOa4,4912
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=JA3X8RSf0n9O8pmWjQmU9PBHtcWs_mLTJos7N9-cjNI,5458
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=cMOGGyXrRlbOLZzjFqJRnn-ynXZhnOrFqSjKw9peXNU,66925
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=NlaRUQa0B43nQtJ71gRX3E16QYwXjwgCaxVEeDyFZoI,10041
jax/experimental/jax2tf/tests/converters.py,sha256=uYSiDOPUNaxcgVwLExAgquiK6M0VFH7j55OnbruOyaM,1998
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=holjJ2qso1R1VD_QSmKj-HnSoh_dkJgkOVLVqZqPIcA,7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=8kvyo8TjjTyIe8steWsggUlQ-OyRuh9QBP8Likg3uSI,60794
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=xQihm9rmF1eyHljrpxpxhFa6fobC9Qu6DzfA_Sc9dnI,69218
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=2ouDpIkKBvLLomEsmYU8jMKIm4iy4xKjONdlEG1Fy-s,6645
jax/experimental/jax2tf/tests/model_harness.py,sha256=-rgFOZu0GAtUx2ozfe9q-QpJ-7CHT3NxsUS8CJrbCsg,13870
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=_n_N5R-PzELJH6xi2MmZRcAG2midz2AB4FyOKeyT1VU,14799
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0,13067
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=bJo4prCjYtiN-PU85avZ_Oy-uViu5SmkVxSjkDXaM88,126464
jax/experimental/jax2tf/tests/sharding_test.py,sha256=F1VlHCHqJeD55v1xYUm1u99xhEtFt5vO19q9ochxjic,26411
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=8zUziluniinxbd3p7FcyWf3lFzJQ8rlyoleaU3oLZeI,20107
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=-YeNKbFpc_WDCC6vZiReai7R4DHQWFG7RTxtcjw41i8,23241
jax/experimental/mosaic/__init__.py,sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY,883
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=nNb91kzKzbbBavHv0JvNZwwKDpgpz3kmoHA5SeUW59Y,3698
jax/experimental/mosaic/gpu/core.py,sha256=9XxZIfIl3TuRk0Yqp99gyT8WNJXa5Y53NYEcvxBZUmw,33789
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=7er-_0rOnb6N-tYC_GlG4md0gxuA2rkd6iCUmYNTee0,66550
jax/experimental/mosaic/gpu/equations.py,sha256=kTtxfOP30c7D_ClQNKEgP-dz9Rx2akD1OFzFtJt9c7w,14417
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=_qXAbLVnoswWEjSRmXKzpmfX_An7TQjd6LgUeP8iJSU,126677
jax/experimental/mosaic/gpu/inference_utils.py,sha256=40O-yjyYci4jQ2CTW4Ry_tyuUpxXH-3zpSsbcxqBdTg,8056
jax/experimental/mosaic/gpu/launch_context.py,sha256=hXl9bGUN4hriwWTuwfxq8CdjAnX59AOVfUULjdGgK9o,40515
jax/experimental/mosaic/gpu/layout_inference.py,sha256=sJGKQjGsYHZof3PxbJJZgAiP52aqBLAEwIns14Q_Il4,26835
jax/experimental/mosaic/gpu/layout_inference2.py,sha256=A0F_fkzC25PsrnqTOUAGF800DAX8qDwILSMCM07V38w,28287
jax/experimental/mosaic/gpu/layouts.py,sha256=oKM66-O-TV_a_1szU-vZht31R6Mm3kTzWcfItNTswkE,11358
jax/experimental/mosaic/gpu/mma_utils.py,sha256=wsyA9Fm_o9oDOD7jfeCAcS3LJSr_Sw92C0Pt0nmZpq0,8968
jax/experimental/mosaic/gpu/profiler.py,sha256=nD82SFot5LowbjHOMSe55weisqCg_DcK_8j7uYMtMlA,15559
jax/experimental/mosaic/gpu/tcgen05.py,sha256=KMpvP4czFo7TsVAorn2rj5BKPYShuoSiC0g5vWTEClo,50808
jax/experimental/mosaic/gpu/transform_inference.py,sha256=EKFhSVptf1zC9VhpX4VmjPYI3sbtXMJ5eupkczcJXsA,18164
jax/experimental/mosaic/gpu/utils.py,sha256=WKiJiWDdR6BcixbWA4chbnlKlnEc7acctieGFc2uBbQ,54871
jax/experimental/mosaic/gpu/wgmma.py,sha256=CYMFA8uP1ZQVK8Wurz01nqOoes-9zckaSWM2AkTlj_I,16982
jax/experimental/mosaic/gpu/examples/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/mosaic/gpu/examples/flash_attention.py,sha256=3_6WQzfc1E5ERKBtHLriX9L5a0A2U_6b_JcIhxwVrr0,22011
jax/experimental/mosaic/gpu/examples/matmul.py,sha256=gumzFptbEqYE5yurGnucOJAva3Uq1ZTuPGWZJmCoZBo,14973
jax/experimental/mosaic/gpu/examples/matmul_blackwell.py,sha256=Rco1bDl9449iAatTL_uB1pKAG_Isypu9GaJq8agUimE,12481
jax/experimental/pallas/__init__.py,sha256=2LNYnbuHP5Rqd03bqTkwNIesmJrYzj29EXmbnr-6mTw,4396
jax/experimental/pallas/fuser.py,sha256=27RRgL3R9gmnJeomX9xOy4HsctvtaTwYKOoqGAIKs3I,1211
jax/experimental/pallas/mosaic_gpu.py,sha256=g1SO3NclVv_CZjeXHQE4MBEI5pOdTK_2Ffk41UCl9io,5092
jax/experimental/pallas/tpu.py,sha256=YCm0l0pBn_UiMDmGhkQpPXcGFhC7E1jzmT-NaPuC93Q,5282
jax/experimental/pallas/triton.py,sha256=HWFJ51-oRjRLY5fiMmlX2DwvE2hT6--7ghBzIMv1Fgo,1459
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/attention.py,sha256=9jJNoBTnzr_yPM5N5Q4DV-J9EYJtqilxG4vjgHH2r7s,23222
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=XcF9gFygjt3KrCZ8Ak-mC1ufNh-6Pvnvk80vBCUu1F8,36368
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=WDvuHgY4W8IG7rBrZy-EVaBvpXunMupz3XNxz2z9EGg,9331
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=AKorrKmQexqSEdNSUKyT1Elu21CVed9QO1Qv7NXh3X4,7646
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=1BvVgUAtMZni4kM_pBTjYMl7Lns8Axg4u5-jc7sUen4,17080
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=3LL6x4gxSVlMhob33M1-0XUrLaYEo3E4sZe0neY0vE4,11127
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=yviTcP15OFIWOu4UtYgT7q9r3BbsuhVbjOJIm546ChQ,16221
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=g1KB4Y4aRVuCMd4ZG89ZwWYx_hDmxCNLDQNVJx99V1Q,11406
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=LdiTCNLeuTYPu15TFM1VmJypLBeqLitTc-ITa_PwbzY,10027
jax/experimental/pallas/ops/gpu/softmax.py,sha256=sKeVjcAwN4va31DeFhMO_-ZLkLkGbgOtXucbG7G2GUE,2768
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=9XXcBpb_pXZGn8EOa-daiO_EJecOLdBtx9QCxCowEL8,5573
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=faIwM9av0WKRRW3Q1yOmhg2_DDvtMKtsW9LztFIiNPM,49874
jax/experimental/pallas/ops/tpu/matmul.py,sha256=x-2MCTChhr6BwQMQXzKcRT8PDVQ0ZvPjGxYOCaAgb7g,2717
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w,650
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=PnH6tp1yv31d5QPtedi7oI7sBgqu0iQ2Pm2lJ4EzqTA,27243
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY,700
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=MRVFNdCdETdHNS1SAAXWV_fufPJHpMsMfSQhiEHKxFQ,22026
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
jax/experimental/pallas/ops/tpu/paged_attention/util.py,sha256=t1_5K06BjSyQKaa-zqQnJDpW4dX1iQ-URfb_hEUJ53g,3254
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.py,sha256=XuQq_I5Pn_5EDon__0hWN9NOrOJ3grU3y0qqc8qKSsg,1055
jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py,sha256=t4sYYuwb6-P6TZ3PZsgDYUXrmjC23iVVrmgXHJE4-WA,31994
jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py,sha256=vCdHiRElQ59pwcdbN9M8IztGbsfdyKka4kk0nGwrqrs,97758
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=Zi9svyA6bUoSKNiAjdyYTqj1ALDgcLR4_fxiPeW1PvI,7494
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=5r3pl293zG_WWLzWkY5WKP-lxqFmLxwGx9hWu9OHiX4,1780
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=bSCSKbj8BCKnHjTeedOCBPfEDURYl1KeD9f4tsznPDs,4339
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=2par2bNjSjBCTHl8LExI1T1yuYt9VAYu74nbfcBYKfY,75020
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=ncsRlbb-oHneOkZqkSpOB6r7V0p_lZY60zBEsar364c,16307
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=6h4bS38tKAB1rKbcuFK1XBH6R14HSzGRtr3lOHrD3j4,40291
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/roofline.py,sha256=lA2cQoWo5mhqX4tmBLyRE5x7qF3YFd3urrK1Hl9d1xE,11368
jax/experimental/roofline/rooflines.py,sha256=xeTOa2AqYqIJoAgBcZ_DI3f4LeREXYcdJMmQeY6zBE4,27210
jax/experimental/source_mapper/__init__.py,sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q,1698
jax/experimental/source_mapper/common.py,sha256=s58vJuszNzoWy-YmpfnuL-noc9zXsEu8_6H7iYwC3pk,2508
jax/experimental/source_mapper/generate_map.py,sha256=k0SrNKPQMAG3fgc_Pw30Vhn2u92nTkt-BbjyhVDs52Q,2129
jax/experimental/source_mapper/hlo.py,sha256=OwbVuuDfu-gYdYSIPOQFWFVXhNprsUrQ4wXTkiU8qNA,4122
jax/experimental/source_mapper/jaxpr.py,sha256=PXa9zMzP2_fD_0vxo8Za-mw21pXp2Aao6IQkQIqmoJU,2595
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
jax/experimental/sparse/__init__.py,sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs,11186
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=nNviA0Ed0Dq-J_mRF1us4DQ4GUP69Gu3WBFzuSbLlWI,13334
jax/experimental/sparse/ad.py,sha256=zJLXiDufoAVcLTHdFitEXdbQUM_XtDiJoIktwDVuNZU,7789
jax/experimental/sparse/api.py,sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g,6596
jax/experimental/sparse/bcoo.py,sha256=6No2R20x7jSI40VGszhurWS0YVl4nH2sWkRImR5vDz8,129568
jax/experimental/sparse/bcsr.py,sha256=qSvCId7ZjWHNZNDVpEPdc_dTnB9446V5xsMiIFdpAxs,41314
jax/experimental/sparse/coo.py,sha256=_R22aml359eVXXBA_3BP4mQfrJAfmdCtD_BzCDLy2vw,24165
jax/experimental/sparse/csr.py,sha256=dhOQNlU57XwMZom4n8VVfhc8t4TP1OxPcwtDurK0il0,24332
jax/experimental/sparse/linalg.py,sha256=MdSVate9YZYqZVzckGeKTfANJGq7LS2TjIDKdcpW3rg,23413
jax/experimental/sparse/nm.py,sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I,8568
jax/experimental/sparse/random.py,sha256=o_g5_F8_EN4vmncmIkFKChU5uX4L3MVM6z-_qdoYUKI,3887
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=VF9ddMVtfVfsJMREIGSjDS41BDWwSVVw3zuwHO8bmL8,37486
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/extend/__init__.py,sha256=sPxASrJd4ncxJpz36RkiO2crdj-MTruQw3bgWbcCgCU,2039
jax/extend/backend.py,sha256=V5LvreaQZSodiF5vceSwkQNKxBWNVZDETtX6-5RgNVY,1251
jax/extend/ffi.py,sha256=9HTo5pYGummGwbjAikbQntYjkbcPnvqtm-l_XpG9PAE,1979
jax/extend/ifrt_programs.py,sha256=TCWvR-K8nHQ9PuUNoMWVAfEuYsutTNrefKwLg4AIRao,787
jax/extend/linear_util.py,sha256=02ixe_vuJbwdBvRNcpQGG4DtE_6cPbIpFCzpus8l1oE,1601
jax/extend/random.py,sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ,1083
jax/extend/sharding.py,sha256=GRpNqvTo7-_NhNk7j9JW3HXE8_WkDocmnj07hfFp60U,1327
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
jax/extend/core/__init__.py,sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM,1036
jax/extend/core/primitives.py,sha256=oDjHgUadWNBnfXGwjWom4QNclXBCIyJpCFyPqUiqAbA,6247
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/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,641
jax/extend/mlir/dialects/sdy.py,sha256=4dpcOKEHHkDh6cXdyEm2VGE9W-8701MD9egZmWDlkRM,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=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=p0gJFj9VvrsZbZavI-xg00Y-PzEu5Rw_6d3upq5u92Y,2759
jax/interpreters/batching.py,sha256=QqKOP4uX4F1PSf96ZrXi6x3DOIrU8wyWqXGTlt1nC2g,2574
jax/interpreters/mlir.py,sha256=EtjPbhUXkCVIwgTezgAWpAFkVMVTenOvVrgm4hl2Tj8,3319
jax/interpreters/partial_eval.py,sha256=LwWxdnS8YvAyfioa3Nh1Q7j368y99eW1TM-AmT1UWfg,3629
jax/interpreters/pxla.py,sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE,1801
jax/interpreters/xla.py,sha256=DBBiXItIWdCZ2hu-Fm8KXRtidNgMS71MxV0zZCtqcGE,2231
jax/lax/__init__.py,sha256=9zQVkYJebreBk-Kw34zzTIdLqGjJ2zuXhwL27lP7D9E,11248
jax/lax/linalg.py,sha256=I9ZzEhenoEzcBnfnU-jCjIspi-aOXHc5KJY3t_45uWs,1631
jax/lib/__init__.py,sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M,774
jax/lib/xla_bridge.py,sha256=EDBYSQ_C4C38CX98VxOnEVhIoyoltkIChLwEV1xFNBE,1601
jax/lib/xla_client.py,sha256=MM2y6jck43GgOqvS81kupZ5RLJpDdADlHTjkuWDG1QA,3030
jax/lib/xla_extension.py,sha256=ZKgrRQujcaA2YcUgWwbjd94Ek-zX7uB9ZKYIcxMPTGM,4100
jax/nn/__init__.py,sha256=vC3v2Eg_WOW8kJWDzbpKS35jDaabCP9iwnOyV1HQEJ8,1730
jax/nn/__init__.pyi,sha256=zmwgLU3MrCVt1HtAoPKebVGdy_pG42vzRXT-K0n5G0A,4288
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=K4eMTRRaAEhlikBlOVl2eEJCkzCqflx-akbJBGAYFmc,12241
jax/numpy/__init__.pyi,sha256=QoG5maSLPHVfOYHo4_d_XsKv5o1Cd33xudc7b4E6kH8,44631
jax/numpy/fft.py,sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM,1084
jax/numpy/linalg.py,sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI,1419
jax/ops/__init__.py,sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc,870
jax/scipy/__init__.py,sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58,1474
jax/scipy/fft.py,sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE,809
jax/scipy/integrate.py,sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y,777
jax/scipy/linalg.py,sha256=GXC1BgLMztvwRMxddJhvJS3obG81HLQ0OJFVt_Iv4V8,1370
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/special.py,sha256=aTtD4vxM5PkdpbYZ_aFDkwX4aIHovWQw4WJkFlzlcxU,2605
jax/scipy/cluster/__init__.py,sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E,768
jax/scipy/cluster/vq.py,sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0,776
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/optimize/__init__.py,sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s,822
jax/scipy/sparse/__init__.py,sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo,757
jax/scipy/sparse/linalg.py,sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/transform.py,sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o,802
jax/scipy/stats/__init__.py,sha256=_rg_z3SzFW5UNdLOwvm1RgTHyRWiu7LulOePjNqE0Mo,2093
jax/scipy/stats/bernoulli.py,sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg,819
jax/scipy/stats/beta.py,sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs,851
jax/scipy/stats/betabinom.py,sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0,792
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
jax/scipy/stats/cauchy.py,sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg,881
jax/scipy/stats/chi2.py,sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo,851
jax/scipy/stats/dirichlet.py,sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE,792
jax/scipy/stats/expon.py,sha256=YClw5PcqAXgq3VD4ukZ19HfTmRCMtrGk6E2BpXb6U9E,866
jax/scipy/stats/gamma.py,sha256=TvP4BE8MROMxQL6YNDQIHSaZCfxumHkjGbCXw9O2iUo,852
jax/scipy/stats/gennorm.py,sha256=hqwz7ueAgdtzs1v0LHTTro2PLKRDjWxTFuu9jXwXOyI,804
jax/scipy/stats/geom.py,sha256=pdp2tCdXOa3CUWyKJCyrFrMR7TBX0La5tVI4ALAETI4,787
jax/scipy/stats/gumbel_l.py,sha256=uGI-yIa7ymknKuVsuol49H-a8ylid993coiWq6mnNuM,868
jax/scipy/stats/gumbel_r.py,sha256=fcbp-XJbmb3DlIDS8BTfKmXhf0CDh3EwMFzmWQ0VMo4,868
jax/scipy/stats/laplace.py,sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA,804
jax/scipy/stats/logistic.py,sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g,845
jax/scipy/stats/multinomial.py,sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU,794
jax/scipy/stats/multivariate_normal.py,sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ,802
jax/scipy/stats/nbinom.py,sha256=hWMMt7D_rLRtiGjlpHcbOXJF03BfA3v9Cu972twYKCY,660
jax/scipy/stats/norm.py,sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4,879
jax/scipy/stats/pareto.py,sha256=xOqP9JMLAYdVXMFpCfCoRqxJNXVeWfJXkEqpOsrUjHc,789
jax/scipy/stats/poisson.py,sha256=Guu0WOaGAwtujTTWVlBZiBg2fLu4JOjx0fumX4m4Ems,804
jax/scipy/stats/t.py,sha256=kPmOWYVodYyD6cTxzX98qNxiFeWpDHBM1WwrdK5Ztoc,784
jax/scipy/stats/truncnorm.py,sha256=q5aSLcidwhZqFi9sNjqvFF9RxT_YLcYBMW2DofDoE5M,855
jax/scipy/stats/uniform.py,sha256=KCmUJttnFPbr-_ECG7O5xd-fy8CqvysIhepltRj_6SU,818
jax/scipy/stats/vonmises.py,sha256=NYGQbIKES93vjVumCkF6IihydEhrhH7JMaJFmGDfCwM,791
jax/scipy/stats/wrapcauchy.py,sha256=pcs6O0cdJTyhsN6LGTOxBm4TyuKkm3HRIdxfZVoz8r4,793
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=fFvvhbWBsHJel1cBJJOBhLeacpk-OJVJzRE-hZHxK90,8711
jax/tools/pgo_nsys_converter.py,sha256=zA_yA3Yzw4BJRWxPrPjkfLiG8uGn2JFbsraSkvnINOc,3411
jax-0.7.0.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.7.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.7.0.dist-info/METADATA,sha256=S6t2j-8liDz1TvrrIR4AA6jWTbKgD2pjrP7rb4EmCFQ,13412
jax-0.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
jax-0.7.0.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.7.0.dist-info/RECORD,,
