jax/__init__.py,sha256=AVZdlkh17lQdzBKkljg3HcrX3vBrVdi4RDlpyLUBklU,9496
jax/ad_checkpoint.py,sha256=NXatsd0qUy0fw2VgkmGsyEWAGHCj04EOr59Xy4YZEWQ,1548
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/cloud_tpu_init.py,sha256=Q12OEXt9BfKAPY2o7bbuDnVMz1QMHvzkxXODZ8-tSG4,1379
jax/collect_profile.py,sha256=Q5CuPjFBUCdZ7W7aSmYwG6BYyBOaF-tboex9ZZho2SU,6148
jax/core.py,sha256=VoOTglt07mSCZydMcHrABI-MBUBbN_xP5XnsqXKJVuY,4407
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=5ZlbQG085r56qJZkGi1tNxeQYTYufJvJYsGyN6N4Rks,1345
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=LQ3XKZaZB9RbYDUAd0IrQK4HdFPjjT9x_FZ4a6ci2xs,1221
jax/distributed.py,sha256=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE,710
jax/dlpack.py,sha256=lPvnhizhMsUdKJzvDl4fUUPfoU_vzCNvoX3Xl3z3Bow,688
jax/dtypes.py,sha256=HRwqpSsU_JD5ZJCjZ8u3MFUzf7JiRXBTaBRedpcJXVM,1224
jax/errors.py,sha256=FyhOBvQW_mPV72z0FXLeqvZ9PPWFfeDYtoxWDVOJv2o,1314
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/ffi.py,sha256=ooQYELAXRz1D7RiwDhUhFYUqfwl3Pr-TV5Gw1KMGhAI,1160
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/memory.py,sha256=uONVdGEZa7XodzKxzpCjwZ17cX3a1eRcrGnltSq7w40,625
jax/monitoring.py,sha256=ik0PsR4rbn1NswE4IJ-i6bpgmV4ueXuaLfAlagUMdNI,1695
jax/profiler.py,sha256=h7sMKCFMzMPSVlnmbXSj7w6bDt7Rf90S5U1x3WkMPCE,1276
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=W3juZxdpY4y0e3IwzDLdtqZEDHeW1DeCc0tsVUCI_6U,9433
jax/ref.py,sha256=t_jQxSaRICBeLL528zsCkxUyAQEU5bXFBa4lTysa6gA,1390
jax/sharding.py,sha256=B5cxsDZdoaoyPhO5FO19OP3HC-3Tb-_nPDOOsmPw92A,1909
jax/stages.py,sha256=KOQqCPRCnLKryQUrTRPVRiNuZqyV4chGi_oY2t3HpQ4,1286
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tree.py,sha256=FfouNfVkwowRidqI_8-zceCtFFxAAuf1JGzCFIZGjdQ,1174
jax/tree_util.py,sha256=Ko9em477wzRwOf0uXExrXJqTjC7msCy_RXvJlkYWs4I,3189
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/version.py,sha256=C9ntXIHQp6CrhbNEz9fBByRNR1NwSbW8EIAAXSidmas,6824
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=woj1yTQTB0DA8FvOYW3HuPvWQAkzLuBNgVPk6Yj01VM,8061
jax/_src/ad_checkpoint.py,sha256=7hOjTwd3EgoeRZ30TSvva69LGvAPExI1fAqfqu33m0Y,44938
jax/_src/ad_util.py,sha256=MPUbyULJ-jdSuZnNHeLsbzZXWwP6vL48Q4iud-U_dn0,4580
jax/_src/api.py,sha256=K2aqoeiJGF0O3osv8hdFAhT70Un0YHoQFNefs3FRVww,136474
jax/_src/api_util.py,sha256=8J3wGTJH-Ac9dKShtuXAm--CslNsx8wQ-ugjFnnlzfg,31314
jax/_src/array.py,sha256=glzuSiztZMJjGruyoWsMOLfglfMoLBJWlIAo7TbvtaY,52876
jax/_src/basearray.py,sha256=qX-m0BJHWhT4sYSK7YdSiE0UwSkuilQiqa9a98BjEJk,7422
jax/_src/basearray.pyi,sha256=GKVBzOsFMtE0Cm5ySrDPD9Uh_5EomLsPJBnsV6G1xMI,14752
jax/_src/blocked_sampler.py,sha256=s0HqkBXTh2t3eTqKV86PHf3YHvRJv4J44J-Oagu3oJo,5920
jax/_src/buffer_callback.py,sha256=MZmTSlA8RnewSdxW7MNWKQuupWeX9hlKgZ4pHdyU6vA,10464
jax/_src/cache_key.py,sha256=xmZeREQiCgKNycUpmHGn-Fq3dVtsnLj2FedsxTmG-w4,13056
jax/_src/callback.py,sha256=dSv3u-23Gf65IORmc3_xLgQTWxgbUF82hAanjumNsdE,34676
jax/_src/checkify.py,sha256=tOPK6O2xoks4I-Qwot4oMvUMSCkMzgipCHzA8bLIDAQ,57734
jax/_src/cloud_tpu_init.py,sha256=KIsKxtuqSCH-32l1FQ-4DblQSsMOPK-T1SQ81rPnvtU,5141
jax/_src/compilation_cache.py,sha256=g2XTMaXwipa3YaLGN-RRUhp1-3Cxborf43VDDzwV4Rc,14429
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=AyEGYRIdQPkcF9XeKHxln2M1QQWNSIz1YpIfLiccNHE,32356
jax/_src/compute_on.py,sha256=9RR7co-uwupbor6OAbi1CZMtfv6COqZV25A2gJHceAg,11389
jax/_src/config.py,sha256=pWDPVUyiqo6tPn_Wn9_B_LbrYHDgUHMo3nZ-G_xwalU,82937
jax/_src/core.py,sha256=wotJ66IOQF-MRzmHjWosItuc85wJRVFFgFfjzSMyT2o,147902
jax/_src/custom_api_util.py,sha256=GgJtgWnv9lTFUeOUjdd0U2xvVHEM4iCRBvqYoWqevb8,831
jax/_src/custom_batching.py,sha256=qiPIrB_5_U4K8GhWXOc8lyRFFMQU_I45ZlxBJ8sahVg,15275
jax/_src/custom_dce.py,sha256=JodPijlnKj1CxFGCIObt0VDbPCjxHKxXTaGJlNoQD1o,17474
jax/_src/custom_derivatives.py,sha256=ij33M9ELc5KhikFkcKhJSi_FRSkivmQ5YZ_XFHZ41CY,81024
jax/_src/custom_partitioning.py,sha256=TeLevSq50QZINQghpcMGnU7G1BGbdNanwVLvgKLZCy8,28434
jax/_src/custom_partitioning_sharding_rule.py,sha256=hPanM4Gq326i8QNdH0LL6ZByDA-grZLLFC-hTcg6orw,21200
jax/_src/custom_transpose.py,sha256=y-WoaBReTrAoexQSiW5w3JCKRCnUE75uza_6juYB-JY,9334
jax/_src/debugging.py,sha256=ehWJ1CwWFDRjrckBPLyBcUCd-hlIOwkyuGcusOMTSlI,34582
jax/_src/deprecations.py,sha256=eu61OgOFcIvKasgby9Qm87yjWtHMpEyJKCzzyMjzspQ,4605
jax/_src/dispatch.py,sha256=yupqk9qny_RP_vcPS9y-2gfcYZ_5O4x2-X6MoWnJWBI,30362
jax/_src/distributed.py,sha256=aLbmUkomKy7icKG0K2Kyh8cYexRnfXE1KVDDWDzuLsc,15694
jax/_src/dlpack.py,sha256=nx-ECIv5fiSe5cxWFymWD3qMmLiUTFiRJlzQnJEweVc,11414
jax/_src/dtypes.py,sha256=to6LlEjnFlL6uRQJzobWuP7YhZ98mz9afwdOAVFA09Q,41248
jax/_src/earray.py,sha256=uBT3Szpd0AWbZiVtgoadlSK8BDnauNOmcTnY007Q6o4,4460
jax/_src/effects.py,sha256=qQCTznjR8_76HVCp_MAAKKXbs1HvIspt1Z3w3UYS5CA,5041
jax/_src/environment_info.py,sha256=Xc9-Y5_DvVjW3iEhen51ATJvIESluEIrWJIF_e-oqA8,2031
jax/_src/error_check.py,sha256=AYMAndx4eSp_Y7bvzsg3ypbr7J2P5dq-LQvYlL-EsVc,13853
jax/_src/errors.py,sha256=VXa9nNqjMBlt1elh_XywuqFRKST8zbfdtw_rDUy3XEE,24777
jax/_src/ffi.py,sha256=8F14CWjq0hIaiRL4AfSLXQYOTCpsGTS3aReXc-84x1s,30695
jax/_src/flatten_util.py,sha256=sG6qaC_yWTcNNbVjc1A1KrwVz3w2q0b1nm7SQGFmKBU,3971
jax/_src/frozen_dict.py,sha256=V9UhNofzdhy7pTM16OOf7i1_fIfmRqLAUH-GSWcU8R4,1457
jax/_src/hardware_utils.py,sha256=YQcgkyPgQN94grS6VvBIVVFL2R-LDCsdY8Tqj43jnVA,2349
jax/_src/hashable_array.py,sha256=Awo1-CwOJJifHgGrlBDRfCgSC3myBNuSGxEjX5FWhpM,1147
jax/_src/hijax.py,sha256=-ZLb9LagqEHX2dvqKE8uZIE7WQYsyXM7XkvwKWfDZLQ,35428
jax/_src/indexing.py,sha256=XixjX2qqnbJKAj7yzpuojl3M_ZFFeiQm8SAapndZVxc,3466
jax/_src/jaxpr_util.py,sha256=f9JMIcOZ7kSakZq5q8OEyyqHJEa0d3Md--ZgG5v-j6E,9801
jax/_src/lax_reference.py,sha256=jC01QA6x7BiRUyO6kl839tTHuHHGHSepcAPjqskW5UI,19915
jax/_src/layout.py,sha256=E3LAwIAujgMyVfm_vVoDF47utfCBuMAV6IpxNxI0eVo,5801
jax/_src/lazy_loader.py,sha256=IEhbhAOHbWusMVAGzBPLKYFPkAXcP2sbK9Q3IcnyG1k,1893
jax/_src/linear_util.py,sha256=20W7ULaVN01NLeI4xQ9xf5N9dllJ274kLrYsqokacSk,18474
jax/_src/literals.py,sha256=HhfOdLGct1I-wBwlrYOe4Syy0Qjxc78PrkGMKqAOCMs,4075
jax/_src/logging_config.py,sha256=4Sk_kyeggOxktYSUfnXmo8zMoGk1Ud10XhYJXo1d1Aw,4272
jax/_src/lru_cache.py,sha256=ZRFT5sY1Paokw0oLVLXSBPj3DV9JEb3WgKDTfy8v9u4,6760
jax/_src/memory.py,sha256=xku_pvnyIR2XpG3brybs0u1MjQ75xRh97AlXgf2w-ro,745
jax/_src/mesh.py,sha256=Gqxs22kFWi-X6LxSgbCeAqbWqDkwUmJzx3x6E0C3OUg,21216
jax/_src/mesh_utils.py,sha256=I9doJNgO25ejIu6lOlGgHduHCraxYbLGhziWxkDDFmk,34504
jax/_src/monitoring.py,sha256=OcXRp8maRJewgZDoKLE8D-9tHZnjINXA_m35ewm9uPY,5631
jax/_src/named_sharding.py,sha256=6n7yE-BT5V9e4azwuV-0Hqlzdi50HuNxX48R6Vg68CE,21194
jax/_src/op_shardings.py,sha256=VvE1zwvbCBY5EyrKUygzCy_AiJCPz8pW3anx-i9qTHY,4388
jax/_src/partition_spec.py,sha256=CYLQn9ST2AVRklueDLwoWYKESJu1c7iYKGOnBmrRaWE,6634
jax/_src/path.py,sha256=BsY5OSYSZ2hazb3ihMD0DUVxA7cQ4BPk5BchZNNf2mo,2223
jax/_src/pickle_util.py,sha256=1Ew-6hHSqhG4TtOkTo0msfIF_NI-a_jefgfr_V7YCVo,2417
jax/_src/pjit.py,sha256=pcA7N0RCWxCwa-W2Kbn-hhk-i6TAOE_CMcQTiEZpjAg,108226
jax/_src/pmap.py,sha256=XLcdm86ydJ_E2ctmNabvTta91mWYb-cgQRg0Y8NmzIU,25454
jax/_src/pretty_printer.py,sha256=_UZ4oTCAFc55iKvstsMhquEW4AmuhfimYHXYchMnM50,3419
jax/_src/prng.py,sha256=_WO3Y_IuZQAv7IfhDm5oSgEcEuQd8_rmsiYAA_-eHoM,47604
jax/_src/profiler.py,sha256=VPF6Wr2JPDSU64WXm0MBihPoMCWK3EFHUk9cbDOJyEc,18416
jax/_src/public_test_util.py,sha256=KDMKncw85neOj-dy5mSzUM5ebOChF8AjoTIsW-uNFTg,13231
jax/_src/random.py,sha256=DYwxo2h70hAkMUX3Ii0zlVXRNpDczF6oc30iO86zXpI,115603
jax/_src/ref.py,sha256=G_7qltcYY4hmq2J-a9IxnuspKOYcJc76bUqtxsDAkNs,1194
jax/_src/shard_alike.py,sha256=vY6VYE9jSN1ZeCyxG7ojmTZerG6ujY6ouvcocW7806I,4001
jax/_src/shard_map.py,sha256=A4-lJp8Edi0AhLjfG9ItYu0DukoVBEJ9r-XlbstJpCg,95610
jax/_src/sharding.py,sha256=_Xe_mXwlDAWPNmq7J8khIMic_PKxEVPSubeJWkzpGHg,8180
jax/_src/sharding_impls.py,sha256=q2bFAnDu4_8xeSv5JrROU811gg_Zsj0d-cMfIpmIvJ0,49364
jax/_src/sharding_specs.py,sha256=j_gpsjDf8F9ClDYJzz9GpDNOkNsOyL_MBMd_0RHV1wU,8324
jax/_src/source_info_util.py,sha256=L29pyn1UcPsYhLXBwhWtkh_3npjjZqBkFV7L1FYONXo,10479
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=Fv_tyEdGs_5mt7rkHB-TSwLtK_aJtUDb-Q6YgeAHjIg,40404
jax/_src/stateful_rng.py,sha256=eGpvQqMc5jJJKpM2kikeyjrlj20_KzPEPlaNHBo_RaM,10530
jax/_src/test_loader.py,sha256=T1xrKBFMjyrtrxNXj1i1AfWh8wB5O2oQhxlhGZgNzB8,7770
jax/_src/test_multiprocess.py,sha256=bimzIn-EvjH6axxcZUVAZc9e28_HKVpvFBjHWbt1mfE,16400
jax/_src/test_util.py,sha256=brOSBDkNSFCM6_CXnAYEwIh2HmD4wKf9FPnKMmEnweA,87432
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
jax/_src/tpu_custom_call.py,sha256=KavdCKWMV780vZ3_bMW-gzC3uz8emvMNQgbgHbtM3Eo,33711
jax/_src/traceback_util.py,sha256=IqRDn71DosM4K9d3SipR2lAi-j_nIpSlbgSbTEmrh58,9046
jax/_src/tree.py,sha256=VgalOSSSFk9B82NHmhAPzkdpBb8H2TLdfW3TVjsxgaY,17117
jax/_src/tree_util.py,sha256=2agsUYRCm-GzziiY5SLgIN1wi1Re1M7bilAV02WJO0g,59318
jax/_src/typing.py,sha256=jdaddrjvthRKzrF6USr77IS2CyOTb-3nj0X4JDq_DZY,3698
jax/_src/util.py,sha256=N14KZLXulc-1qwt0deld9XByXsN7xqptdjGDWABhVNw,29210
jax/_src/xla_bridge.py,sha256=BdgbmRwiXujbumRYTz_Hmc1oYjayProfYw8IsGP48C8,45339
jax/_src/xla_metadata.py,sha256=NSNRRplS2wFQceGO3ePpF4pVTFWKWX2M684MzQ0DsFU,4875
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=jzgz_nolFfyXa8C8B4entrkUefUZyRMmzCq6Un1i5kk,9188
jax/_src/clusters/cluster.py,sha256=BbWPDzudcjSxAGSFbazo3zc6drUFqoelDX2ZPrFYnJk,6035
jax/_src/clusters/k8s_cluster.py,sha256=K299M1mW1vYa_6iEuJEAD9HRtig9jKNginp1wJKdV_o,9487
jax/_src/clusters/mpi4py_cluster.py,sha256=ZC4O9Wyktd4GobCjmN9AFS-W3XeOFFotghBHnD28qEQ,2760
jax/_src/clusters/ompi_cluster.py,sha256=57kq-MvL70Dt-C5WYgTeZ1Ivlg6wTEzWkD96jIfyk3s,2470
jax/_src/clusters/slurm_cluster.py,sha256=OI-OiVtZdO0u0-3ANX0NeP5mj_-gLWYDGaytsDYlV6c,2555
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=gYLBtg57q6ljtvS8fjBb2KlZNFJx2MSnlI5lkm3xtQI,81776
jax/_src/cudnn/fusion.py,sha256=0FF_DACwZ2HRuRUQA9LdAPgYKFADMhItn8AHed-jmZs,2866
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=yA5NtYqWb3n0684uMN7vITD3_gM6h_YjJi-u8nq0foE,25703
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/cli_debugger.py,sha256=7Bk4tfCQgTR69y63z9myy4SRqnP6riFBD5dl9tMMLG0,4834
jax/_src/debugger/colab_debugger.py,sha256=sKhZXzKk4yBX5qIDgEHCvfHBZiC9x_cTEC8ETGGuGmU,7798
jax/_src/debugger/colab_lib.py,sha256=EMRpaQg80wuam14kGVyS9S4GzNIUlMClIVNn5OlfYpU,4400
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=Dc4iHNo17TwH-7x9_g2rriLWxDp-twj7oJbVlrZDesk,78257
jax/_src/export/serialization.py,sha256=xzRqQiz7DaL9nFj7xiH6krd204qPQ-np6ZZuiZw49D4,30787
jax/_src/export/serialization_generated.py,sha256=rK2pFV10r5kmfz7aLcJAHBuQp2nJAGeWBsppf1vaTBE,42428
jax/_src/export/shape_poly.py,sha256=-x2zUBcLhoksUh1iIn86XTzRsKv7FjAUpmw69r3RjEA,86343
jax/_src/export/shape_poly_decision.py,sha256=RQQyg3jmJPdMmpCONwU8Ns-vzsEE783z--gHN8rbFW4,20478
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=Rc68o6EA7zPLY3nK-cyF8Tq0WPdvnk81rg9sLb22QRg,1269
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=5ZBF-5vRHhgLbkHA-0uhAr62HPLwVTAWW0Ia3k_YgzE,13841
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=ZEH_dUDZDNrMt_rH4pkZHMSH-FfkihomcuJfIZDB5Y0,16576
jax/_src/internal_test_util/lax_test_util.py,sha256=tyMSRQ_AGkmMTKSmYFONpToMKOcRtT6E8uF3ilbvbW0,13016
jax/_src/internal_test_util/test_harnesses.py,sha256=8hX-J6mxr3KHC5NVGZNIuHB4V0GTNM2rmMxY-ocinY8,126705
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=cGj4fw18G6PZuop-GkBrP9lWbiJSoUrGydbVRWaV0Qk,30355
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=MEP5BTGV4rsVb3hBWjZPhwfuPJQ0Xhq8wzaYcgTl7G8,18480
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_cholesky_solver_potrf.py,sha256=_8NkcH-yAwc4vw7XZ_jsBO9o_B2hX1jtoaay9Hoz9b4,33552
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/export_with_memory_space.py,sha256=MaBa7vvTBiVm6MdHZ3BMM70d9AqkswobRMQZ_PIZUGY,11314
jax/_src/internal_test_util/export_back_compat_test_data/export_with_specified_sharding.py,sha256=38XlkdNAC1MqYjce5gWxD9kp85k2Pn2rfnGBe0mEnU8,20488
jax/_src/internal_test_util/export_back_compat_test_data/export_with_unspecified_sharding.py,sha256=kByM7J491zPYtJuYqv0rYZ8Tq4UlF1lf8MCH-KSPaLI,18890
jax/_src/internal_test_util/export_back_compat_test_data/gpu_eigh_solver_syev.py,sha256=ZsvHcnMMUmSDc5rYkVx8WSrgPw4R_i5y7dwHRnBxczI,48280
jax/_src/internal_test_util/export_back_compat_test_data/rocm_cholesky_solver_potrf.py,sha256=irhAx5uy-9FY2CFapLfm8InSeq29ea2Qo63wAWG9Bac,34008
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/rocm_lu_pivots_to_permutation.py,sha256=rUuBzUMM-oqjSkvJHaW5I8T6dTW4rICZk9NfcHzhlZE,4466
jax/_src/internal_test_util/export_back_compat_test_data/rocm_lu_rocsolver_getrf.py,sha256=2FANkwqxZVKiuMgNvayC5nD1BJQVSJ8Y_OiLn4L5yqA,27624
jax/_src/internal_test_util/export_back_compat_test_data/rocm_qr_hipsolver_geqrf.py,sha256=qrP2_cvYZRNqnuK7bV3eq4pe80iScxqoO2-kN6uL8Zg,32328
jax/_src/internal_test_util/export_back_compat_test_data/rocm_svd_hipsolver_gesvd.py,sha256=hamwN-QKxXnWQmbIKFTEVNWDsUnry5agSwqEtpI-5Bk,85625
jax/_src/internal_test_util/export_back_compat_test_data/rocm_threefry2x32.py,sha256=gSKaXH6Zh_5rc5XA9GY3HNQamO0vMzfymP-hsVOLV5o,11011
jax/_src/internal_test_util/export_back_compat_test_data/rocm_tridiagonal_hipsolver_sytrd.py,sha256=d1WM0eY98IBcFg8uPASHz0QkD0bs2p9nRz3omMcWZHw,40914
jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.py,sha256=nIpwkX34vIPo0SN3ad52D4MhGJUpHqD1bFt19gkXEZ8,8209
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=OZQs7gjQeR0n0YlCSJm5RRSe1kJvhrkSdX2R5DQ9cgA,9600
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_boolean_constant.py,sha256=3xVgNk6jzNMGPD69ES00bBY58EUyJX0cpBYjKRzI_6o,13509
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.py,sha256=2LW9r0tASE2LEzDgeIyqGDR-ViDgmMIcqDkIrG-Wb8s,114219
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=RT0zT3fOwMPk_l6tMttym9jyT1Gs13H6YpWGg2euQ-M,66930
jax/_src/interpreters/batching.py,sha256=FOXlEEroe2qvWEGP_Q1KRifWoGEbdgYYKIvnk6yqraM,35592
jax/_src/interpreters/mlir.py,sha256=xUmBoB-dqKxSFXRlwkL3drRnPP0_C792aS55vUfTpKc,139004
jax/_src/interpreters/partial_eval.py,sha256=8xvYkE2TOBCf3aPOzW8stWPzbY2uhRYCTq_NMQzOLBk,115685
jax/_src/interpreters/pxla.py,sha256=8hTX2hp3iO5GWZ9sRJ-vdmgcuZXhUrKM9dSq2o1b7Ok,142027
jax/_src/interpreters/remat.py,sha256=gKF7nV2oAHdQ7IPo5M2rCpfLvCj0t8aTrvka1auixIg,5453
jax/_src/lax/__init__.py,sha256=bu37D3AJ6ANSqsGl2DRcJenVTTSuog7oj5jftMq7Kng,8510
jax/_src/lax/ann.py,sha256=w65ooLRE2IIoVBOOIl0MUc_PL7W7OawcyTe8gTuNz3Q,17161
jax/_src/lax/convolution.py,sha256=aSS_SFkJ-if9rEVRY_q9gKsEwPFUIB7AZDP0B5qIZN8,50169
jax/_src/lax/fft.py,sha256=q6_80wqlrHhkI13LhjFp4Doux8TkRxjIb_lNdAgNgbw,7332
jax/_src/lax/lax.py,sha256=x96pXM4-O1AaElWnTvGONQn1-CRWDGnxl8Q87YV6UaM,363595
jax/_src/lax/linalg.py,sha256=P-oWQ4M47mqimyPoOB2LurtgepSmYPZxqYmg15hn_T8,107297
jax/_src/lax/other.py,sha256=E1n6P-LXfYNbi9maGIDHRBvw994z6XojQvYRC2Kt-lc,13863
jax/_src/lax/parallel.py,sha256=224x-HgUu2Ljn5ees92ZFRpkyTUhh1rrOTrK3y0Rmuc,117694
jax/_src/lax/scaled_dot.py,sha256=EGZl_9mmseMvxzFdUcn1zbb-dTB4k9VCzsKn01hKcIQ,13719
jax/_src/lax/slicing.py,sha256=wxv06ka6riafp48qTy4fSlbC27ZdGE5LketirBsKQqc,150336
jax/_src/lax/special.py,sha256=aQpun6B6VDi0FG_steT1WHfQWYLwb_k7Gi5PYxvuyT4,28920
jax/_src/lax/utils.py,sha256=x3cVoiC8Pa3qF3xPskvhyn9XDFLqY_AJe-9AIjuLSO4,11284
jax/_src/lax/windowed_reductions.py,sha256=wIj1eigkfWyYG5ihQLg9ie0cu4TojMB_ORHMqm1p2qw,47494
jax/_src/lax/control_flow/__init__.py,sha256=JsuWyikJkp-5IVosqvubQfR6K1qXVeKMmSgNWLDzjso,2093
jax/_src/lax/control_flow/common.py,sha256=aF2wy1_-CvEYOmwv-Asu3FNRXhCo1CxY3SdtXuvrm3M,6464
jax/_src/lax/control_flow/conditionals.py,sha256=-cgk0gRbWFB6Y4tfQ53VfPNr6wKY3t6aw3tw2TZqHk0,53567
jax/_src/lax/control_flow/loops.py,sha256=4vtwZQRcBHaA6zQy6sKqavnQpLfLDT1q82KXScfe5sU,131447
jax/_src/lax/control_flow/solves.py,sha256=wUz3WkVBBhkwWcjYb9dI7zmq5E1eV3GGufRsJp9nIYA,20683
jax/_src/lib/__init__.py,sha256=Cr1yla-BX0YK2wyVjRFTm528D-GWubU8q23nx-XXxLM,7737
jax/_src/lib/mosaic_gpu.py,sha256=andYvo1TC8Eru9Pd_rM76L66Ozami3Q5awjdTkdIE10,1017
jax/_src/lib/triton.py,sha256=4S9rXZUP_3e_mYDTutLQKhUIwyLXdJ2HurR42Cb-dOE,2237
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
jax/_src/lib/mlir/dialects/__init__.py,sha256=zsqbkEnPPS1LY5XzzTp0TkAoCd9lAwICwX4g2YHBGtw,2089
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=Fr-NG42fW3NnQ_zdIjRHi9FQCeuGdQGcBJ4iafQ9E3o,48648
jax/_src/nn/initializers.py,sha256=VcgC8CnVGD-VqBPqdG4rlLTvTKaNbXeLgFICJ1uMp8c,27202
jax/_src/numpy/__init__.py,sha256=_RvbF3GcGcbXqVDVFXdVDq-MuJwf36zomNpgQCZWw4k,5680
jax/_src/numpy/array_api_metadata.py,sha256=7sw3kcQGzk3AM2fK39A8YYGl6DoommXsi916ZJKwH0c,3667
jax/_src/numpy/array_constructors.py,sha256=etLfNX5epC03-cRA_pJAWZV58Eow-_-e-aOBO8kAYOM,16697
jax/_src/numpy/array_creation.py,sha256=CgP_O-sTCTu85JQfMgNxYHem4UZ5Td6JbxVouXUAE_k,30955
jax/_src/numpy/array_methods.py,sha256=Sm9gA4Y2TEi-mxP0DFtCXIRmsoYNGnmP1IaEVHX2_nM,58897
jax/_src/numpy/einsum.py,sha256=1fIFSn793iFo_vI_UAjf4iytZNOaFXlyKbug8HwPx9U,24369
jax/_src/numpy/error.py,sha256=kqoCUrAfFL1zmtVJfydI3MrpB9NNuYxTqEVmX4iidzM,5450
jax/_src/numpy/fft.py,sha256=bpoFtzhhvOTsSsHTsun3XZcnc6ITKciZiOit4baax0c,52329
jax/_src/numpy/hijax.py,sha256=b1C8pSY8yi4s3XfWN81TyjcblIEgz9zbp4-OC5Lf_Y4,11812
jax/_src/numpy/index_tricks.py,sha256=VTGtSNYaMu6wzNwScnpeCbSiRySAcW2tIFW3wX39jM8,10034
jax/_src/numpy/indexing.py,sha256=cstuDTvYnTwnVMUlhBoqaWNLb-N81U4Kz_wb0ZPGjA0,68154
jax/_src/numpy/lax_numpy.py,sha256=ixosowJiS7k4u8H6j3VAI-GGh2eOQG_ZGsqrMx2ymog,337530
jax/_src/numpy/linalg.py,sha256=rCf2e-N0_QVADKJtuBEcojhXa4pPLveRTZsOf32_194,79364
jax/_src/numpy/polynomial.py,sha256=jAuRK7qso8BIl6yBHEP9k3W58yo7UHfrYfZTcvzCGsk,28420
jax/_src/numpy/reductions.py,sha256=5epTSX3ZUnCE_TRsylKidY-w0uKwJCLJ2v-22CXuHwU,109384
jax/_src/numpy/scalar_types.py,sha256=xG5NsqiCJZ9vMZb_p8dXRszToF_Vg_WmMR7xgp7WkZQ,3928
jax/_src/numpy/setops.py,sha256=7v-Gywj4VtrxHn-F-tg7MIf8C5SpOOV7FBiomOlin1c,47996
jax/_src/numpy/sorting.py,sha256=ZoJRnsQPXHHc9Gt8GmKWtNp0OTbMNWyYJjduBzvDdLs,16949
jax/_src/numpy/tensor_contractions.py,sha256=GH0XyCm3IaJSDwH9CySe7Fr8xNBq6AVhk-VKvQMJVGY,24613
jax/_src/numpy/ufunc_api.py,sha256=YjlCW4wwLPW8EKWEW7uRqFT4S9GtdcYeZ9ur6H2Jw1M,24310
jax/_src/numpy/ufuncs.py,sha256=AjrH8OL3xDb-3zkgdqqGKqwrygDRUx2ufbgSb1tCa9k,118226
jax/_src/numpy/util.py,sha256=7LjisAwskfIPhWeYNHYsNtPlkQ-FohAyXj6oAP6WYY8,17319
jax/_src/numpy/vectorize.py,sha256=8-VLF5kGcy5Sgv3JjWAoR8oHIOaSSmoYgJUoplRfB4A,14192
jax/_src/numpy/window_functions.py,sha256=jZsnPyv4nF6IXNHMCiX_9-gBGDdJfyb7V74fLN8TaJ0,5473
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=nVoktEZ1Znq_aO-e9GingBwJrMcsaNTnbykiWWWXRj0,18146
jax/_src/ops/special.py,sha256=zYKBMe29eMqKXDeaZ5BK8YehQdpAxAHPFKTmNjwKBa4,4380
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=gYUNwlRQQdAQhrVqzOaoxG5K62gECyYctAHeOdCH_Cc,61643
jax/_src/pallas/cost_estimate.py,sha256=9WMlS2n3SS8K7EzuB8Ye11H_ClsS1fzQTB4jbYgxPaY,8425
jax/_src/pallas/einshape.py,sha256=ctbDEgkYfg3fuZekVXKAnmEw8Wk_CZZlLgInamlKT40,20577
jax/_src/pallas/helpers.py,sha256=gFfgdxnauM0YPBABOujNMg-3s7nxvHL-HChhvhh1OfM,9193
jax/_src/pallas/hlo_interpreter.py,sha256=Capl5Usd130g-SBE2QtmxHvBAFVKBZ81Z27XNKi34Zg,19055
jax/_src/pallas/mpmd.py,sha256=9JI1IbtC91sqYMZ8qIhtJy35hZRhPmCqfGmfSTUEqKE,10768
jax/_src/pallas/pallas_call.py,sha256=uYQICMQ7GJtgtMz7zcJVWkTkueKmCAoLKlvkIlIgHEE,61225
jax/_src/pallas/primitives.py,sha256=bjesdDLgUjWrsQ7GnjZ268orNYuD5Mb8nQBinb3L-Gw,50130
jax/_src/pallas/utils.py,sha256=1Uuh9wdaTWR5a0NzeXwUx_rX7pDQQecQOrs800VGSmY,15746
jax/_src/pallas/fuser/__init__.py,sha256=9vpH6Fvikt8NEg8EA0o3ryFBh1oz37RMm1MudOZ-WxY,1265
jax/_src/pallas/fuser/block_spec.py,sha256=PgUPKEQ_f7_ggrFbAvQUNBbwn4rRBuj_69B4KaE8VDE,81563
jax/_src/pallas/fuser/custom_evaluate.py,sha256=nVKiLrI7qaXn-uJN4z0BL3o8BGtCO6dMnKmAAoQDfDQ,2804
jax/_src/pallas/fuser/custom_fusion_lib.py,sha256=_SoN2KEuNapYT_DT2TxA0ISvnJMOVmCkHp1BHXnv-CM,8951
jax/_src/pallas/fuser/fuser_utils.py,sha256=nu0u_l7Xv4mi0X3rRY2ZfIXvQogesdAAT0QIhBfirU4,1279
jax/_src/pallas/fuser/fusible.py,sha256=YqNyY5VOl-FkNB5Bvr8KFXkEykn4DaRev3xe6WhF93w,4455
jax/_src/pallas/fuser/fusible_dtype.py,sha256=wBzbo79t2KlQ64RpRsjw46Sk1XzakH8EJwnSYFDMjtc,18237
jax/_src/pallas/fuser/fusion.py,sha256=AvfY5PSPzoAT5Vu6NpMHPqobSgNfvv_600VbWIlcslI,1515
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=ryniYU8aRK-H4R6MwTNeEkX24nJgE903O6FMaDJ-C_g,11197
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=5vMKf88EBVR0eLoHUlRuuLrXEZw4Wl1V3JeBh8pgc8E,16807
jax/_src/pallas/mosaic/error_handling.py,sha256=6lVpREI3p92dLNfEdOHkcA0uK6h_MgvcmBFx6W0bI3g,5607
jax/_src/pallas/mosaic/helpers.py,sha256=pf0G5tV09sj7zZLi7zP0L6158qJy4Gz6VBJm7rZIhVY,2693
jax/_src/pallas/mosaic/lowering.py,sha256=uLkUChRbeGGRt_KcwTfBQLiAPukJOV7IKfF1mQ25A7w,156977
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=8YxHubcb0OyDk91tzuoKiZbkAQyeSxdk9y1C9MPE69E,19028
jax/_src/pallas/mosaic/pipeline.py,sha256=2YSefHPOfY0rrGPG8atoX51mQc7p2EQPAxi8OEAAjss,83924
jax/_src/pallas/mosaic/primitives.py,sha256=SpsmItMLooJNfLOalCeY8FT8QEkh6ZPDDFNDIBbVcSY,44950
jax/_src/pallas/mosaic/random.py,sha256=BkF4uekd6goGSk741Nd8tiKxH9Pv2kjAQMB4Ak42eog,7900
jax/_src/pallas/mosaic/sc_core.py,sha256=VOx3IkmLv8P0QXlOrQxw2wLg9vqfqScZloEKNamuEnw,9729
jax/_src/pallas/mosaic/sc_lowering.py,sha256=QJIGgoSrVgjNVj6PbCya0v0k4ioBL3_NkvxIEbziDko,43573
jax/_src/pallas/mosaic/sc_primitives.py,sha256=umi-rb-t3U0P275Tak5Aus2ffDMpaU0KHAKhyx8EX-s,44071
jax/_src/pallas/mosaic/tpu_info.py,sha256=cMQxSB5WZAYsb5ezdDxkTBNc19FpbqpyOmxOl3pIi_I,20921
jax/_src/pallas/mosaic/interpret/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/pallas/mosaic/interpret/interpret_pallas_call.py,sha256=Dt69FAfTkf8Xs2xTwB8p8vqF880p_u_pl-nWDfVwPb8,83130
jax/_src/pallas/mosaic/interpret/race_detection_state.py,sha256=7zhjevGaCJ1ABJsSSbQBPrtZGBOWZnLnz3IkBFDUsU8,6395
jax/_src/pallas/mosaic/interpret/shared_memory.py,sha256=ljzR3b3pGFCBtNmOToOnPJTV0pEOefG8sWS9fzDxpQM,29386
jax/_src/pallas/mosaic/interpret/thread_map.py,sha256=Kl58ZY9PHrirbkU_Bu_03R4pBlexLHBgKmLXaAgWTOE,3003
jax/_src/pallas/mosaic/interpret/utils.py,sha256=KwAr0mI-6GwU2D2C4NSRAOr2r3BI8FdT2DqF0FUZwv0,15311
jax/_src/pallas/mosaic/interpret/vector_clock.py,sha256=-Z8fzIB8BO0rx4cxzV-QeoOt3w7khYY5_BT1xFj92nk,1363
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/core.py,sha256=At0FPmzOuc1PLfLKh5WEdDn7ZLmPfNva_15AjZoqt_0,56660
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=E9FuTAyXSKBU6CAgU07yRo7u7JgWkXXjSK_GzY2aJRw,14774
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=SZ7Xl5yBnOOUrsXJMwGqdksRuvyDKMRRfflt9EL2D-k,156923
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=52mGScyyqS68mGTkFnS5QQ4JPvv9-qJtly2b3Uu6PUM,5713
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=IPUlwfUGqcGeRjoCMEmY_ii6pxOO9VRZ4XRyW3OXnzY,40976
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=e6-r-JPhnod6LZ2A7agdHMgjV3hKd_xJdpKhlYKa6vE,160507
jax/_src/pallas/mosaic_gpu/torch.py,sha256=wQW3PedREaamxDQ-SbDQY2wMM_Ovsy28ibbGizAOtVI,11211
jax/_src/pallas/mosaic_gpu/interpret/__init__.py,sha256=-aCGt5dPNdKLi-ALblRdGErT_9Or12mpWJ52yApKN0I,581
jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py,sha256=tElz2D8IIng4kvgdDbxKvleqJY7DZYUSVlhJg5ZeC28,29231
jax/_src/pallas/mosaic_gpu/interpret/interpret_pallas_call.py,sha256=2NNXxB6aEFFN9q_qOx7_FVAYvTc3BGogyQarB-p0uD4,18357
jax/_src/pallas/mosaic_gpu/interpret/jaxpr_interpret.py,sha256=UBJKodx-MYY7HBelZhGaAunYmkewo-Y-y_cnOih0a0s,17754
jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py,sha256=ODyCPiAWUumCSD4058Phd0iM67Uk2wTWw4vA9Ld-4gI,14635
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/core.py,sha256=SGp_aTNJuga6uxio9Tz-AUwEx2_ojwnPyCGuxA2BXBQ,1066
jax/_src/pallas/triton/lowering.py,sha256=oHi6ZqafHO1iRxal0SGxM4wo0tEV8ly2KH9H6Z_E3Io,96700
jax/_src/pallas/triton/pallas_call_registration.py,sha256=K4Zye-EkFU24UcUjNTyBRE7CJUkHnxExJ0Q6MIbFwpc,6997
jax/_src/pallas/triton/primitives.py,sha256=xZ0g6vyFmRPlVXUVEz3o5jTHt7oadNw3uY8U_kiQSIs,18628
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=YHiLcxEwNKeBgHPh2R6FwajmeBbNCkbOgskwhOu6BMk,15857
jax/_src/scipy/integrate.py,sha256=vHyXnhczC0l-Q7ZUWQOIakc1Knqlne4MMShb1DPIHf4,2334
jax/_src/scipy/linalg.py,sha256=1MF6IF0LGXaOj9_WKAAizWuxmCQg6W-itcxql5N6Q58,87208
jax/_src/scipy/ndimage.py,sha256=6XBQq3MXu2F0i1UaTCAG6MOMPfbPNn4hMnRNHhoyVVY,6945
jax/_src/scipy/signal.py,sha256=6fIT_yS2JXnzWeQkAiiK08cUP1nXvjZ_QubtPws2Lcs,47846
jax/_src/scipy/special.py,sha256=o1ECsi5AxKouF-_XMNKoU7XCL9cGsEg8Q-B0vHp5mGM,100824
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=3eKk6Mksbm22y8mkbexLet8VeDzY0ieGJhXBiQvGkfM,7938
jax/_src/scipy/optimize/bfgs.py,sha256=RmbR6n91EhFgkxJ9Oe-HB1_0RdvD5voG6ZMMOnWoFvQ,5733
jax/_src/scipy/optimize/line_search.py,sha256=FWUF2_9f_HwKcuxC399M4a3X0MuRSTRmjx3NOgjR9w0,13459
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=cIYkGl-Amg3Q-nTaw8Q3zz-Tzu6S08kkGPGstUpm1p0,27270
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=aaLxpNR8DQKQJhmTpub6uGvatnWpWQYx9J1d-NZiv3Q,10651
jax/_src/scipy/stats/bernoulli.py,sha256=MadOZcnOe0L4ODrF3FcBHln-MQ8WSaUHDEjucsky1us,4580
jax/_src/scipy/stats/beta.py,sha256=jjRE4LlyBt2BO2jmmN1z5PEasjl4LykhKTNc8846etU,8026
jax/_src/scipy/stats/betabinom.py,sha256=R7bAVggksU1siFT7xsaVCU2CW14uQhNCuBEoCbHgh8g,3514
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=_1nDUgHNtYlNTILqyZq9jxT4ZDdElwltJ6I5gYVmMUY,8361
jax/_src/scipy/stats/kde.py,sha256=4zc6vcIZ8mYVqplb8cLjtLQsfeRL2JwdIFhCx9V8feo,10429
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=zfIQvzbhCHx2LdLN5dOZjEFa55FV_YLnE4xfy1mFIw8,8908
jax/_src/scipy/stats/poisson.py,sha256=L6QHQZj2GOMCsOrD5npoOGVw9q1el6x-EAZ7LslUWM8,7323
jax/_src/scipy/stats/t.py,sha256=-POac0PdQa3tNpgksvZaVRHYrrbQQrlyXj6l8iP15gk,3226
jax/_src/scipy/stats/truncnorm.py,sha256=9KvkdVwpUu5CvN66y0tp6esEKDB4tuSSHmyezd6lQjk,9151
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=b52iytwSpPVYmiqoKl0I9yaNmmdjmujHf7mulxO8mSg,1026
jax/_src/state/discharge.py,sha256=1i_wAIku8bksrsLO5BpoVp1fh81sRz5NKZz3bZcVPJQ,37603
jax/_src/state/indexing.py,sha256=BlhmCS3_bsev8CNN9SSThBjzypQIlNA-3iKxKYyudC8,16442
jax/_src/state/primitives.py,sha256=DXa5-LS6KGOXgVcpOyCgOeKfdodDRDYO4wPlMQbqzk8,44538
jax/_src/state/types.py,sha256=LdE9sif5Ur87ZMcz9BRmEHpG5ZB8nie4hogmJA5sk9k,19684
jax/_src/state/utils.py,sha256=LfeYz8GbPN-CLfrmI0PZJmIqtPcW5zLkbnUgB3zZnKQ,3848
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=4rPNE2rU0JnDZ53Kzxi_ByZ-8tq6cylu5lhWgyU5MJY,3611
jax/_src/third_party/scipy/special.py,sha256=RDDuU2aB_e4K15z8-vkkj2YTmsDquvpnLjOXKuV6Rzw,9235
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=2VVzY5L9ib_AnxEfUrnvFc9w1ok3lxPCm6d0vNMowmQ,26618
jax/_src/tpu/linalg/qdwh.py,sha256=YIwsQpwqge7l0nCB1NLfy3Q6La8UdEFEtMMhimKrXLY,9601
jax/_src/tpu/linalg/stack.py,sha256=XPaFJtuPGfry1ssSqTNKdHcUs7yRUyzeD6aeYz0SrU8,2570
jax/_src/tpu/linalg/svd.py,sha256=gTmZuWdGkMGadhalcA1wGCSDNOKO9CQsr7IVnUej8TQ,10701
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=eOLn-UowNy-bEXngwaNUMXbUka5qTqmbQXQUYS8Ou00,13834
jax/experimental/__init__.py,sha256=V0WZhTy7yIYVX0BkO5Tpk5FU3OS1aqfXoIHPoxOZSfk,2047
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/fused.py,sha256=3wOnRpaGhJZKQPd-Q1TRY1HbV9v6ekPK4UbxV0rrNnk,7189
jax/experimental/hijax.py,sha256=9k9LLPkYeG9KToukmJUkShi3nsqpO9lHSetPP-i0EOc,1430
jax/experimental/jet.py,sha256=aNnK8XCbKY17Yu5SJmhWpbUfSjrSlkFTHllIuaRQo18,27945
jax/experimental/layout.py,sha256=fR0lxc08SqRZ0jv9bh6QACBFt_rFP_4naq5y43pU5GQ,738
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/multihost_utils.py,sha256=gq_NZzWA84RyGpmw7sdrv9VoC_FEeCi67TLHPPm07fg,25965
jax/experimental/ode.py,sha256=naggyXx7EoRZkGESbjIYsxEsTxxeIgUEbDlFEwIIVwE,10866
jax/experimental/pjit.py,sha256=iy2zCGu1eqrbbYTHm4UT9QCcHXzgAn0mcnxL_jukvaA,1195
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
jax/experimental/random.py,sha256=srL15IIcOo8A_DX_4WaGpDggdALBWPJBxkQV_f1esVc,721
jax/experimental/rnn.py,sha256=Is1S72fhAzzCLzBELP17FOGkk1Ggw3I8MSiGeekuni4,20189
jax/experimental/scheduling_groups.py,sha256=N5HcxT8b0A00K4RgaDsSngWCxL-A2aTVZgjlH2MZkrQ,7920
jax/experimental/serialize_executable.py,sha256=T_hxV_VhKZy--jmRdDCN1ezF2_YsEcwESTMqAMFYLJI,4632
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=daJk2CGjpmspNgkUMVCBvEosmbJ8BxgcGvRMMFwAPrM,1472
jax/experimental/topologies.py,sha256=AKCCES0n_D3WKyluII_vVRhV72Fy2dHamjAgSKVtyYw,2078
jax/experimental/transfer.py,sha256=jGPl9y0Qb31h_4LPJmi2FTZVghIN0bmk1sqH02tVRm0,2535
jax/experimental/x64_context.py,sha256=ediyKqtdUW15sPTLTlxB8K52BKSgrMAC0AUnFbNuUT4,1192
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/pytree_serialization.py,sha256=l4kWZ7t4g5_M0tpw-ecMB9qnWa-vF-IjnVtes8c6rWU,21706
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=agMetED8ST3oeyk2yCgw9nJ08VcNvP9zWuzQEPfZpMI,2798
jax/experimental/array_serialization/serialization.py,sha256=_N5k1hvw626vcFDRBIbso3cfmLyBIEZBrDrLdEbvmmk,13608
jax/experimental/array_serialization/tensorstore_impl.py,sha256=xEJKzJHMrKMgb3ATybgIBiE-QhSqAC1ykWv4yqASvoQ,25324
jax/experimental/colocated_python/__init__.py,sha256=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
jax/experimental/colocated_python/api.py,sha256=IhXs8bj18Lx921Hb9khYCHpA7x9olBUB77CKYdNrESg,6737
jax/experimental/colocated_python/func.py,sha256=utpdl_Lu3nRKdnDuBUgdkRUddssYP7GmA8XzwtuTKjI,26215
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
jax/experimental/colocated_python/obj.py,sha256=io1RlRZVHTO7iuvVUVCrCRRdoUpUnEMfVaHLjsPrPN4,8244
jax/experimental/colocated_python/obj_backend.py,sha256=4Gvf3tCdb-YDjPtn3qHx0E5JJMgA00UQjs558Q_WD0g,3982
jax/experimental/colocated_python/serialization.py,sha256=J7FsXxyogfy6Bjde4lekus53NDQX_lqhuTw0pXWSamE,12470
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=0r1UkLBOJBsXql-Zbvwg3SPg7hl4RLt3dWBG0UzE1xI,689
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=L6KfjSa5l5o0wKF-gAmT6ev5m4fxDdXM-IVig4S0gkg,29053
jax/experimental/jax2tf/jax2tf.py,sha256=YSuyxVvoNoAMYdaRcz4i9rZdo9EsCTpeUzZUaXxvgOk,38114
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=iq7-_tGfhd-GH2fR3er-D39QeVSna2RDtH8oquVwdm0,23317
jax/experimental/mosaic/__init__.py,sha256=m9n2VxXEoL9plQRUzwShRTvRYAeNyJKba2qzJQp8vQY,1038
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=VtphsOX4MQp2F5OJ5Hf_qvGhvAY4_u8RaJOC0zWErqs,4282
jax/experimental/mosaic/gpu/constraints.py,sha256=Bvayi-zy-eAdrBDTujg9b1vWPaublmqEyVDNiM_fbDE,33871
jax/experimental/mosaic/gpu/core.py,sha256=6F-n5yN2jZ5H-D2wP8d4i4AhqEYUOC8gd3FWbOx0fIM,42806
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=Fp-5u7xlBBI-0W7YRRKtJXaHrrxRWCjtjJteiYx_Row,89549
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=ibX1Oev67_mszGRV_3vNSROY_Yiyti0ZvEp9bFDqlV0,176501
jax/experimental/mosaic/gpu/inference_utils.py,sha256=bFJiZBK8xgjfFCH99EuDO5M6QxWFZAW1TD1iuoqGD1E,7982
jax/experimental/mosaic/gpu/launch_context.py,sha256=BWABwdO4u3L0sZoQV7-2wICMg9nzDmIyEVFqtHlkjRg,74211
jax/experimental/mosaic/gpu/layout_inference.py,sha256=kerrmE2TjIobV3CdjoqI_ZoJjWR2SIbfz5U0EtL85NQ,83651
jax/experimental/mosaic/gpu/layouts.py,sha256=_jjfzrIllcAAirwHhf5SNWJnkcO5u9A_yGK0quQZGzc,9032
jax/experimental/mosaic/gpu/mma.py,sha256=igcTIoZXMinenjg69ChoQW-D29tNcgnr9ASLu3NHecw,8752
jax/experimental/mosaic/gpu/mma_utils.py,sha256=AF7QKWnJua5ukJmNtL7KStGmaZaCSaEBtu6wudhgl-4,10707
jax/experimental/mosaic/gpu/profiler.py,sha256=fbT7kHwnWtmSt3rV3NoZwz0coBtL7ybRZWeyUVOD6RQ,14668
jax/experimental/mosaic/gpu/tcgen05.py,sha256=dBZVnfPAfcaF59eAOJqJhpBhw9z9L5iSvjvXS8zYl38,71275
jax/experimental/mosaic/gpu/utils.py,sha256=YYYUNoG0W_IS08e23rurVIgSwjxTsLUSNljKiz26VIQ,72444
jax/experimental/mosaic/gpu/wgmma.py,sha256=6Mo64a8U3IbCuLzZAbVMK0IdAxLajAnZ7mWHBk0QJAM,18749
jax/experimental/pallas/__init__.py,sha256=6TNp6hUNasUpcu4uouYUac7k1zwTNPhbPCYCHdaSKB0,4044
jax/experimental/pallas/fuser.py,sha256=FmH7hj89DZq_Ijhkx41IFSWhhZs-aR1_RWTHWrtUQA8,1294
jax/experimental/pallas/mosaic_gpu.py,sha256=k3-dqJh6PXlcyptbFXmH5OhcueTNt_BxUlTzRS3Tjlc,7672
jax/experimental/pallas/tpu.py,sha256=A3_JxhEjNdJNTYcxsV1Nhyk5s_mgf9z6Iz9Pw9HGxIQ,7126
jax/experimental/pallas/tpu_sc.py,sha256=zprBAiz-NKefopOEGrAyIeylr0w0wzNDHiQrKh2_86o,2573
jax/experimental/pallas/triton.py,sha256=J5JZFmxqk2uMdv0KOd5E1TdGjcwh05loPwRa0rlkLiQ,1703
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/all_gather_mgpu.py,sha256=2eE62KzS18ha07i7LQD5uAZTE9VeVg1u8QEubLcgPBU,8869
jax/experimental/pallas/ops/gpu/attention.py,sha256=_LWBv9V-ck1DKnnd1P7eeh1-4ppK2m8Cp7NBEy-b5Vw,23260
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=MzCoBwWhtJ5c_CZBeOhap1UgNsRvDS2winJtX695m3E,36790
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=JgXkYYtK2dYVlSD0MT88PkHmNSFp1l1lTrYxIppb-oI,12239
jax/experimental/pallas/ops/gpu/blackwell_ragged_dot_mgpu.py,sha256=sI_VYsT6C7d3XOIITo9GNIcm__LLS7ny1RZ5hd5DmwU,15921
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=WMStX-HjJDvRNGWU56Jq8UqwLTLcwj1ZUUcIIot5cgY,11070
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=Pf7EUNRDjpcYIXqTFoj5bKZBxaxA2View-kWrb3QDdw,17129
jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py,sha256=i86qlK5JF-1OQmHieWVLRy5BtsolRYQi7wyUQUTHMVA,12428
jax/experimental/pallas/ops/gpu/hopper_mixed_type_matmul_mgpu.py,sha256=qkftSh32aiD7WcKqo9a4-QFJule-Rx0CDMXnlmAI-dg,12605
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=ifcUo1WIazMEXijjhGbFVQJ6_jdrh218CtVq69RoU3Y,11271
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=MKaQ0GKMrhf2xYGvzvHQdM1AxeyLHc8uFGnCZHNmU3k,16191
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=5_zDeG7fM4drMKKiDGWs-pJNzBfOR1t4av2jjD_cc-Y,12297
jax/experimental/pallas/ops/gpu/reduce_scatter_mgpu.py,sha256=6nKNL6hNOJG3po-3EE3kyvW-G3CFABlkNFAAXB3uuk0,9705
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=1pxIRygmfsWZJRuGAnEOIFdQJD2QtT17mX2fTOLnDE4,10091
jax/experimental/pallas/ops/gpu/softmax.py,sha256=3E-coLRFVOYWyAcC6uLhIodrhfbl8wa68rIAEYz9mBM,2766
jax/experimental/pallas/ops/gpu/transposed_ragged_dot_mgpu.py,sha256=GcT28NhUWoaKxYyE7paTT486FrpPEp5L25aLfKFIwxs,9941
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=ZcttPmkBLcKOQfpxhJSWA2RFtY-6Dek0L2z2wehbALA,49781
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=-hkn5qlRxNqi9noij1SI0FAl3lodvHwDRYAt5Da_zDw,2100
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=_BEerBHpykoijwcqpagwKf--pXIWtt479Yz0qzzGlC8,22009
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=XBTKcqySDW4Bl6lDQvhpu0PAJK0o_L3fWuRE359qy9s,1036
jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py,sha256=KQnu-8EJJDOVqDpK4ZpisE4dtV7gW2TtoJMWru28ij4,31880
jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py,sha256=FuPLdgVEeGscJiCLp5RY0eS-V-klAFsznTAzHBienAg,97793
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=Sbrnpu1_b4VcC-gZysO6VTqZ3RqwLUMl7rcEv3FBfRk,7588
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=5r3pl293zG_WWLzWkY5WKP-lxqFmLxwGx9hWu9OHiX4,1780
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=lEDoysDSPHvGHJIwmChwx0oDgFO-R_2vPEFzN3Llqqc,4358
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=xHalLnBA2G-3Xh0el1FVixVGA8M9HLfeix8CxSQXj7o,79473
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=sONDvYMwDwDzi2LBrVmmp2g10iBcjam6sC09rRCcxxA,40387
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/roofline.py,sha256=Vhsj7Hbc6vF-Quyy8vrawziQluXVpHU4ETCzu7i2vog,11818
jax/experimental/roofline/rooflines.py,sha256=C4g-_7iV1VilK257r6plGTcMHNDImAgosZFI2HWVdNw,28455
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=QEDIXY1F9fwzgACNIe8KwxqH3ICXHL8ROthsa4ANFt0,7375
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=8m4B-xYwUfXp9McGxgMMDiyWagueWwekQz1bRLXUzJU,11419
jax/experimental/sparse/_base.py,sha256=mFppdPU_4s59k6zVLh8aSEr0NDvF4X7tmS-a1mhELlI,3423
jax/experimental/sparse/_lowerings.py,sha256=UWISqTt4FkrLAzwMe6SK1DJgJJzXSrV-ICedFiL1wyo,12832
jax/experimental/sparse/ad.py,sha256=zJLXiDufoAVcLTHdFitEXdbQUM_XtDiJoIktwDVuNZU,7789
jax/experimental/sparse/api.py,sha256=nwPo_11UAjrsAIQN9aElbNIFN4iwIjhawizgDafh2Ho,6632
jax/experimental/sparse/bcoo.py,sha256=pnXWUSHsP8itl2yuTGYDwFB1ipEWf8JASU47Tp3JxRc,129497
jax/experimental/sparse/bcsr.py,sha256=Sy3WoNa92dhSu1ODTfix2qC6KJb3N0K-HnNkaPkkpGY,41296
jax/experimental/sparse/coo.py,sha256=qOiTgC-dzk5B4I0YALC9zzY99J_wyZkO4wTm9M_G3js,23892
jax/experimental/sparse/csr.py,sha256=3ur4IZ-izRkajT1Z8AgNQjUrY4mKZg9TvFtIbAuPzjo,24138
jax/experimental/sparse/linalg.py,sha256=cLuUOoNcdQFnt_BVQKMBIl4Ab2xHVJWxFYrbrSF22WA,23531
jax/experimental/sparse/nm.py,sha256=tq1R0q1CU8Xpm0f3zLXR_DFmwU6iyqu5mj51nMPYeR8,8527
jax/experimental/sparse/random.py,sha256=iygSupHbydsxPp1hnLrGvQLU9_9vaFQ43WzCx_3zHVs,4306
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=5WJv_c4L4iEDX0JUXbLhvDsrE-4iVFAzAzHt_XNFhcw,37977
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/extend/__init__.py,sha256=ED6qQKaAf27YT4aaYK5vuZj3-qU5ez0QmiCj263dIWU,1645
jax/extend/backend.py,sha256=f34yAuA03Q4G98r8FOrQQtjVPM2i-5E3fwQ01eV3gcA,1386
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=AH1xG4kTvYokmplG53LF9qhuIx-RtDGtvoGWdveuEcY,1144
jax/extend/core/primitives.py,sha256=BQd8FTrHnEFY4daqeBJnpVsV6ALMgxM6mz1-N7hjwA0,6321
jax/extend/mlir/__init__.py,sha256=wGke5QNa1sJB6PqTteCbJsSaKVwzwPsb0NUKgWVg12M,998
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/mpmd.py,sha256=TzdnqMzVUtGHW3pFmnUa16PDC0xPgOCqFi8hUV6tbKM,642
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=AQhDO4X0q4OUteW82lbo0vJvwdb7M1JNFyilqX1Noak,4850
jax/interpreters/batching.py,sha256=_JqcgCOGwIjnNn_C8tMfq3cQGmAQnJJ-rAC52GoBYyc,5850
jax/interpreters/mlir.py,sha256=ifKNAIltmsU_Bmq45z-7MNElFkLY9rz42e9tHuzjIs0,3264
jax/interpreters/partial_eval.py,sha256=DQTJb6lRalIkppjv5-8L6iA-hS6VutlPHvw1XR2RtTE,6780
jax/interpreters/pxla.py,sha256=fIF4_AlXxxZupnO0accVkE3CEe_EVURIVmAb8VFxba4,8271
jax/interpreters/xla.py,sha256=TYm_YJ1oYol39-yei642UjRy2W-tHaxgJ34bw1oD9W0,899
jax/lax/__init__.py,sha256=ZWDfSGrx1arkkcQLa8pBBstV24lQ35-brvHhOZFFB6c,12252
jax/lax/linalg.py,sha256=-ApN-HR0U_5z4VJdw6IAoq96II92n3OJ6CR_1KyzoNo,1717
jax/lib/__init__.py,sha256=i2G6ZoULU2uH8HBmxui_CTzii_5bW5ibXbHPlIvkONs,660
jax/nn/__init__.py,sha256=G9v8E63aJYY0rgGMX_bTRB52uBR8a0qEkDQn92zA_z8,1782
jax/nn/__init__.pyi,sha256=aqHPaDoGq0xSkCrUPqBc9N3zoWTTBRFNKeL9qn8t0T4,5042
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=Z7l21dXZYBZR5eTDGBpqieDMKqvuGNUEOCUAnGFr4ec,12953
jax/numpy/__init__.pyi,sha256=xLRGqbu3WMpMBmQeKQeXb0KeJEhPvSPSjhP4j1PuPhU,45044
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=e0IKYyULJG2uWRjvIC7AGfhNN9lAWJVFTbJjQuh8iEA,1428
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/special.py,sha256=jwos1-Kn5wS2I_XYMuy0tfmVb-KKCRuW9WVrqv3I3yc,2621
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=6lR5cOlMmlSosez0hLa2BbKA4h74HWHL8Dugqj9PPIs,867
jax/scipy/stats/poisson.py,sha256=OBwWU15Qbjv6HBrob8YXteB9f4StLmYAuEYyu_h14Nc,825
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/jax_to_ir.py,sha256=Ok893_stWzeC_OIp8EnS9jDUcukMDjlqUA13DDjbkzM,9213
jax/tools/pgo_nsys_converter.py,sha256=C2gz5kMqZRqI8t11HOswanwcGDlrAHF_PTh-p4hkVrI,3407
jax-0.9.2rc2.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.9.2rc2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.9.2rc2.dist-info/METADATA,sha256=MJW-0CNgTwOHpa1EPPht5M1QV0HqQYnzUC4zApfVPX8,13335
jax-0.9.2rc2.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
jax-0.9.2rc2.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.9.2rc2.dist-info/RECORD,,
