jax/__init__.py,sha256=ChbGeWH5CcTPiN8ZZfdnMpEilPXkrSFDnBYcqyuZbB0,8990
jax/ad_checkpoint.py,sha256=W7rJzoWBbohkjgjF3GZvU63ae_Za4sL6v_z3il5qRrU,952
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/collect_profile.py,sha256=ZGiQR6LFtx4c0dCmMzJiuZIYggesXLM9TCePUmdsU-8,6157
jax/core.py,sha256=niywwF0GGexS4DZmZ5XGNRKaNb2b2mRCPd2aRM48w3Y,1617
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=fnGTHSIfrPu3kUmlJfPMsYKdhfdXpx6LVnaOXVKq7L4,1307
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=VuZubjoZ2dL-bNNtdvuG-eNCtlNdqlUzgcAS-BCxOPw,1270
jax/errors.py,sha256=cGk2LbKFAzkIrbaNKu2wNDFPzbnwz7nS4H-BLxzQP6k,1453
jax/export.py,sha256=CU-7uyZLGQmkRpq2dMt-QHP1XVW7BZaRrobEkmz35fk,1594
jax/ffi.py,sha256=hIHyQ70fdjF3YaNal-btdyB5eVHxVHAv51FP4VgUk50,1229
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/monitoring.py,sha256=ik0PsR4rbn1NswE4IJ-i6bpgmV4ueXuaLfAlagUMdNI,1695
jax/profiler.py,sha256=ruT_OwprsxMyLylTWyXsS3xTCb4eLSWz9ypm5YmIagY,1442
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=Mzj__kEvsGMacwD6tEY7goGpdTriK4dHDK9r5ToDXEM,10028
jax/ref.py,sha256=77I6A8lbOcLprxCBy7XR2arFcGM08p8JnP45Y53CLU8,980
jax/sharding.py,sha256=YRznqQmhzlVNU_7jwBjZXCoGfIApX0uyFi0bJ8AZ2es,1588
jax/stages.py,sha256=XDxWVwXuXQgRBVrgaskb5sTzumDl5rgcDG8t2HtcL_4,1290
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tree.py,sha256=FfouNfVkwowRidqI_8-zceCtFFxAAuf1JGzCFIZGjdQ,1174
jax/tree_util.py,sha256=e0ygHErJRAHtdXxlx0DA9tCP-capcg5LGZSZs9RnxsE,3331
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/version.py,sha256=pXM6YzmD0rLc9A9Dt_XemkwtSfdZjjTaKZJsESPZgUc,6849
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=3m5Wdgnkl0mF-lJ6VSjOWNHwvnhO8QhjGhh5qhn_v-s,4836
jax/_src/ad_checkpoint.py,sha256=yW7qi6uxn9aKWXmjg_aVwMiZCTHtGVEpP4sCNonUgZc,57971
jax/_src/ad_util.py,sha256=rDHDzDjmqNlW0FJI4nE6PshxZknT9hJ1bXaekU0bUfw,4891
jax/_src/api.py,sha256=tEuxzIdBK0cQ7WY6yhmnpqncDSGluzeeh7t7x1ewPms,125707
jax/_src/api_util.py,sha256=wCmFFhpaYD-BLQu4nJBcMdzoV9m9yig2rsyDzLSbqGI,33043
jax/_src/array.py,sha256=W_iry0uvqBKfOt7KWaGUljcVE2iSq33JEZPpa8nxL1c,53795
jax/_src/basearray.py,sha256=Vj3rUlNSWS4ZJjudKYHhv7qudKTOQjASpvtPzM3gKoU,7122
jax/_src/basearray.pyi,sha256=KnRpoBFoumnVNTi-aHvPw2D3lWK77S-hM6KirGqryvk,14755
jax/_src/blocked_sampler.py,sha256=s0HqkBXTh2t3eTqKV86PHf3YHvRJv4J44J-Oagu3oJo,5920
jax/_src/buffer_callback.py,sha256=Hp1zPwQjCLhsLzTjB9UJG8Mg8Wu7P4rp27tUpDE9sHM,10514
jax/_src/cache_key.py,sha256=BgJnB6EPb9cHsxmK1HBBhnYhxd05vyi8enHu5rqlDv4,14726
jax/_src/callback.py,sha256=YP9uocwHG2xR77bDt9p11hmbJqj-LhJW7w9qR5sqNyc,34980
jax/_src/checkify.py,sha256=0qr52hQCaBcEa-4pc43wzJ3ndnn9_1bbqaqjHX_nisM,61366
jax/_src/cloud_tpu_init.py,sha256=wzctNetT0NdiVk0l2tbSEXaMxQxVtiC27Xtw5f9bWxM,6441
jax/_src/compilation_cache.py,sha256=GYAfSErgc1OIVgTO67py5UXGVgara6m-uSWwcKyV1i4,16462
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=L9UA4SNUP2nNIVQKh80KNMAU_53Xps7HGyRnI10s7yU,33492
jax/_src/compute_on.py,sha256=7NqVHdybJez9PZ5rFMyV6qEExA5d_NMls_4DAv5Ahxs,19613
jax/_src/config.py,sha256=WaHQcj7o6-ZEGrIoWeXLivFYCrWk8r-ZpAiqjcelhfE,89833
jax/_src/core.py,sha256=b1QNzLXiJ116OM9EjZSuGjFfKdot5nPylimk8xFS0LI,163318
jax/_src/custom_api_util.py,sha256=GgJtgWnv9lTFUeOUjdd0U2xvVHEM4iCRBvqYoWqevb8,831
jax/_src/custom_batching.py,sha256=iIl9WwzR_MmbH3tnhfT0uU31VeTos00bmC7LI9aEQKo,15328
jax/_src/custom_derivatives.py,sha256=T64HN46SOTAFMKT6G1Ex9vKEbeipx80Pc-dPpLoBqK4,99798
jax/_src/custom_partitioning.py,sha256=LN4I2n7JI4faEXIp0D0nFblyWO7jcgwhELzI9M9s7xM,29614
jax/_src/custom_partitioning_sharding_rule.py,sha256=NpdehLzMqEELhHbeQd6-jRkG6siLiYm9pmgQBVW5XxA,20765
jax/_src/debugging.py,sha256=yLv2U0jj8WxOpoqCvY3TP0R6En8kMunuWzhIPSIZFyQ,36840
jax/_src/deprecations.py,sha256=vnv0u2JuHIEjVKyRBFzQ0KGnhA56rqAW8TL0wtO-ypM,4714
jax/_src/dispatch.py,sha256=_mdDhoMeupa2ga_pOazMKqvhdZ6hZxGJ6YmWsInW5RQ,31798
jax/_src/distributed.py,sha256=0_cs1jqZKULTbCdkZ5uCT1BboUeJ_VE6vYnCHVQNxpM,19678
jax/_src/dlpack.py,sha256=Ru0BOtf7NbDajI7XzbuE_a4Kt3Vexps1nlNp0gmjbKc,12221
jax/_src/dtypes.py,sha256=d0j_bb16sZWVxC_HHAHjpSRtJJBDjdRWEsq7PkeBRhA,42237
jax/_src/earray.py,sha256=gXfpTpbpauyPfxI1G08OilC2Iku3a7CWkjJd3zFSKeo,4227
jax/_src/effects.py,sha256=-Sx0aMLVVM0EPBjzbVbD90JCXhJEvv9ZIQrOq6u4hk0,5739
jax/_src/environment_info.py,sha256=f2s5UG-n1L9wpCHu9pbxLTOykCOM_AkhDH7MPNGmJWE,1999
jax/_src/error_check.py,sha256=QRDj93GrxUph5hcaSyRZ8ujrS4_UWGuBWGyq94nYXh0,13868
jax/_src/errors.py,sha256=dbAF0eS4jJLUiOlbYJn_E_65NBQpjL5V2Wb4Uh9TYJ8,24578
jax/_src/ffi.py,sha256=sUjRDE2sE0AlNEBl6GtzXbXcR1G7ZyN5fLoM8Fo81p4,30420
jax/_src/flatten_util.py,sha256=2Zm2plIPflJ_OcX3ZewPMj004zCMvDxjZMT4P_ohNrk,4002
jax/_src/flattree.py,sha256=h2fgQ0-q8K-xyhFtT8_Zs84KqmcTRU8jTVrWqwY_76o,13448
jax/_src/frozen_dict.py,sha256=_48jSkeKkbNyKgmwfy8S5W9qIvCLzfIfQ_XxJbJZMuU,1331
jax/_src/hardware_utils.py,sha256=2lAna22_EovcHyjPxK_zAV3_Wczi5E4a3KNYvel6DjM,5744
jax/_src/hashable_array.py,sha256=Awo1-CwOJJifHgGrlBDRfCgSC3myBNuSGxEjX5FWhpM,1147
jax/_src/hijax.py,sha256=Veh8i_Ta6lpgeA6odxIv6rjozcqU0l9EIepsvfNHjHc,69421
jax/_src/indexing.py,sha256=IVaqbEObsorp29vE4y8kXTGKG-fHIbgb7k9UrCMPey4,3818
jax/_src/jaxpr_util.py,sha256=aZ08kI1hQjKI_I_o8Xiz5DLmNpJmi8xbaI572pOf0NQ,31142
jax/_src/lax_reference.py,sha256=T1Hlg_ubHuNusj9dywTkp4kzg1aVxFIeQP84Nytg-zo,20809
jax/_src/layout.py,sha256=49oZN1pFnh_UrOYtEnO9q3Dk7lLNjkq5aCpeIAFDfw0,7463
jax/_src/lazy_loader.py,sha256=IEhbhAOHbWusMVAGzBPLKYFPkAXcP2sbK9Q3IcnyG1k,1893
jax/_src/linear_util.py,sha256=Um_dewR7yGwnGBTazAIqKI3hP4ervmicInv42C-JZbo,18553
jax/_src/literals.py,sha256=ulGl6r5F1bPnLynQwHOVSRcNBipPvWaPDnU3W-sS5sg,6295
jax/_src/logging_config.py,sha256=4Sk_kyeggOxktYSUfnXmo8zMoGk1Ud10XhYJXo1d1Aw,4272
jax/_src/lru_cache.py,sha256=F_1XpQNAqIQdY7c_li4t2TEO2IjQuTwBITt-9s5YO38,7044
jax/_src/mesh.py,sha256=OxQidFPvS-vy9E4dAlc7N75i9RSX2eAQesxqqLANuW4,22821
jax/_src/mesh_utils.py,sha256=utdglxhoW0Cu4CVZNOjz60nchGxWhFLAfVUV9MGUoHk,36861
jax/_src/monitoring.py,sha256=OcXRp8maRJewgZDoKLE8D-9tHZnjINXA_m35ewm9uPY,5631
jax/_src/named_sharding.py,sha256=EBe8awxsvm4v41i6JF1KkNq5lUtDIjX8ySHRNy55wWc,23368
jax/_src/op_shardings.py,sha256=fFrlvDZAENwmRW8uu9Zg07m7v4R9wLW8VbaR2zerJRo,3907
jax/_src/partition_spec.py,sha256=7159GWoNUUnR4Hu_nwDbPzDKlmXSGvvOkSCLFgIovg0,9745
jax/_src/path.py,sha256=kX7Mt_vwwncT-CLWsl3ykZTAb6ACHFDHKF_rE_yWLso,2210
jax/_src/pickle_util.py,sha256=TymdLYgrO_zREdtjkUzmlkXP8FZUQ0KWz5ATy3SKJbk,2248
jax/_src/pjit.py,sha256=TLrEHtaThTyocVmbo6aw2ducIZmPk_rLQGZELcGsimI,117239
jax/_src/pmap.py,sha256=rW8ggK1xLEHaFw9i_ONPJDZVH8jgmntlMh9dd1kYxVo,37194
jax/_src/pretty_printer.py,sha256=kbHKiK1TXhG1492L8P59ovTzKqfUoejtt_fQ3bEubt0,6634
jax/_src/profiler.py,sha256=2Cr4SZoka8Gfzf0CJelMDM7amIMqghwUY_zTyqNN-4Q,25588
jax/_src/program_order.py,sha256=pDg09kM-EubhdpgGHA8MZK1WBXNWR-XiX7df-9NQA9Q,17696
jax/_src/public_test_util.py,sha256=KDMKncw85neOj-dy5mSzUM5ebOChF8AjoTIsW-uNFTg,13231
jax/_src/ref.py,sha256=ZEAIx1AMn5HBLMQuVLQoiSmlL7--Mtp0mNXbavgxdu8,1501
jax/_src/shard_map.py,sha256=Bk-yUZQCMxf9IVF9hvNFAO7stL7xb2WLct-PBqa0HDo,109287
jax/_src/sharding.py,sha256=cSPFRbEyy2jRwVbALejKdGeKkXHSq2CDqJ_82wMJivU,8222
jax/_src/sharding_impls.py,sha256=qRNJzRbSEFw31yPUH2XkYJH9HZxqUOROEAohWeTxo4c,40295
jax/_src/source_info_util.py,sha256=NajKNyCTUtd-qcBIG5o8-E-oqDamcwCUB3UHcD4Vvk0,11136
jax/_src/sourcemap.py,sha256=AuxvuvK3fbnIi8rDuwR-j4J3YjWlduLj2Y9QJKcdVIE,6624
jax/_src/stages.py,sha256=hfr_79Q1mCg7UuIsXZkpOwVwL5eWsciYAA09LDinrKI,43991
jax/_src/test_loader.py,sha256=Ai1We-HhlytaQu-A-pZ31A6vLzClHaUwxOJqrC8tayI,7410
jax/_src/test_multiprocess.py,sha256=bGm6lv-DtdkGATuPkFRE3G4IY5Xg2vqNAR9TbGlDhBs,20333
jax/_src/test_util.py,sha256=Gk24EYte7eZPFntaXZnfocwBXIly6JE2foynnypAA88,78224
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
jax/_src/tpu_custom_call.py,sha256=6afGvrWGS-_fFv3qlwflu3qMmmLn4JEJR4O88or_9a0,40641
jax/_src/tpu_info.py,sha256=riyjS1laV0nO4vnZxxB4aUgAz4czUttgKJvi0zktqA4,25908
jax/_src/traceback_util.py,sha256=TRQJWEJtaAqfST-O0Y9WEZoX3nGC48UqiDJpNbcZUUU,9169
jax/_src/tree.py,sha256=VgalOSSSFk9B82NHmhAPzkdpBb8H2TLdfW3TVjsxgaY,17117
jax/_src/tree_util.py,sha256=OjfXAJBqKBbL7E3Ul2nyGEkUov2QYT3fxVek4XGJ_BQ,53091
jax/_src/typing.py,sha256=tXFymVkv2VVSveelKufwxb6P35Cs7PaVB3qHRi7qBn4,3750
jax/_src/util.py,sha256=JvPBK45zAaDCadUsMxKRhyy0PcDn00ZPQW-oKKzQEp8,25804
jax/_src/xla_bridge.py,sha256=EkfM2koW7sgKGCC4pwvSKTzBpvdVPGOnwJaTzMtDca8,46909
jax/_src/xla_metadata.py,sha256=deRzON_-5Dr8hzr-u4IDGYCmHj1tm2fTOpLbRFh15Dw,20467
jax/_src/xla_metadata_lib.py,sha256=Bn6bv8JFU24G4jcp7g2tgH-WUdqJtHVISwJRTrLwUDI,1457
jax/_src/xla_transform.py,sha256=FToTdVij6dOxZAsLxl8HbtF-TxFAIr8aIH8uQqG3ieQ,4062
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
jax/_src/clusters/cloud_tpu_cluster.py,sha256=wjdgmU6JTKruMZy8rYgIHjY2aiQsV9Vk5SNzXttzKVc,10270
jax/_src/clusters/cluster.py,sha256=dlgXWj40fEJ7Pt92kYsm7eAGcB89j-TtpCv5wpka-5c,5889
jax/_src/clusters/k8s_cluster.py,sha256=VhplqSuUYuX1JIMJBRjlKBcLSbE03AGv477BgS4aIqo,9493
jax/_src/clusters/mpi4py_cluster.py,sha256=OZ2O5ZIPFvIcf9acNyg923wS7Zej2WGH9KHheSe_wDI,2790
jax/_src/clusters/ompi_cluster.py,sha256=KoTwNiumERMILw0RuSUNcBb_dAmwiOOGkt3Pe2Ja9OA,4048
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=WGDAcaJWfnVFOBSEj1-95Iv2qKu9PcgcaM7UU8ushE8,92541
jax/_src/cudnn/fusion.py,sha256=tTEC2LrHCGhoDSMeRErrkT2Q_fTRj6HMkgD1lqigv-M,2836
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=evv1CwrOv-C3Xx5gTo9Wv92wIiT-Z9zFVxLDlolkm00,29334
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/cli_debugger.py,sha256=7Bk4tfCQgTR69y63z9myy4SRqnP6riFBD5dl9tMMLG0,4834
jax/_src/debugger/colab_debugger.py,sha256=NvMyRShLM9MX4Fp-MZJ0tTVG8RIWgXrBVWe6Lfa5D2g,7819
jax/_src/debugger/colab_lib.py,sha256=I3CUTT9M8hCNMwQ7NJWRKs-WL776UV415hE4s_9IooQ,4337
jax/_src/debugger/core.py,sha256=eyfWAU035lnJBS6fcnWP7d_TpL6t1e3FG5sSBj60xA0,8622
jax/_src/debugger/web_debugger.py,sha256=thfhP5NoWNfqd5PhJVzfPITMXP_8sB-PrQyJx3qzP1E,3463
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/_export.py,sha256=CnYAJ1QZO_Zg9Ust0j6fJJjXE7S9vTOCUjKPxwUYk-g,73621
jax/_src/export/serialization.py,sha256=TN6Z_coxIX9ZVfKqeKo26R8BsK_WseXjDVBI_H1Dlyk,38337
jax/_src/export/serialization_generated.py,sha256=iqc-M2s5AVrFVRcVFwYKkelwPaVBJ9hVw9clg-9Us70,51650
jax/_src/export/shape_poly.py,sha256=cwM6eGPIRR_HiCY25h-IgZG1Ig9DkwbjD4Lq1d5JNoc,87073
jax/_src/export/shape_poly_decision.py,sha256=PhFsP_dD5k1iVcQtU-hkHIMgqGcIJ67vYJWoIf9TMl8,20878
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=JwZavJJSZ-EpL3G8YfQsXh8foR4IK8d1wQYvLW0C3C8,1276
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=M5Qgta6dg0fnWUmrYkxZOzMfRIhI9LuLmqcY0frjopI,20955
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=CiNQ5026LYGwCnzF-7Rlj4uajlZwMKBAJRTRrs6bwDE,17182
jax/_src/internal_test_util/lax_test_util.py,sha256=m9r5u0wVPacgofl7RMTI2zs4kXDF989wLS7EnztSPIQ,13348
jax/_src/internal_test_util/test_harnesses.py,sha256=Lwy-E83HgOkYzjmal1eZWP3uoKw1QayiKINhi0GvHJM,126806
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=nrAZsjCv8G-Iht7d6KOdlMSTyeY2GJ0CHUDfEcp3v7c,34266
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=nIPmNYdRBZMALtXTGOishvfJ_YsmemmIH2XsXgmOArw,80033
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=fUNUEIgm6ekrFjxM8cdvtxiZf2kpsg9HtdGS0C4y2Kg,19732
jax/_src/internal_test_util/export_back_compat_test_data/export_with_multiple_meshes.py,sha256=YYxf17ohJxRupz6Eb0jHQXj_GUzN8bzonULHaqrzVOc,89037
jax/_src/internal_test_util/export_back_compat_test_data/export_with_specified_sharding.py,sha256=l7uV8ZZkikO-ECfmr5icPwfDz4CbDipTcZgdxF3y6Mo,28564
jax/_src/internal_test_util/export_back_compat_test_data/export_with_unspecified_sharding.py,sha256=MJDMzPbdjdMLpEstgf84kyWYiLpTDPRBh1wsJMHT46M,24402
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=aZpELM1Bn8mtWdrpJ8vfeGM1sjLgfijZbqIljqo63aY,94370
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=1I7GYt1PMBlDzkctBde5qawKPMP4U67YI8JwXreq6xI,139362
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=xbzvO3M53cpQNMNaB7dQUsCLCPaT4u7vfqdeIC_nKsU,50319
jax/_src/interpreters/batching.py,sha256=vTRhgIHvAQvMlyem37hPoU97PRUmFj70Ol5hpW0x6bo,33934
jax/_src/interpreters/mlir.py,sha256=mWHCvUiR9pWu2BOwRAspZYd_bL4gJTwr9_7H0sNHpaU,151493
jax/_src/interpreters/partial_eval.py,sha256=tDngx4hUPwwgnPRBd2-67hm7nzn4lQj5cxXN9wHfz_8,107307
jax/_src/interpreters/pxla.py,sha256=eKqJw6QQWR_yQLu0bgz3qwt8dnc6lDD7-vH6m-_AeYY,85205
jax/_src/interpreters/remat.py,sha256=6zTMT2yBCiBnduX0EWVNDAMttmoCmHKwOolFyd-EI1Y,9697
jax/_src/lax/__init__.py,sha256=sCX-9dMZXELEc0sJ-EQKbpBQMXl9h9WGa6I2PT1uoQ4,8802
jax/_src/lax/ann.py,sha256=yhP_H9ozhmAhVg19foKBBjt0pJlyoE565g1_XqrvXgg,17967
jax/_src/lax/convolution.py,sha256=yIVquMOpoam32XdVS1_jPJ4_UvNXCmvvpRMKgxACwxQ,51741
jax/_src/lax/erf.py,sha256=oQD8Y7n-ay0L3As9P88v7jJY1VFYxbgp59sjzRtU4aU,48923
jax/_src/lax/eval_jaxpr.py,sha256=uo-xr7waAyaCg_KyhSG9lhIvFpqRoWhs0ociuKmEPsc,7387
jax/_src/lax/fft.py,sha256=4cxoC2qGMhojmlFUL_YidyN1nsGuHdog6q80FyY6rY0,12157
jax/_src/lax/lax.py,sha256=AX-cvOn7eHJkMbiD7ecxDskdzAba6_A-4ROEvvJMDDE,427961
jax/_src/lax/lgamma.py,sha256=V2WhwZxMclGFxEPuvXZO5zRhdt0WafypP928c7Qg8FU,20855
jax/_src/lax/linalg.py,sha256=wyxKuQUi3rnzN61Cnr425q5D39wXfVfH0mxEgjPoumE,126218
jax/_src/lax/other.py,sha256=E1n6P-LXfYNbi9maGIDHRBvw994z6XojQvYRC2Kt-lc,13863
jax/_src/lax/parallel.py,sha256=o6o4MWi5dVUOFLeWOYe5uurwxIDmXuRuwfEDdB7LlPA,138807
jax/_src/lax/scaled_dot.py,sha256=e77gfCdyMxWybL-Ko4n6n7f1qeObAs-wkER8crovQ7k,13736
jax/_src/lax/slicing.py,sha256=j7z99kzkeGfk6LUTi-CUM3p0lrybqgraYg5EKohcFYI,156765
jax/_src/lax/special.py,sha256=WIU7rnflO40d8YIZVREeGGnRV1t2-rY8ajG76jjViAA,32897
jax/_src/lax/utils.py,sha256=jcOK0mGQl_i8Mm_VtrXqTPIqRtgqdJkalZZAz_qGmU0,17317
jax/_src/lax/windowed_reductions.py,sha256=jOtT6jBxGECNJF97bWYXzNtqfJIoQx_BzFklL7EuUsM,42495
jax/_src/lax/control_flow/__init__.py,sha256=JsuWyikJkp-5IVosqvubQfR6K1qXVeKMmSgNWLDzjso,2093
jax/_src/lax/control_flow/common.py,sha256=DV9y5jmjhc9vptDduwDFdZTefFNzzpXuUb06Wh56q80,6211
jax/_src/lax/control_flow/conditionals.py,sha256=huXMX3xgulJs9K5BjHEUO6XbEBsAeIX2WoCkUBqLdlo,59113
jax/_src/lax/control_flow/loops.py,sha256=9k7ieslwdILP6vs2V5dEPX1FHNNlc-sepRumwz5URWE,146725
jax/_src/lax/control_flow/solves.py,sha256=cDPW9BOsVaou02WhdJGig57dCsihTmBwnCcHOO0LKNI,25067
jax/_src/lib/__init__.py,sha256=2SrNd1zfiaJYqBGlJ_1rszgKSrKaVtiJrkMt1GE9bM8,7550
jax/_src/lib/mosaic_gpu.py,sha256=UBM6XbB1uYNCLKc-wgsMvfQgwsKB2QnY3egOim0M5Y8,1026
jax/_src/lib/triton.py,sha256=8GDkBIS0RyhwoFR7mGSjyzN8X-rZhBOaP3vpq8XWQJ8,630
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=tQ6nLvEjl7Luz2BtPQHvlLA1c2Hyl_mMD4cXlGoNnfw,50939
jax/_src/nn/initializers.py,sha256=rW969JPc2So9JiItesR2n1L847Rf8oIos7V8ccTPwso,28311
jax/_src/numpy/__init__.py,sha256=8gtl7Z9LdI_kSVnV2Mt7vnRg1R62x7cB6RVl0t1Pk6s,5752
jax/_src/numpy/array_api_metadata.py,sha256=BO66bz0vbbsMo25UQHzYg2OiqPkG6IPabHDbvjzENv8,3652
jax/_src/numpy/array_constructors.py,sha256=zJVyKm6KOK0_YPNJ_HM51J4tT51qXMDHxC9F-mg6Ay8,17528
jax/_src/numpy/array_creation.py,sha256=8g66OWuzP17USo22dhrab4qBFCYY3-mG0xTf1jNJu6w,33278
jax/_src/numpy/array_methods.py,sha256=N0EN2qT2d44KKJ6d5O72eUiH-83sWoUYKofiF2cMsIs,62543
jax/_src/numpy/einsum.py,sha256=njYqqbUXYIzflzteuy-ATuO8vflr4fFe1Z8u0qHOybQ,24670
jax/_src/numpy/error.py,sha256=kqoCUrAfFL1zmtVJfydI3MrpB9NNuYxTqEVmX4iidzM,5450
jax/_src/numpy/fft.py,sha256=bpoFtzhhvOTsSsHTsun3XZcnc6ITKciZiOit4baax0c,52329
jax/_src/numpy/hijax.py,sha256=R93mUGX522dnS23aEBIoS1dF9RcVSWoQfuhFwxLV7Cs,28822
jax/_src/numpy/index_tricks.py,sha256=bTlkOaAGNP35qBc1Cuf40Ts60G2YMaDaT0VPGk8OwDk,10028
jax/_src/numpy/indexing.py,sha256=he2FUq1kjNro0OcLKAOBra2Hp-5UJOD2KH4HYJHKprw,70070
jax/_src/numpy/lax_numpy.py,sha256=SSOo4uYBEsGVW38_BWz_EGI5T7BVt8oXwtlu1stXPuU,338153
jax/_src/numpy/linalg.py,sha256=fsu0L29_oCERzp8lbZ7TtibZxRkr_FbHCSZQp3oQLck,84108
jax/_src/numpy/polynomial.py,sha256=3LH4dWCGdBBwTelV_VfXYs_1Hugx7tBZTnq0bepL2hs,28617
jax/_src/numpy/reductions.py,sha256=ruwC6wLHsXLoN49Nw4o8Pgc6WSSVBzCwZnA-206J-RQ,123176
jax/_src/numpy/scalar_types.py,sha256=WjlDgTwmtssAWai3Ejd2sznfcuutMKS07sqLCNPx2A8,4040
jax/_src/numpy/setops.py,sha256=fGvIZ27FtKm04O3pukEl07KQKIehrruCH9er-pAtf_Q,48550
jax/_src/numpy/sorting.py,sha256=9OoxI3QXsRa7TCX9DkqX-sOtXANWqbe8tMkTy0hsfKM,24760
jax/_src/numpy/tensor_contractions.py,sha256=hVpzmSE2861nDbK1_A7cQeeDL8Hvck2WlJHfRE8W6pU,26265
jax/_src/numpy/ufunc_api.py,sha256=HBOi4VNhinbFZwC45G_epusQ3X_bw8bZCKjNJgTpo-Q,24376
jax/_src/numpy/ufuncs.py,sha256=hiXNIwBkBX6Ixg_jHjbvfygYYI1xGsLTmMLyJtRZ9Aw,132633
jax/_src/numpy/util.py,sha256=yeVvMhIevMdI_ofkDGfTMmvTCIhxKDmiyfgADBHgA0U,17707
jax/_src/numpy/vectorize.py,sha256=hRaOYzYMvVgmUv-z6TsMXEHOsFrZjGsX3aa5C7iBDUg,14228
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=V7vCrKoMk0rM41vk1Sowip9oe1XQ66WRv-fkli14DS4,19417
jax/_src/ops/special.py,sha256=zYKBMe29eMqKXDeaZ5BK8YehQdpAxAHPFKTmNjwKBa4,4380
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=V_KulwFu172ra_Cx5FE4eVEflWJysYPlo0eJZxa48e8,57442
jax/_src/pallas/cost_estimate.py,sha256=UpEC06rDhNIiMTLbLIa2PzFQx9g1_Rjk6Q4kFjzzOsc,8903
jax/_src/pallas/einshape.py,sha256=5waWuPP_XeJg5IOa6D8ZHpiihvjUL5N2oCX7ECP2Tao,30299
jax/_src/pallas/helpers.py,sha256=TLHE1ri_UquT0V5BlGEBgNCKZfahMOrjxHGD6cjI5Sc,10128
jax/_src/pallas/hlo_interpreter.py,sha256=Tp_HnkhesURwlFNNcVeynCPYrk60ii1qeTt8cVrJLcg,19707
jax/_src/pallas/mpmd.py,sha256=SOM77pnO3yE9kdSWsYUS-5-hGh4ds-WPVZIeZVfHBQ4,40476
jax/_src/pallas/pallas_call.py,sha256=48UQZRKUgioB8n_xeL4iRWLiiCRHCWklw5cOoglIL-Q,49712
jax/_src/pallas/primitives.py,sha256=Uqn_-ZzBy5NoYgdSenU_8fovChN_RxkMUnuEQFIVvd8,47756
jax/_src/pallas/utils.py,sha256=wpDWJLJv5o-Je7nedf5Xs7yZ1hsr0bI1VzjCkAhBtLM,17261
jax/_src/pallas/fuser/__init__.py,sha256=a-D_lbug8V-DJp85stF3Mll4j45YvVFed0TcktCPWK0,1547
jax/_src/pallas/fuser/block_spec.py,sha256=8hk3KQKNBsE__CCsMqbRLS3EVMTrJXFFrq7VYBpYblQ,122633
jax/_src/pallas/fuser/custom_evaluate.py,sha256=QXIkpbjIDeCxC4kJ5ZfD2gA-D04IDMxcQA-_nTHiCq0,3589
jax/_src/pallas/fuser/custom_fusion_lib.py,sha256=uHBL-eZC08U6UHdJcADUWKO9nAKXQ740mLCtFKBB3jY,10324
jax/_src/pallas/fuser/fuser_utils.py,sha256=i1n7OvkHmdfY5pq4kvlgcJjN3uDtOwH8HPcIox_KiVw,10619
jax/_src/pallas/fuser/fusible.py,sha256=0UN0MhnI-52_pKpXvqT8YAnBs7_kGJBSrA0mCVNsjNQ,6954
jax/_src/pallas/fuser/fusible_dtype.py,sha256=2IvODMUJHvJfsKnRjAc_SViB3nP0mqwVbbBzaGt-ZcA,20770
jax/_src/pallas/fuser/fusion.py,sha256=ntcEIIH49imrIZ_0_oKhio2b4Z0Snt9lQoqDoy6vhrQ,1545
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=yEAkiL6Hz7cSyvfItgAa-oY4JCerhBCGVLrF3NfM7ic,15359
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=SwAXp32Ouk7uLTTjbUll-qPvM4UGa6a96F6URy4vv7o,18831
jax/_src/pallas/mosaic/error_handling.py,sha256=LRKPwwB8O15JLnNJrb-L8UkDdQ4yLbONxlFgBjHrSyg,5850
jax/_src/pallas/mosaic/helpers.py,sha256=w_WFJ0_Am2Z9aCFEHXiSI8TjYJIvg1TovnFwUHZEo4g,7266
jax/_src/pallas/mosaic/lowering.py,sha256=ifCzvQ4xgsjTgbx2xiC5c6JB0__wdA4-MJ7ZhRKptTg,217760
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=Fj21m9rt3ZnYV_1L7aaIoJK9iBomlILEk2F47bZpfDw,28177
jax/_src/pallas/mosaic/pipeline.py,sha256=CKzC_eZLtt5o-xFVfBT1mwwGBxOQQWeT2AdORL2_qEU,106909
jax/_src/pallas/mosaic/prefetch.py,sha256=V1eGPinebMHQ5RNAchLzYUy-XRkPFQe382_zUE6nZwI,5951
jax/_src/pallas/mosaic/primitives.py,sha256=isvQxHknI78Uz6nAS87KkSzcL9Kgk-K6hf6_a8WDAbY,62562
jax/_src/pallas/mosaic/random.py,sha256=rFLIEMScXO7ZaEFKgZnFosT-7oyBXribRG4CQ0qRSIQ,7958
jax/_src/pallas/mosaic/sc_core.py,sha256=i44BLjrnKFn-VNRB_CfZQHc7L2V2cQnVj0IG0r49mds,9638
jax/_src/pallas/mosaic/sc_lowering.py,sha256=DOqmKSttIb7bDq3vDvs0G-Dhvx9Hp4XfmQ7YipYgA1g,36994
jax/_src/pallas/mosaic/sc_primitives.py,sha256=N3U_wmoVbUIkhOZeZbGiOMvey3hjpvSJp9xihPE_Su4,48406
jax/_src/pallas/mosaic/tpu_info.py,sha256=NN_BUuyq6MQuPAADzOlDGeeLro7OvtY3VyndWEF9G_8,1163
jax/_src/pallas/mosaic/interpret/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/pallas/mosaic/interpret/interpret_pallas_call.py,sha256=dp7srY0A7wuteGG1WkjJ_4sUqw_JAzv5OXCA2DDT81Q,92862
jax/_src/pallas/mosaic/interpret/params.py,sha256=uYhiY0c2akxqivjHyA0nEWGslNLQgXkgZYmD9l80dUo,8886
jax/_src/pallas/mosaic/interpret/race_detection_state.py,sha256=Mf9hoVBQn_0hWhae-H69J7yuatf6KxIxO1DxSH05TsY,6592
jax/_src/pallas/mosaic/interpret/shared_memory.py,sha256=QXhyzfgUR9139vYABID2_0G5kRH76BHDmZ2b5RIjOWU,38173
jax/_src/pallas/mosaic/interpret/thread_map.py,sha256=mFsaeAkd6WcHKJRdEqR9qjfeXXriN6OVVTrfV9oCzJQ,4643
jax/_src/pallas/mosaic/interpret/utils.py,sha256=_BTHZROiH3LNIFf6wqnMd6jstZNrJ7j9fOXmC9tPm88,15797
jax/_src/pallas/mosaic/interpret/vector_clock.py,sha256=GvxQkJzFLltL2y8ofrnlSNEXD0fuAow21wHFXKsjEv0,1690
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/core.py,sha256=DS8OCu9AubMo9AqjuPnl_stxgLV3NU_vdYlBjhU_mbE,80262
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=7dFYQNUihDp0wnWgPE-eeXgkhA1lfVT05t5sZoEARGI,17362
jax/_src/pallas/mosaic_gpu/layout_rules.py,sha256=YaB92GfsRfsN_yzvSkRDYycqeSnEybgcDYBADiBbJNM,6899
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=caDb6dh068_DTuWgR7SmCho8WlAvt3NIpuHkr3lE-dk,172062
jax/_src/pallas/mosaic_gpu/pallas_call.py,sha256=ULZAnLwAN8B0XWKm5Fe6jOzd7Iu5VlOCgQZbXxozVi4,4208
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=TRtgBHeTfhW-cK21pPOTEWVfDJLpWdLiWRShqbftbnE,6889
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=u_jcD7MQ00QgRc4HkTYHi_wD9JLZF_uu9gRDQn4mthA,46742
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=-X5mJGzhw-IIxgJy-Ungox8qgNecGuzaHAvpNZw5CgY,222273
jax/_src/pallas/mosaic_gpu/torch.py,sha256=wD8WcxT4CLIEjndMi1HDK-uCl3y9a22kln4T0y9TvoI,11297
jax/_src/pallas/mosaic_gpu/interpret/__init__.py,sha256=-aCGt5dPNdKLi-ALblRdGErT_9Or12mpWJ52yApKN0I,581
jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py,sha256=S00onPMPUwWUEM966FED3IKAjb_8Y7KfDv7va5IAEGc,72339
jax/_src/pallas/mosaic_gpu/interpret/interpret_pallas_call.py,sha256=tD8J9Oy92oexZCzgdSTgqeHGGkS_qTpJurEeyDXDfXo,22369
jax/_src/pallas/mosaic_gpu/interpret/jaxpr_interpret.py,sha256=koDV-S_eyo3QSjcjSBaCOnuPWGnBkUcCgF8ej8H1McQ,57398
jax/_src/pallas/mosaic_gpu/interpret/params.py,sha256=SBSKpx4xZwiUXrjynhsPxdnUNWXR1YmwrRAejCkzRPA,3935
jax/_src/pallas/mosaic_gpu/interpret/race_detection_state.py,sha256=5Q1docMkkL7id4EOkyFSneRBISOa8dkxeAqsJ0FPf0A,5041
jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py,sha256=s4dQ1BYO9WWPUy0NTmE3yEttvUJMxkzvv4tQiN8GESI,63238
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/gpu_info.py,sha256=bmIMRTdBXqrdVvhsuy3xgrq5f7aP98As0EcovTbHXYY,7056
jax/_src/pallas/triton/lowering.py,sha256=Uh6K5KX7mWO0dSBK5pXTKLT3cTurRrdFxtRwfjqXurM,103506
jax/_src/pallas/triton/pallas_call_registration.py,sha256=tuyaZeAjPVwWB46KVVMGCuSupKHEowJJq6rQJyO-DKI,5892
jax/_src/pallas/triton/primitives.py,sha256=5CICEZu7trZQfzsICtSte6JyyDMnyk9Bj2a6Lvt9oRE,21381
jax/_src/random/__init__.py,sha256=hh0pFqUQA8XBIK1PE_LAS34NgkZnJ9Herk2JS8SV_0Q,2627
jax/_src/random/core.py,sha256=1Arsuv6CxXOSKsjS91m6QtILD_CBfG4ss7fFft5tQdo,163756
jax/_src/random/philox2x32.py,sha256=QYDZLKfVZ3whc6f4BseJavFFSMFksVB1OoCMVx7g8kc,7776
jax/_src/random/philox4x32.py,sha256=gsVnINWJungyM68DNzF-4FrSKaSr6mDz4ZZxpxphDeA,8878
jax/_src/random/prng.py,sha256=xjWaUZWne6_OBgS0WJEV9_UBQnoOMhRkbfxiWcUVABw,31204
jax/_src/random/rbg.py,sha256=zm17I3It5iCanKMZ84yOexXQpLSathpXZsC06VEcUKw,4122
jax/_src/random/stateful_rng.py,sha256=jo-VL1Udb45LcYjNI5HSq6hwTK4xXIglRg_GK0d49bo,10474
jax/_src/random/threefry2x32.py,sha256=0G_uN6WRmnxsJ6gCvhSo0xPnn1uaTKpwPS2794hIig0,14314
jax/_src/random/threefry4x32.py,sha256=CggHyOgiBqyO3SNueEIx0VGvymFZmzrvDduMrv6RCRE,10883
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=LsUgA1Zxsk4qiAZ2kAnwmNZx2ealZxPazHO8NyPHiDM,17320
jax/_src/scipy/integrate.py,sha256=vHyXnhczC0l-Q7ZUWQOIakc1Knqlne4MMShb1DPIHf4,2334
jax/_src/scipy/linalg.py,sha256=4OgXOlC2iThwZuO1WPaNinZTjj884NskkD_W-lRGyAk,121643
jax/_src/scipy/ndimage.py,sha256=z5tuwZl6HbDqf8nmDKRes_8M97NABr02gK4DWN9Hnjg,6948
jax/_src/scipy/signal.py,sha256=g8pxxAQ9lTfejT0tHYB3hI2R4YfDA4uhrOtyiTeyNSw,47290
jax/_src/scipy/special.py,sha256=jfrBsdm1HtX6nXaHPmVtrHe2neAYuz0PsVOmMQiKUXY,132462
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=n431SEm2svv-KWjN2sdeVpV2HV8lwSIY8UYhwzRdGbs,13486
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=d46t9j28faor-9Tcn25FscW-GNjbgmjYP3gL3RNFQEs,17634
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=N7aWuKeWmwCoZjV0DPe08ycdmONWI0l6eNKwzxyYpDg,10855
jax/_src/scipy/stats/bernoulli.py,sha256=A6TUtOXfIR1aDxu-i59Ry_VBsEVQ9WBhtdi2L9yN2Qs,4500
jax/_src/scipy/stats/beta.py,sha256=jjRE4LlyBt2BO2jmmN1z5PEasjl4LykhKTNc8846etU,8026
jax/_src/scipy/stats/betabinom.py,sha256=e01yXEhvPVAibWwpxteNon_PnW7kL5mnCWnbRieGOGs,3514
jax/_src/scipy/stats/binom.py,sha256=KIydYZOcwN109i6rodNR-Lda-cog16r2jrQcDarUeeo,2859
jax/_src/scipy/stats/cauchy.py,sha256=hJxpFNPHr86knTjXNqkMYqjuxiB35m8Ig7FLAyNQwWs,9273
jax/_src/scipy/stats/chi2.py,sha256=M-ZBULVHOakfkNlBkLDRlLlcYQbg3ycuFQGFwPgTEAs,8041
jax/_src/scipy/stats/dirichlet.py,sha256=G9FAdSoODRw-lZkZlGsg1sq2bbqMLZ47wG8P4o_oaow,3237
jax/_src/scipy/stats/expon.py,sha256=IsPRFkP7SQKQIgMbQpWAd71w3niz1Wh96tkX1KrAH7c,8134
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=bGlj98GOZBylcVLFxqn4V2cfUDwvczn0wiVISFs7t-M,10462
jax/_src/scipy/stats/laplace.py,sha256=B7maGyE9v3s0ai9o7mwIFwaDKrJj54KF8FuRZ9iToi8,3324
jax/_src/scipy/stats/logistic.py,sha256=DRV7CEGAdzV4CzEKFOTnSyXK5g-WVx_rt02krU-x0q8,6244
jax/_src/scipy/stats/multinomial.py,sha256=h_INEuZ01Pv3ZjCa9kfi667aJlVB5l43v72d4nhu9P4,2612
jax/_src/scipy/stats/multivariate_normal.py,sha256=6_Ncq9FvbEC8wbw5tCzt9k-QLZIy_iXryzskGfKNSE4,3549
jax/_src/scipy/stats/nbinom.py,sha256=O5dnvw9yt3VCmjctRlzc7mq9wOPBjBKqyJIk-XsoRE8,2637
jax/_src/scipy/stats/norm.py,sha256=WfaoFZHNC8Os7edpl1Cx3hoquO_V2dv5S-NF5o9Y8UM,8462
jax/_src/scipy/stats/pareto.py,sha256=SMCQZf4ewVElu5aP4fz0ZG1drarIUN_qmG0RKoqDqzQ,8922
jax/_src/scipy/stats/poisson.py,sha256=ulpmmnckI2pbH_ABljfK5u9X9rkj4Kr4kkzel7lYHDo,8161
jax/_src/scipy/stats/t.py,sha256=-POac0PdQa3tNpgksvZaVRHYrrbQQrlyXj6l8iP15gk,3226
jax/_src/scipy/stats/truncnorm.py,sha256=ovB6SpJb9wTF8T_nz1Vy5O6z3SBYu-GyyJi4UT16WlY,9787
jax/_src/scipy/stats/uniform.py,sha256=KxP4IfZuxXqLcaag5G91lxler0J-jJRsBpxNse6bU-Q,4287
jax/_src/scipy/stats/vonmises.py,sha256=2EN54QHM06sXmsCO82oh4YubBWghkzfLk4a06Fi0qxE,2513
jax/_src/scipy/stats/wrapcauchy.py,sha256=4vJw9eZn5iMgzpR3iJmuFhFq0GTCqQm3P___4UB-czQ,2401
jax/_src/state/__init__.py,sha256=5pm26HOqMHKV2-5JeAE6oa7wdLcu6IaVuIoP61PZv60,974
jax/_src/state/discharge.py,sha256=AOBkAWa-xgZrUX2GtS4sD20kLPCQzeTiQystAzObJ90,42094
jax/_src/state/indexing.py,sha256=DffbfaGEzU77a5j0q9MkXxH6V6ZUyyS2yqrveFjMoEQ,16913
jax/_src/state/primitives.py,sha256=sCXPOSvOEDM1iFUrkVUEO8sH5qwBZpmLKPxvq7eRDIo,52980
jax/_src/state/types.py,sha256=WM50-ofb-S_bxUEgFCD1_lfJWCzkgIsCGVABMAGP1Kw,24540
jax/_src/state/utils.py,sha256=Wnjr2fazNdZMJfqGp-WtZdZSKplZyfK8BcICQHh4-xs,3826
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=bsIkAVAcfQm0HbPQZjhMNgmcdBSjMb9k4fb2eYMeR0A,2619
jax/_src/third_party/scipy/interpolate.py,sha256=7A5nuFfrGRG8cFM2KU_7qRpsdbN4YpzGhc7esSWopZs,6800
jax/_src/third_party/scipy/linalg.py,sha256=jgDDGByArWWjetmCdX6LCdzPT2Ao43TCe56syYpBxFw,3878
jax/_src/third_party/scipy/signal_helper.py,sha256=DxAw44hcufCKd5bhOD1jz5TpxPBr6wMhjAzO3p4edwQ,3572
jax/_src/third_party/scipy/special.py,sha256=jrWq2tDIBzeYgOeHt-XFXg6VFMl4c33lpvciu2jTSoY,9415
jax/_src/thunky/__init__.py,sha256=-aCGt5dPNdKLi-ALblRdGErT_9Or12mpWJ52yApKN0I,581
jax/_src/thunky/thunky.py,sha256=9xoAJ1fp71GIpzqhdcmDxs50qaIgQPpFfProcil7_6Y,70051
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=FUmUJ8EXXNSVhgCP6l-Ffc5oPA-MIYKS8LfADSs1qOc,26676
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=k5vIQP5EwKd1aQs-VhlN4Q9zCdQD4ooMdLW9fWdhmko,10663
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=RoUjVf16M2DTMfdWhvc6TcX55T11gLj6XeCowN7v8EM,13872
jax/experimental/__init__.py,sha256=HGWeZYGRKgKmpBJEQy_KD4sYUoc9-MF0y0hTjCppG2E,1240
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_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
jax/experimental/fused.py,sha256=PGJPfeeVRq-bJrQjn6UMyM5ZaJoj_RyzZK3rCLV6X0s,7568
jax/experimental/hijax.py,sha256=CZagI6vR6BzoSGcA1qVyt_U-j_JAcxDAMl_eUPUIjIs,1670
jax/experimental/jet.py,sha256=Ui8z2Ph_wqQfus044xJyb8UJUH00SPl8MKAFoo27M8M,28310
jax/experimental/layout.py,sha256=QQo8aBUEn5AKSNb6QWWLfnVFcavXSKh0xTHkjBlhGSk,778
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/multihost_utils.py,sha256=VkphseESHu7l4_HWLNW9f9DXQM3myGzfk-bXHRhLY8Y,26563
jax/experimental/ode.py,sha256=c0zlLkhxIfgZCDJt2XnChOUlSI-ItQrd8zLtvSh5Im0,10627
jax/experimental/overlap.py,sha256=HY9J5tmO5ta7yHJt_FRtNL4WhlPxAcm3t4g_LmNE6e8,644
jax/experimental/pjit.py,sha256=g70rw2xUYiip7Xe10P5OqeeNgkdSO9Z425uTTWJDOJ8,1132
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
jax/experimental/random.py,sha256=OheWK1WzdbKbMTU_QtKo0VZvHwknjV0eZXzuSt5OZU0,728
jax/experimental/rnn.py,sha256=YmDCHCMwh-CqjkumpHD7pQF9ekkHCEgP2Ob9htBhdsY,20117
jax/experimental/scheduling_groups.py,sha256=qFNfrEioPSEvVfF8n6bdgeKWNC-rRBt-3uKn_z5tV-M,713
jax/experimental/serialize_executable.py,sha256=nigeScAfUlGbpsADFDojqFJm9MWLkaGIzpQ3yQvGKHg,4967
jax/experimental/shard_map.py,sha256=daJk2CGjpmspNgkUMVCBvEosmbJ8BxgcGvRMMFwAPrM,1472
jax/experimental/thunky.py,sha256=ONWH0p7quyKc1W9SyCM5eZhfbFrwiWwnqS2Z1uBNmhQ,1834
jax/experimental/topologies.py,sha256=ISxVhyAC0vjAUzbPTCjRWHmSS_w38ioaSCVaz1szj7Y,3272
jax/experimental/transfer.py,sha256=xB19nADORbWPF9lQpN6hhf2e6eV6HwBBbrj4hsBD6zM,2912
jax/experimental/xla_metadata.py,sha256=XWaiFypKPpsAe8gtILBLWWPEsq2YOaTZXMOIIxbp8kk,4154
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/pytree_serialization.py,sha256=Q1xp2Ucs-976jfi49k0Oh7KBe2kOSdB2XecSzkGw-1o,21899
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=Xqaoqlr4lb6nPlPENiTtVqNEu1myUGZXNLOC3e0iQxc,2688
jax/experimental/array_serialization/serialization.py,sha256=d-DBlH4PJCFO3nKgbdmlxt9X5Wsg9XKWk3cG-rqdYzo,13515
jax/experimental/array_serialization/tensorstore_impl.py,sha256=e5Zjd41C-X5FZFb5mjdQbHwhxWh_xfhIkYYW1yb46B0,25586
jax/experimental/colocated_python/__init__.py,sha256=wZwEqaKE8AMWBRS0Xp9BubV0YXIAUWp-FQZv0t8gdZ0,940
jax/experimental/colocated_python/api.py,sha256=Iw02uQfoFrXcDQUra33-N_WxAHj_scN2_qW4KvgKhvs,6998
jax/experimental/colocated_python/func.py,sha256=zgixzVrYzq91oB8Ta7x0csgSlC1nUXcyvHvylcm6VBQ,31327
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
jax/experimental/colocated_python/obj.py,sha256=iGUxeIMg01jQzKT-RMiHHl4sl4EbiM8TRnkCHZKFv48,6577
jax/experimental/colocated_python/obj_backend.py,sha256=4tbCoNp69yOVCIP4Nt9Wu04vSgRQxPRp1TuoQ3jHoLE,4054
jax/experimental/colocated_python/serialization.py,sha256=DpJ9fw_BDmkhQ-D3v3NXz58aFSnXf8a-B7Z5SCWauGw,12648
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=SPo9xcmW2yND7JIKvNjGck61Rk-TFVrLUCAqEt6ayes,29234
jax/experimental/jax2tf/jax2tf.py,sha256=H0mFIq58-mUrPohhNQHu6PJTMN3lZroQKxV4Z7C5otQ,38234
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=J5VUjJw31j0RN2uFFGElLcUAyv03QHpXw_SxqL4yXtY,23880
jax/experimental/mosaic/__init__.py,sha256=djMqZrdvyCamCGea3mRbUsZa_4sszOlnulsFKXXZ74c,1337
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=H79-YJoMGF3jx2VVQ_g2QA297cHs-UJ0J8nb_Cn9EUw,4392
jax/experimental/mosaic/gpu/constraints.py,sha256=_Yhezfm4w5r5eQFLMUTL7_UnYr6bNTfxMpwUoV4M6Cw,53986
jax/experimental/mosaic/gpu/core.py,sha256=nIyp_eZcHoHq9NgnyYQv7rE6WUKnnIiblzaKK1J6jnw,50453
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=svqSjH8njDb9O1dhPKv96fTBv79lHP7p9Dm8nv73HMQ,108268
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=B_OYpsYc028Um7C1HVa1dEd81S8ryG2HhKwNezLOf0c,247474
jax/experimental/mosaic/gpu/inference_utils.py,sha256=XP0eClB06JBfUqY7cfi7U071zbXC7egObBaBDABDmtA,6983
jax/experimental/mosaic/gpu/launch_context.py,sha256=kiNz1Kzp29EfgaF0Mt-mXQKhZcHfX9KQnTazfYXiYLw,104455
jax/experimental/mosaic/gpu/layout_inference.py,sha256=UCwlf342ZgZbRj5Evc1M-taj7IIwqeOaOJN2IO_DDe8,111982
jax/experimental/mosaic/gpu/layouts.py,sha256=fPsxfV9bv7RCD8Gab5IwvydWDe-c6UszuRvawm5HdkQ,6587
jax/experimental/mosaic/gpu/mma.py,sha256=_b877xhWLVZoFHikozx5Qu8kRMMGGVp8E62fJ0DAP20,11354
jax/experimental/mosaic/gpu/mma_utils.py,sha256=7yjGdni0MydHAfwKc4lBs5moxz6WGNQHJrc2kEp7ejo,10748
jax/experimental/mosaic/gpu/profiler.py,sha256=r0l2TvXZvVuSGOoZtKf9O1xAILgMrmcMNjcaKFpSY_I,18052
jax/experimental/mosaic/gpu/tcgen05.py,sha256=6OXGeiE931RfuLF4xA3OjOeKVuCJ06nhbKM3KhcrXp0,86119
jax/experimental/mosaic/gpu/utils.py,sha256=rqI8uuHD-yeJBB8mkb8rVEhfsUQ20T22pWPaciqt6og,92731
jax/experimental/mosaic/gpu/wgmma.py,sha256=gBwsEW9ILF5CIkfV_vLDSWRzkD09MWgF1vixJTR4oTA,20115
jax/experimental/pallas/__init__.py,sha256=UebpaUOPRL-tP_7IpY52vcZpWIFMUcdbJH0sNQx7HQo,5985
jax/experimental/pallas/fuser.py,sha256=ltpmwcG8UeHh85OrjrT1an7kw791shQNc8aBq4RrkVc,1894
jax/experimental/pallas/mosaic_gpu.py,sha256=4VJ6zY0N64szlItF5D_IKP2HTo_1SYmOKBujnuZykuI,9819
jax/experimental/pallas/tpu.py,sha256=JjjOaEjM0rC4VpSGroxGAMz743nojpLSr9O3HBFSFGw,7384
jax/experimental/pallas/tpu_sc.py,sha256=tjqz0l8uXj2KAaOSqbxuTCDwDRVP_Zcm4Nq02bCOsho,2719
jax/experimental/pallas/triton.py,sha256=LCgjTSVCdTqJ60FJ6lM_1wcgYPUYS6zR0Ijq2fmH6gY,2243
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=RSKrLu2ZytrI0v0o4hudLnYP0lINJkCrsfjxDAeoB-Q,8975
jax/experimental/pallas/ops/gpu/attention.py,sha256=c9tcicTRj9AS7c4p7F1_hC2M3cvIsjqrGpkcp9siIc4,23487
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=nQKTK4-VFlbE_ngSHKvHBMd4XkEQqCO9ShzPsiLdAlo,36570
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=L1TN9eGOjQRl1esjnRSoUK2t5sIPIcuytgEvVQmngrw,12229
jax/experimental/pallas/ops/gpu/blackwell_ragged_dot_mgpu.py,sha256=gDc1S8cK4SzjOsRfaVyfgNACbsekgk__RRE2cUpiXPw,15818
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=lLri-LPat9Fm66cZjqEWnX2ZSAaloqoVl1HcgvBjHJ8,11298
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=ZRaf2dPP-nov-6JZAoHF4VsrFPvXYetyss_4W9XV2iM,17317
jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py,sha256=KNhxU83T0DRRgNmksmGcMvdx5bfx9LWuXRn9dV36hwA,12694
jax/experimental/pallas/ops/gpu/hopper_mixed_type_matmul_mgpu.py,sha256=spBIoFHSji8UFtg1z5p1kTXVlDYI9NcpfoHqa9uVQ94,12736
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=gTsWoTx-8822N8Vo7ymycdKGzVuWYd-4_pjx8QsP2gQ,11361
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=ANqN2HeOJBO_9vjtPL7DEPsAmQvKM-fOYUi8ehSFVZY,16308
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=gXSAwzXCHsH53d8Dc64jFhmD44NnXKYgAKRe1RgJoEI,12175
jax/experimental/pallas/ops/gpu/reduce_scatter_mgpu.py,sha256=Mku8kameSwtmGTfAANTieUjgSz2O_MeiYLoMmMIG3TQ,9811
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=gzeLnkSuDw6XfT28NevWq-SFhrDkEH8fAX96TmOuOKI,10205
jax/experimental/pallas/ops/gpu/softmax.py,sha256=27wRZNf-lPr1qbM8bS6B7bPEEmxLpMF5QnUspub_Ei0,2873
jax/experimental/pallas/ops/gpu/transposed_ragged_dot_mgpu.py,sha256=sJ7rrbU9rkBDEe-2iGq02frrZDuw58HQ7SLXTtWyEUc,10135
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=TQA89uTwpQl4UT77U6GVfe7r7_-_JqUDcXDB_0lF0cc,5355
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=cxXLqap7-eDp-SRvS154bncP7Uixro6PjKPWguNgLGM,49865
jax/experimental/pallas/ops/tpu/matmul.py,sha256=a7HEgbL8LGoQDGJY3jaMnFrpzoufJhWYEal-e5LwZNo,2728
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w,650
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=4sHd4dWGgSYZ5HRqgznZtj0pGdBMr2CB3EJJNw8WD6k,2122
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=gXaBx-M6omIV_75lciicUCFB1BlQAAm2Es9YGroMjjk,27217
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=VxBMIVOvY1XnaHVrNaIz4qa0JjoNbgPgFCRiN6nr6gk,2925
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=BQmfFyeB4nDxtq3P0NHAdQfUBYoFZYU_YbzSNk0UTG4,21888
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=XjArV6NQLT5U7kUHKhGejeRYi-SBU0hGean1o0xApeE,2578
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=zknHb0VaC63u3G3lRr4WRqnRwD_ZpgAgpRNaGjBKAME,97915
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=76dL8VuI0CfEDZrVwOQc1Hqkk2DFcxDSbDkux5NmKfM,7644
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=5r3pl293zG_WWLzWkY5WKP-lxqFmLxwGx9hWu9OHiX4,1780
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=ZWZMUa2UIjqWrxWlQ9KhvEaUflk7GjzzsjIt6qVvX3Q,4438
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=Ab3LoCcT7LszFkbv73BkBJjm6T6AXZff--CYt-_XLXA,78667
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=Z3pbdqmibifMZ31fIUIEvS3pfF3VCwQczH_ZUu7o4tA,16285
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=zJ7yQeI3dmWIA9aZgnpErwS4KB2jHJDGvKboWOGTPks,40393
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/roofline.py,sha256=a_jVgWsH0gUU7d2KOUuxwAcm4LKODwxwU0e_JV8bYPI,12028
jax/experimental/roofline/rooflines.py,sha256=F4M0HrAkxfBlmonABvHejvzmwQHGK6p6kdRipVoOkNo,28578
jax/experimental/source_mapper/__init__.py,sha256=ZGiZxIuIkYEpyCw8cMv5bHSCC5MIBr0Elf_Ig_wArYs,1632
jax/experimental/source_mapper/common.py,sha256=1_sE6BmKFz4fr2nCZybC5t2adjakSDkZsVmJuUiPifY,2500
jax/experimental/source_mapper/generate_map.py,sha256=k0SrNKPQMAG3fgc_Pw30Vhn2u92nTkt-BbjyhVDs52Q,2129
jax/experimental/source_mapper/hlo.py,sha256=mIheqhdp6H2aTn7hjqwD5QjHQAf7KqbtSNJDnpw_Fik,7219
jax/experimental/source_mapper/jaxpr.py,sha256=0IMw3kt1UwmZO5AJxanYy2tDzPan__s2rtIQa_ykG4U,2610
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=e2FJOBxWAnG7DQcyg09rYUEMs88SS9lcy4T04HQ93j4,3336
jax/experimental/sparse/_lowerings.py,sha256=UWISqTt4FkrLAzwMe6SK1DJgJJzXSrV-ICedFiL1wyo,12832
jax/experimental/sparse/ad.py,sha256=kAM6UdvkNzKuZ_CeAQf7JIg15SlJZnL60h9RbmJ2a78,7941
jax/experimental/sparse/api.py,sha256=X2YePcfk0652J8z0pc11-gPRIBfl6438mlcvB0-ILIA,6641
jax/experimental/sparse/bcoo.py,sha256=LlseWCtaJCRjK6v_1OUhAOHnrt3OqxE8i9v4vhSnObA,129479
jax/experimental/sparse/bcsr.py,sha256=sWBeaOUOyv6OBkpjlaPvL4rm3A3Fjn8sSq0g9LuzqB0,41264
jax/experimental/sparse/coo.py,sha256=8OKWb0lEzLrl-SostqPExyO19Q41y1o_eZQK_DAj8tU,23964
jax/experimental/sparse/csr.py,sha256=70rA2SXv5CPQ6LvV5YUipOcEhWhx-wu0AFN_TM4hUYQ,24292
jax/experimental/sparse/linalg.py,sha256=qQBnen6-tLmc6ftHiR5nqCDwmjp-LrVJLOy0FpvvoFg,23493
jax/experimental/sparse/random.py,sha256=_bbiUKD0gTdBhtNU6G7T3mV7doM66gz-tFiSgXlblls,4333
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=AgzVbgLsQMn0YwrWogvQN3nK67o-7IV9VbTcni9RarI,38888
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/extend/__init__.py,sha256=-hZkYp3ZGP2L4-2_9NB-4ekYFcyjtpYTHRvVJAeRaKg,1670
jax/extend/backend.py,sha256=QOL10ZfJMgYh7HWGYPQ1p8y0RD3KEHR075OnsOb_SP8,1370
jax/extend/ifrt_programs.py,sha256=TCWvR-K8nHQ9PuUNoMWVAfEuYsutTNrefKwLg4AIRao,787
jax/extend/linear_util.py,sha256=02ixe_vuJbwdBvRNcpQGG4DtE_6cPbIpFCzpus8l1oE,1601
jax/extend/lowering.py,sha256=nVCHA5MZUEKU5aPUb1b6d5YGdEr5IZokRsLunO7olRQ,880
jax/extend/pallas.py,sha256=676BusN50DcftzgrGqar0G9mCCLvR_kNGTukN85b07U,834
jax/extend/random.py,sha256=upxZOHOGTZrDEmQfZrI8CYr054f_7zEZMluNYrabZPY,1171
jax/extend/sharding.py,sha256=GRpNqvTo7-_NhNk7j9JW3HXE8_WkDocmnj07hfFp60U,1327
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
jax/extend/xla.py,sha256=6vshsQMLWfD0pi9jb3XiawEtjLjinZyoRUYCAE2Aakw,968
jax/extend/core/__init__.py,sha256=POgATGL_MXMf9bTo7zTD3Yj5f-TmKh9pB7NrnBsdHS4,2361
jax/extend/core/primitives.py,sha256=SEVvcR0mmDw1r5jBZCStY7dlV0Cr05CwyvJMXw34Wj4,6695
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=0zs0fI80iJ0F_utFP1ygBnIW_xC6xmYQCLMKReHZQ3U,1478
jax/interpreters/batching.py,sha256=CKnro5kKE3fJXlLAJ-lw1wEZ9xRfDbjBVgtk0Wo4N5Q,1276
jax/interpreters/mlir.py,sha256=CDxW22rlBt3hx-i8Pb_9izwx9b_hBnkhIrusjYuG2L0,2823
jax/interpreters/partial_eval.py,sha256=-VkTi04cGSYzeLU2yQgsGbH2o25V-D8P65kqbIGwwuU,1614
jax/interpreters/pxla.py,sha256=Q_UFZ3EnciG8jGzbWrK3xmUg0soh_5SLLLtQrYcB--k,2341
jax/interpreters/xla.py,sha256=6nJ2ku4ycSMXfT-cCYgdxbfDgImPKVk_P5p0n9imQJc,883
jax/lax/__init__.py,sha256=GacmkfPJQ1HZPcoiJ8zxHTXq_WN6dFMWiTeqdUIapMM,12291
jax/lax/linalg.py,sha256=NVBRgXWUJtmL37gjy13AtcU4nNvE4dIsMUpJnoktCyE,1757
jax/lib/__init__.py,sha256=i2G6ZoULU2uH8HBmxui_CTzii_5bW5ibXbHPlIvkONs,660
jax/nn/__init__.py,sha256=G9v8E63aJYY0rgGMX_bTRB52uBR8a0qEkDQn92zA_z8,1782
jax/nn/__init__.pyi,sha256=6cT3nbawcW7yGmZFX1vRdnlNBxEqhfpV34NWKdRhD7M,5172
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=FZ-f2ytxdtK2tmlsLsRJgYoXKLirlRYfpq8CHZUx1Sw,12533
jax/numpy/__init__.pyi,sha256=q7qxmjv1DEN2QQUXDoXJayI8VggjFNDRmvnjmuC3zFs,45939
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=LCGoDRGvc-VwI1k0CTbOuEmfNRqd4oUfnyX1WA5q9QA,1752
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/special.py,sha256=deNAFVrZx9d4hYxhLOYH8iPbxifS3oddAyTQbiIoDqw,2578
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=9DOHrGu7OpKwIKU1ViXEsDa33RaN-WDZPrAKAkb16cY,9185
jax/tools/pgo_nsys_converter.py,sha256=C2gz5kMqZRqI8t11HOswanwcGDlrAHF_PTh-p4hkVrI,3407
jax-0.12.0.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.12.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.12.0.dist-info/METADATA,sha256=yRswnEedWIPD4m18qhV7PfE9fq0oupKFT1XpOqmz7ow,13262
jax-0.12.0.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
jax-0.12.0.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.12.0.dist-info/RECORD,,
