jax-0.10.0.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.10.0.dist-info/METADATA,sha256=8jBnA4K1bYYOfJdYSKMRr-RYARewfzSqglrux0J8M6o,13257
jax-0.10.0.dist-info/RECORD,,
jax-0.10.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
jax-0.10.0.dist-info/licenses/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.10.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.10.0.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=gxfSiEWARUtn7DNCvqV8Pz6DyQJ3Au5oLkEN2wmpytI,9727
jax/__pycache__/__init__.cpython-312.pyc,,
jax/__pycache__/ad_checkpoint.cpython-312.pyc,,
jax/__pycache__/api_util.cpython-312.pyc,,
jax/__pycache__/cloud_tpu_init.cpython-312.pyc,,
jax/__pycache__/collect_profile.cpython-312.pyc,,
jax/__pycache__/core.cpython-312.pyc,,
jax/__pycache__/custom_batching.cpython-312.pyc,,
jax/__pycache__/custom_derivatives.cpython-312.pyc,,
jax/__pycache__/debug.cpython-312.pyc,,
jax/__pycache__/distributed.cpython-312.pyc,,
jax/__pycache__/dlpack.cpython-312.pyc,,
jax/__pycache__/dtypes.cpython-312.pyc,,
jax/__pycache__/errors.cpython-312.pyc,,
jax/__pycache__/export.cpython-312.pyc,,
jax/__pycache__/ffi.cpython-312.pyc,,
jax/__pycache__/flatten_util.cpython-312.pyc,,
jax/__pycache__/monitoring.cpython-312.pyc,,
jax/__pycache__/profiler.cpython-312.pyc,,
jax/__pycache__/random.cpython-312.pyc,,
jax/__pycache__/ref.cpython-312.pyc,,
jax/__pycache__/sharding.cpython-312.pyc,,
jax/__pycache__/stages.cpython-312.pyc,,
jax/__pycache__/test_util.cpython-312.pyc,,
jax/__pycache__/tree.cpython-312.pyc,,
jax/__pycache__/tree_util.cpython-312.pyc,,
jax/__pycache__/typing.cpython-312.pyc,,
jax/__pycache__/version.cpython-312.pyc,,
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/__pycache__/__init__.cpython-312.pyc,,
jax/_src/__pycache__/abstract_arrays.cpython-312.pyc,,
jax/_src/__pycache__/ad_checkpoint.cpython-312.pyc,,
jax/_src/__pycache__/ad_util.cpython-312.pyc,,
jax/_src/__pycache__/api.cpython-312.pyc,,
jax/_src/__pycache__/api_util.cpython-312.pyc,,
jax/_src/__pycache__/array.cpython-312.pyc,,
jax/_src/__pycache__/basearray.cpython-312.pyc,,
jax/_src/__pycache__/blocked_sampler.cpython-312.pyc,,
jax/_src/__pycache__/buffer_callback.cpython-312.pyc,,
jax/_src/__pycache__/cache_key.cpython-312.pyc,,
jax/_src/__pycache__/callback.cpython-312.pyc,,
jax/_src/__pycache__/checkify.cpython-312.pyc,,
jax/_src/__pycache__/cloud_tpu_init.cpython-312.pyc,,
jax/_src/__pycache__/compilation_cache.cpython-312.pyc,,
jax/_src/__pycache__/compilation_cache_interface.cpython-312.pyc,,
jax/_src/__pycache__/compiler.cpython-312.pyc,,
jax/_src/__pycache__/compute_on.cpython-312.pyc,,
jax/_src/__pycache__/config.cpython-312.pyc,,
jax/_src/__pycache__/core.cpython-312.pyc,,
jax/_src/__pycache__/custom_api_util.cpython-312.pyc,,
jax/_src/__pycache__/custom_batching.cpython-312.pyc,,
jax/_src/__pycache__/custom_derivatives.cpython-312.pyc,,
jax/_src/__pycache__/custom_partitioning.cpython-312.pyc,,
jax/_src/__pycache__/custom_partitioning_sharding_rule.cpython-312.pyc,,
jax/_src/__pycache__/debugging.cpython-312.pyc,,
jax/_src/__pycache__/deprecations.cpython-312.pyc,,
jax/_src/__pycache__/dispatch.cpython-312.pyc,,
jax/_src/__pycache__/distributed.cpython-312.pyc,,
jax/_src/__pycache__/dlpack.cpython-312.pyc,,
jax/_src/__pycache__/dtypes.cpython-312.pyc,,
jax/_src/__pycache__/earray.cpython-312.pyc,,
jax/_src/__pycache__/effects.cpython-312.pyc,,
jax/_src/__pycache__/environment_info.cpython-312.pyc,,
jax/_src/__pycache__/error_check.cpython-312.pyc,,
jax/_src/__pycache__/errors.cpython-312.pyc,,
jax/_src/__pycache__/ffi.cpython-312.pyc,,
jax/_src/__pycache__/flatten_util.cpython-312.pyc,,
jax/_src/__pycache__/frozen_dict.cpython-312.pyc,,
jax/_src/__pycache__/hardware_utils.cpython-312.pyc,,
jax/_src/__pycache__/hashable_array.cpython-312.pyc,,
jax/_src/__pycache__/hijax.cpython-312.pyc,,
jax/_src/__pycache__/indexing.cpython-312.pyc,,
jax/_src/__pycache__/jaxpr_util.cpython-312.pyc,,
jax/_src/__pycache__/lax_reference.cpython-312.pyc,,
jax/_src/__pycache__/layout.cpython-312.pyc,,
jax/_src/__pycache__/lazy_loader.cpython-312.pyc,,
jax/_src/__pycache__/linear_util.cpython-312.pyc,,
jax/_src/__pycache__/literals.cpython-312.pyc,,
jax/_src/__pycache__/logging_config.cpython-312.pyc,,
jax/_src/__pycache__/lru_cache.cpython-312.pyc,,
jax/_src/__pycache__/mesh.cpython-312.pyc,,
jax/_src/__pycache__/mesh_utils.cpython-312.pyc,,
jax/_src/__pycache__/monitoring.cpython-312.pyc,,
jax/_src/__pycache__/named_sharding.cpython-312.pyc,,
jax/_src/__pycache__/op_shardings.cpython-312.pyc,,
jax/_src/__pycache__/partition_spec.cpython-312.pyc,,
jax/_src/__pycache__/path.cpython-312.pyc,,
jax/_src/__pycache__/pickle_util.cpython-312.pyc,,
jax/_src/__pycache__/pjit.cpython-312.pyc,,
jax/_src/__pycache__/pmap.cpython-312.pyc,,
jax/_src/__pycache__/pretty_printer.cpython-312.pyc,,
jax/_src/__pycache__/prng.cpython-312.pyc,,
jax/_src/__pycache__/profiler.cpython-312.pyc,,
jax/_src/__pycache__/public_test_util.cpython-312.pyc,,
jax/_src/__pycache__/random.cpython-312.pyc,,
jax/_src/__pycache__/ref.cpython-312.pyc,,
jax/_src/__pycache__/shard_alike.cpython-312.pyc,,
jax/_src/__pycache__/shard_map.cpython-312.pyc,,
jax/_src/__pycache__/sharding.cpython-312.pyc,,
jax/_src/__pycache__/sharding_impls.cpython-312.pyc,,
jax/_src/__pycache__/source_info_util.cpython-312.pyc,,
jax/_src/__pycache__/sourcemap.cpython-312.pyc,,
jax/_src/__pycache__/stages.cpython-312.pyc,,
jax/_src/__pycache__/stateful_rng.cpython-312.pyc,,
jax/_src/__pycache__/test_loader.cpython-312.pyc,,
jax/_src/__pycache__/test_multiprocess.cpython-312.pyc,,
jax/_src/__pycache__/test_util.cpython-312.pyc,,
jax/_src/__pycache__/test_warning_util.cpython-312.pyc,,
jax/_src/__pycache__/tpu_custom_call.cpython-312.pyc,,
jax/_src/__pycache__/traceback_util.cpython-312.pyc,,
jax/_src/__pycache__/tree.cpython-312.pyc,,
jax/_src/__pycache__/tree_util.cpython-312.pyc,,
jax/_src/__pycache__/typing.cpython-312.pyc,,
jax/_src/__pycache__/util.cpython-312.pyc,,
jax/_src/__pycache__/xla_bridge.cpython-312.pyc,,
jax/_src/__pycache__/xla_metadata.cpython-312.pyc,,
jax/_src/__pycache__/xla_metadata_lib.cpython-312.pyc,,
jax/_src/abstract_arrays.py,sha256=vEspfYvTlWGI9BzJidpWC1iigJGAkm7TC_QCF3I4W8o,6154
jax/_src/ad_checkpoint.py,sha256=rE46PYvu-fxxtCDeDh-Jn32p35uDH5Rexh4f_Uk9eqA,44760
jax/_src/ad_util.py,sha256=5fdKdaVsHt9Ge6z6tuPkRcA3G3XU7nPEByu96uz_D8I,4614
jax/_src/api.py,sha256=ZEHY7RVHYztalTldJh4K8hTyUdlrtH9-dIxOAzgKnPo,111002
jax/_src/api_util.py,sha256=lrMRHFydhGZ-vxFt-UOGWrvZv7GR5p62ZcKxnUiGy0k,31169
jax/_src/array.py,sha256=NK_tJMjb0WQa6b9dlmXPjc99An_5usFySsQmoQ3alQs,50460
jax/_src/basearray.py,sha256=g0QKD-iu2USDw-W4HRu1RFOQe04bZaDjsR-kmveKaug,7453
jax/_src/basearray.pyi,sha256=qTYC2f1P1cLQ4NVaua4hdjv-01CQl0Zm-u2yoEZl8Bg,14699
jax/_src/blocked_sampler.py,sha256=s0HqkBXTh2t3eTqKV86PHf3YHvRJv4J44J-Oagu3oJo,5920
jax/_src/buffer_callback.py,sha256=MZmTSlA8RnewSdxW7MNWKQuupWeX9hlKgZ4pHdyU6vA,10464
jax/_src/cache_key.py,sha256=f9mcVW4cWH3CLHFoZHhtPGC1i7eUguaKL5mkhpwCyY0,13309
jax/_src/callback.py,sha256=DLpYS03weuq0AMXTUDzMTH52SI5PIGxXfmBkApnaohk,34196
jax/_src/checkify.py,sha256=GHiY1gqJo36k9FnODsp2t4P6ZdbkRTR4634I8fHlP8s,57033
jax/_src/cloud_tpu_init.py,sha256=WikCN0NjkMn4ACz6oqIhw3Axyx7gaftFWsqixFA1pcs,5050
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
jax/_src/clusters/__pycache__/__init__.cpython-312.pyc,,
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/k8s_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/mpi4py_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/ompi_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/slurm_cluster.cpython-312.pyc,,
jax/_src/clusters/cloud_tpu_cluster.py,sha256=j2isj3e-Tc_ioF7bsMtz0U3ZgkE7m8C-m0fhsmFBRaI,9596
jax/_src/clusters/cluster.py,sha256=dlgXWj40fEJ7Pt92kYsm7eAGcB89j-TtpCv5wpka-5c,5889
jax/_src/clusters/k8s_cluster.py,sha256=K299M1mW1vYa_6iEuJEAD9HRtig9jKNginp1wJKdV_o,9487
jax/_src/clusters/mpi4py_cluster.py,sha256=sp3lT2edLQtTb7L1Nx9zD4nGdT-WOlqP4ufw59ZXV4A,2781
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/compilation_cache.py,sha256=VABLW01sPDTvyZecd9uHuSeAuK-AtZlWkvQm0Ztmaio,14609
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=WEwxYOx1YVSuXzvSc4cnLQwlFvxjd1K_KJ6tB1KyZVM,33009
jax/_src/compute_on.py,sha256=NXLi8rJQ7fnDe4qI25REDWU9goHZs56cZquXDjNhcN0,11455
jax/_src/config.py,sha256=qaK8z4azVzQXFpnLNOLJFCrdXJv71UIpQeGmsZLSWq0,81435
jax/_src/core.py,sha256=2koTxavpORo5T7QYrOgpwtQKNn2JeVSXliAPOncIVRA,150385
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
jax/_src/cudnn/__pycache__/__init__.cpython-312.pyc,,
jax/_src/cudnn/__pycache__/fused_attention_stablehlo.cpython-312.pyc,,
jax/_src/cudnn/__pycache__/fusion.cpython-312.pyc,,
jax/_src/cudnn/__pycache__/scaled_matmul_stablehlo.cpython-312.pyc,,
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=5usEfB03VHp5M8dVdb02TTqN-l_kvepZbJXIj-YHy6A,82427
jax/_src/cudnn/fusion.py,sha256=0FF_DACwZ2HRuRUQA9LdAPgYKFADMhItn8AHed-jmZs,2866
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=zxtTYcsEt6bBSI4nZ_xtWyAXtY3YIxr-UlYAZ1Wv-Ts,26987
jax/_src/custom_api_util.py,sha256=GgJtgWnv9lTFUeOUjdd0U2xvVHEM4iCRBvqYoWqevb8,831
jax/_src/custom_batching.py,sha256=cyTYtuLA89mIwwgBQGx9YbrFaucCoeogaDDMfrRvfLg,15183
jax/_src/custom_derivatives.py,sha256=ltpVkId9utcDwZllcE1_58BIZ3K0JAm7hJAR-FlgfrE,79989
jax/_src/custom_partitioning.py,sha256=2X5QcNM4VyA3P0fS51Atn7P-KuOP8fpTZznkZHqvBDw,28466
jax/_src/custom_partitioning_sharding_rule.py,sha256=NpdehLzMqEELhHbeQd6-jRkG6siLiYm9pmgQBVW5XxA,20765
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/__pycache__/__init__.cpython-312.pyc,,
jax/_src/debugger/__pycache__/cli_debugger.cpython-312.pyc,,
jax/_src/debugger/__pycache__/colab_debugger.cpython-312.pyc,,
jax/_src/debugger/__pycache__/colab_lib.cpython-312.pyc,,
jax/_src/debugger/__pycache__/core.cpython-312.pyc,,
jax/_src/debugger/__pycache__/web_debugger.cpython-312.pyc,,
jax/_src/debugger/cli_debugger.py,sha256=7Bk4tfCQgTR69y63z9myy4SRqnP6riFBD5dl9tMMLG0,4834
jax/_src/debugger/colab_debugger.py,sha256=bngN7OtnM09xCmvRnJnoP7EUOd0_FxNRh_SpikNGECc,7710
jax/_src/debugger/colab_lib.py,sha256=5gxJDlgQqFFp23jYd8OexzUE2yKv5IJwUpSXoGpndvU,4321
jax/_src/debugger/core.py,sha256=raht1TFiKJfVUmeSqQ20H5w2_oVcLpX7z6a3BpqcP_k,8610
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
jax/_src/debugging.py,sha256=bQyfDehLCbEYkKByoJPPEbqpFd7ykPBaGDxjDmKTaQQ,34369
jax/_src/deprecations.py,sha256=L0-2QiMSrPq2twCVLA0iICgjjJ4KDR4az4-35Qf7W-g,4491
jax/_src/dispatch.py,sha256=pGI42wHqFfuv-JkB6pFpNfZRWMvX69QVeNh_Pm_gdYQ,30422
jax/_src/distributed.py,sha256=zbCY-qNEXmvARnEfkHWagY_1Ijq9Kzu9XC5VQ8MYWXA,15694
jax/_src/dlpack.py,sha256=nx-ECIv5fiSe5cxWFymWD3qMmLiUTFiRJlzQnJEweVc,11414
jax/_src/dtypes.py,sha256=HaXxSgzpr5Qyo9dXf5R4i69JsyuW4xoFtYxS9YMX15M,41424
jax/_src/earray.py,sha256=gg-ofv8qjedwxqOy7ES8C5WEVVwmCItk145tEOq5qqo,4458
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=MLcFqK4oPj5elgC6GWqKli_SyJdbbq6No41493WWeZ4,24751
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/__pycache__/__init__.cpython-312.pyc,,
jax/_src/export/__pycache__/_export.cpython-312.pyc,,
jax/_src/export/__pycache__/serialization.cpython-312.pyc,,
jax/_src/export/__pycache__/serialization_generated.cpython-312.pyc,,
jax/_src/export/__pycache__/shape_poly.cpython-312.pyc,,
jax/_src/export/__pycache__/shape_poly_decision.cpython-312.pyc,,
jax/_src/export/_export.py,sha256=tAjoihYnIqW9PNM-QaGxr_UuNqJxgkyfRBV1WGof0fw,79460
jax/_src/export/serialization.py,sha256=yVySwDfv3L4_0xNlMJJytg4Yl0SgttG6sNHEgXSudW0,37113
jax/_src/export/serialization_generated.py,sha256=6V9ZWnuUA7bkskOzzqLRhS01CN29yMACLybfWdUPc_8,50902
jax/_src/export/shape_poly.py,sha256=PL5mIoXiwJciCR9TbvO4ETFm15Yqs0SoZGsm2eeyRb0,86068
jax/_src/export/shape_poly_decision.py,sha256=ZkAOWUlx2y6JotHVieJCrO_Pes7cCMO9WX1_w24QLB8,20522
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-312.pyc,,
jax/_src/extend/__pycache__/random.cpython-312.pyc,,
jax/_src/extend/random.py,sha256=Rc68o6EA7zPLY3nK-cyF8Tq0WPdvnk81rg9sLb22QRg,1269
jax/_src/ffi.py,sha256=3y_uQCJ6UCU2LMaKbc9cfFHoLxpA2ENjmxDLZ3NZVlk,30230
jax/_src/flatten_util.py,sha256=sG6qaC_yWTcNNbVjc1A1KrwVz3w2q0b1nm7SQGFmKBU,3971
jax/_src/frozen_dict.py,sha256=KKEuFtV9F-GH71O4x0NZqLayaNIsV-A3Xqg5lpEcQyI,1474
jax/_src/hardware_utils.py,sha256=YQcgkyPgQN94grS6VvBIVVFL2R-LDCsdY8Tqj43jnVA,2349
jax/_src/hashable_array.py,sha256=Awo1-CwOJJifHgGrlBDRfCgSC3myBNuSGxEjX5FWhpM,1147
jax/_src/hijax.py,sha256=lw9Tp87Y1rLagvwAHvEESFgE1UKoV3D-vEmAcEYV2HE,39151
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/__pycache__/__init__.cpython-312.pyc,,
jax/_src/image/__pycache__/scale.cpython-312.pyc,,
jax/_src/image/scale.py,sha256=Ik5jtc7cjTjzYzB7RLS6mOApHbgfPJJ3Dfzv83rJ37o,19204
jax/_src/indexing.py,sha256=k3pMnw5hr8BneE2fxnCzc2GvBn_3BU6e9BrH4ufJ3YM,3377
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-312.pyc,,
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
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/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/annotate_data_placement.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_cholesky_lapack_potrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eig_lapack_geev.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eigh_lapack_syev.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_hessenberg_lapack_gehrd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_lu_lapack_getrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_qr_lapack_geqrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_schur_lapack_gees.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_svd_lapack_gesdd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_triangular_solve_blas_trsm.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_lapack_sytrd_hetrd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_solve_lapack_gtsv.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_cholesky_solver_potrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_eigh_cusolver_syev.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_cusolver_getrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_pivots_to_permutation.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_qr_cusolver_geqrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_svd_cusolver_gesvd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_threefry2x32.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_cusolver_sytrd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_solve.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/export_with_memory_space.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/export_with_multiple_meshes.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/export_with_specified_sharding.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/export_with_unspecified_sharding.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/gpu_eigh_solver_syev.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_cholesky_solver_potrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_eigh_hipsolver_syev.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_lu_pivots_to_permutation.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_lu_rocsolver_getrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_qr_hipsolver_geqrf.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_svd_hipsolver_gesvd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_threefry2x32.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_tridiagonal_hipsolver_sytrd.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/shardy_sharding_ops_with_different_meshes.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_approx_top_k.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_rng_bit_generator.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_top_k.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_ApproxTopK.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Eigh.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Lu.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Qr.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Sharding.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_stablehlo_dynamic_reduce_window.cpython-312.pyc,,
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=MaBa7vvTBiVm6MdHZ3BMM70d9AqkswobRMQZ_PIZUGY,11314
jax/_src/internal_test_util/export_back_compat_test_data/export_with_multiple_meshes.py,sha256=3ndTSFY4mLYIfi41P6wtINwv-pnXXZokMJMqco5y3is,41690
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/pallas/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_boolean_constant.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_gpu_add_one.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_matmul.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_semaphore_dma.cpython-312.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/triton_add_one.cpython-312.pyc,,
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/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_util.py,sha256=dBLniYzagqkiB_b2Y3d75Ua9u3-OnkNmbNv7CD0eLN4,17217
jax/_src/internal_test_util/lax_test_util.py,sha256=oAg8V0SwubAntFNfKAryoFfQpfNnHLn7NGv8VNg_Yo0,13158
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-312.pyc,,
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/internal_test_util/test_harnesses.py,sha256=8hX-J6mxr3KHC5NVGZNIuHB4V0GTNM2rmMxY-ocinY8,126705
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/__pycache__/__init__.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/ad.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/batching.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/mlir.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/pxla.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/remat.cpython-312.pyc,,
jax/_src/interpreters/ad.py,sha256=djDEGc7vc6RegiyeUGYhuRXZrSXEpqt3ef4Y4iQxsFw,58155
jax/_src/interpreters/batching.py,sha256=sfnp-5tGVRKewttveZ7pDDQ22dH9F6ksTv7YVxy1HLg,32402
jax/_src/interpreters/mlir.py,sha256=qCMNAKKt1ZZw8qW9m6qCvWHR4thVTwxaVfav_48vEGo,141222
jax/_src/interpreters/partial_eval.py,sha256=3D5LPNmI3L8kgcqKSZ0gtn6o0mCF-TtLnPF4z5lgK1M,106156
jax/_src/interpreters/pxla.py,sha256=CZxsXnGN1wvXIY7s0vJvpeh-szQ8-eX40hEpbHatI24,85266
jax/_src/interpreters/remat.py,sha256=zjD2Xu0L-ROhhNAgfKT5ESrMx7UX-vF7F5-2Vi22KMk,5349
jax/_src/jaxpr_util.py,sha256=kDAWQAy9olqgd9yEELNWbjXYRkmCbe9l058pKhGF2io,10513
jax/_src/lax/__init__.py,sha256=RCwdy4QblTyQxzwoqj1yxmMWHi2Tk0Q2fy0Dy6cGeaU,8592
jax/_src/lax/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lax/__pycache__/ann.cpython-312.pyc,,
jax/_src/lax/__pycache__/convolution.cpython-312.pyc,,
jax/_src/lax/__pycache__/fft.cpython-312.pyc,,
jax/_src/lax/__pycache__/lax.cpython-312.pyc,,
jax/_src/lax/__pycache__/linalg.cpython-312.pyc,,
jax/_src/lax/__pycache__/other.cpython-312.pyc,,
jax/_src/lax/__pycache__/parallel.cpython-312.pyc,,
jax/_src/lax/__pycache__/scaled_dot.cpython-312.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-312.pyc,,
jax/_src/lax/__pycache__/special.cpython-312.pyc,,
jax/_src/lax/__pycache__/utils.cpython-312.pyc,,
jax/_src/lax/__pycache__/windowed_reductions.cpython-312.pyc,,
jax/_src/lax/ann.py,sha256=En9crn02gbbxxtl4u6dMMPRbmJ7cYBSoXUbN75UMwKc,17128
jax/_src/lax/control_flow/__init__.py,sha256=JsuWyikJkp-5IVosqvubQfR6K1qXVeKMmSgNWLDzjso,2093
jax/_src/lax/control_flow/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/common.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/loops.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/solves.cpython-312.pyc,,
jax/_src/lax/control_flow/common.py,sha256=aF2wy1_-CvEYOmwv-Asu3FNRXhCo1CxY3SdtXuvrm3M,6464
jax/_src/lax/control_flow/conditionals.py,sha256=LtyJ6cm-0u7lal8yFVa5Z5PoKvDecrpvdFLbGz9s1GQ,53612
jax/_src/lax/control_flow/loops.py,sha256=kvwOJoNib3fjBNe5xvYKxXNkazwjJ_efDmwgyvn8V9g,135180
jax/_src/lax/control_flow/solves.py,sha256=OitRduN73UlWVrx4nnBhlTNgOmtJAMtgkc-C4VFAZ60,20734
jax/_src/lax/convolution.py,sha256=irz2YjCCpdUhpzc3x_dDt-47YitpPwjalDSKGGgBvcE,50559
jax/_src/lax/fft.py,sha256=LkxZCyMttQ_N53WTCmExc8OxnVmC5a8VH57u0TrG2PY,8871
jax/_src/lax/lax.py,sha256=1NpCIClILwwQejdrSDt-mfXN3jwYqW9iAdyCsaX9DMY,370018
jax/_src/lax/linalg.py,sha256=VM0a8b-TFYxIbW2DigHfkF9Bz4VSynnhqtVGCf3MnS0,119353
jax/_src/lax/other.py,sha256=E1n6P-LXfYNbi9maGIDHRBvw994z6XojQvYRC2Kt-lc,13863
jax/_src/lax/parallel.py,sha256=jJI1mYnU5NbHtDIb4n_YZAIYNTNeZ_C__U_wWNX8OM0,127116
jax/_src/lax/scaled_dot.py,sha256=EGZl_9mmseMvxzFdUcn1zbb-dTB4k9VCzsKn01hKcIQ,13719
jax/_src/lax/slicing.py,sha256=PsX9x2kcTk3HXGSLyvMRN0-jyiFUCs8sHCFMsf_abmQ,150298
jax/_src/lax/special.py,sha256=aQpun6B6VDi0FG_steT1WHfQWYLwb_k7Gi5PYxvuyT4,28920
jax/_src/lax/utils.py,sha256=gt74K2RbcO4wTiYS9SHkcvqPjOD1JYiEbmvEzwcIzt8,12089
jax/_src/lax/windowed_reductions.py,sha256=bFcCSez8mN1DDIkghOO7UJ3o396b-_3bjeSyxjbFjC8,47727
jax/_src/lax_reference.py,sha256=smNLw8sUa9W1eL3gzumuKTxDpDnQww96K1Q20a71c6s,20313
jax/_src/layout.py,sha256=hh6Ofbi5Pk1IYl1kCGokD40QbTSf9XUdOs8vVaniEls,5728
jax/_src/lazy_loader.py,sha256=IEhbhAOHbWusMVAGzBPLKYFPkAXcP2sbK9Q3IcnyG1k,1893
jax/_src/lib/__init__.py,sha256=XFWlw5xMHDoAo9nxsayF0K1cOI71lK9wx7nEhO1the8,7564
jax/_src/lib/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/__pycache__/mosaic_gpu.cpython-312.pyc,,
jax/_src/lib/__pycache__/triton.cpython-312.pyc,,
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
jax/_src/lib/mlir/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/mlir/dialects/__init__.py,sha256=zsqbkEnPPS1LY5XzzTp0TkAoCd9lAwICwX4g2YHBGtw,2089
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/mosaic_gpu.py,sha256=andYvo1TC8Eru9Pd_rM76L66Ozami3Q5awjdTkdIE10,1017
jax/_src/lib/triton.py,sha256=u6N-heI4favwsraseGEwzdIHe3dKjeubO6Mw2cG8Y3g,2205
jax/_src/linear_util.py,sha256=Um_dewR7yGwnGBTazAIqKI3hP4ervmicInv42C-JZbo,18553
jax/_src/literals.py,sha256=2tKoRX_Ql6BB8jMuFipFuVgyvO-dDtn5bdVL6ks7uZs,6035
jax/_src/logging_config.py,sha256=4Sk_kyeggOxktYSUfnXmo8zMoGk1Ud10XhYJXo1d1Aw,4272
jax/_src/lru_cache.py,sha256=GjEK1v9YhMfcSqvEEGxDleyRwbf1UG3pgZwMzWXPWCU,6728
jax/_src/mesh.py,sha256=O7IPYCAWx2zIyoV9zFkgn4CZfWtGnX4om-bdia_Umcc,20557
jax/_src/mesh_utils.py,sha256=3FcmzYsbqCtCMQWdXLRSbonp0OsprgJhQKv6Fb5Qfxc,34490
jax/_src/monitoring.py,sha256=OcXRp8maRJewgZDoKLE8D-9tHZnjINXA_m35ewm9uPY,5631
jax/_src/named_sharding.py,sha256=IoRrMPnaDJV_azDf4LhTU89KwMgv1qB0cuL8yZaYfpI,22004
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/__pycache__/__init__.cpython-312.pyc,,
jax/_src/nn/__pycache__/functions.cpython-312.pyc,,
jax/_src/nn/__pycache__/initializers.cpython-312.pyc,,
jax/_src/nn/functions.py,sha256=0AzoB9Hw2Tu80ZNUxIQ_dSTFJ1TvlNNFRQkBWUlEwqs,48401
jax/_src/nn/initializers.py,sha256=yC9d99yIGJ0N03oauzFkWTw1ap3jirt0pEAC7jLQg-g,27333
jax/_src/numpy/__init__.py,sha256=_RvbF3GcGcbXqVDVFXdVDq-MuJwf36zomNpgQCZWw4k,5680
jax/_src/numpy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/numpy/__pycache__/array_api_metadata.cpython-312.pyc,,
jax/_src/numpy/__pycache__/array_constructors.cpython-312.pyc,,
jax/_src/numpy/__pycache__/array_creation.cpython-312.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-312.pyc,,
jax/_src/numpy/__pycache__/einsum.cpython-312.pyc,,
jax/_src/numpy/__pycache__/error.cpython-312.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-312.pyc,,
jax/_src/numpy/__pycache__/hijax.cpython-312.pyc,,
jax/_src/numpy/__pycache__/index_tricks.cpython-312.pyc,,
jax/_src/numpy/__pycache__/indexing.cpython-312.pyc,,
jax/_src/numpy/__pycache__/lax_numpy.cpython-312.pyc,,
jax/_src/numpy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/numpy/__pycache__/polynomial.cpython-312.pyc,,
jax/_src/numpy/__pycache__/reductions.cpython-312.pyc,,
jax/_src/numpy/__pycache__/scalar_types.cpython-312.pyc,,
jax/_src/numpy/__pycache__/setops.cpython-312.pyc,,
jax/_src/numpy/__pycache__/sorting.cpython-312.pyc,,
jax/_src/numpy/__pycache__/tensor_contractions.cpython-312.pyc,,
jax/_src/numpy/__pycache__/ufunc_api.cpython-312.pyc,,
jax/_src/numpy/__pycache__/ufuncs.cpython-312.pyc,,
jax/_src/numpy/__pycache__/util.cpython-312.pyc,,
jax/_src/numpy/__pycache__/vectorize.cpython-312.pyc,,
jax/_src/numpy/__pycache__/window_functions.cpython-312.pyc,,
jax/_src/numpy/array_api_metadata.py,sha256=Xu2XKu-3r2cRI8-bgY9zd3OkROMO0vEk7qw9Xy-3XxY,3636
jax/_src/numpy/array_constructors.py,sha256=cf3yDMYtqrjqMpMkYu65iOnXUunORFHkeluTZOlpw9w,16699
jax/_src/numpy/array_creation.py,sha256=MvCx0yLDqF8Fp3oHejq_KqOU0SKOISkbZqFBdH9qe2A,31162
jax/_src/numpy/array_methods.py,sha256=c2SYE1z6ml-yjYGa-TFkeuSIFPVvG75suhGDV0Ypn18,61280
jax/_src/numpy/einsum.py,sha256=h7ZWlw1PWSEfWGhwtaeHzPgsxZUfHnP7wi9OtinD9BU,24439
jax/_src/numpy/error.py,sha256=kqoCUrAfFL1zmtVJfydI3MrpB9NNuYxTqEVmX4iidzM,5450
jax/_src/numpy/fft.py,sha256=bpoFtzhhvOTsSsHTsun3XZcnc6ITKciZiOit4baax0c,52329
jax/_src/numpy/hijax.py,sha256=KULofdIvnj-3xv6gO4ELwdL2WnPl9spWd9HYoEUTjjo,16921
jax/_src/numpy/index_tricks.py,sha256=VTGtSNYaMu6wzNwScnpeCbSiRySAcW2tIFW3wX39jM8,10034
jax/_src/numpy/indexing.py,sha256=Q6eTKrcROt8AkkpNTQll2Z1pslQiITY7tm4QCX1ZTrM,67868
jax/_src/numpy/lax_numpy.py,sha256=Hi5Q49dCvtNopE5S8-ZRrcjQjV8I9Q2-cSO1uBWtDCM,334272
jax/_src/numpy/linalg.py,sha256=Tpwhbsp-nSSCr1XQi3-75i_ulWVge2XO2myu3lfA4Qo,81817
jax/_src/numpy/polynomial.py,sha256=jAuRK7qso8BIl6yBHEP9k3W58yo7UHfrYfZTcvzCGsk,28420
jax/_src/numpy/reductions.py,sha256=sBmNjhWXJHP_ZRT9lD5GnOYazlP-WvV16_8mV5vk5x8,116211
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=J5MYZif0wwLdbLHmTUJnhs1A9zhgNwJ3L1NdD1ZrAhc,24322
jax/_src/numpy/ufuncs.py,sha256=dMEfBmZg3btm1LsoP6JMCZ0LZu9JF6aiBfTyBzE3DRY,118238
jax/_src/numpy/util.py,sha256=bClkNYrtK6J6ymt3BFwmQPuy6XgL1nLFCnCsjKPA9oc,16482
jax/_src/numpy/vectorize.py,sha256=Y-cmOE2qezjEY2fRrgIz1fcaP8Oyu7vJvpk5Tu6gCnY,14215
jax/_src/numpy/window_functions.py,sha256=jZsnPyv4nF6IXNHMCiX_9-gBGDdJfyb7V74fLN8TaJ0,5473
jax/_src/op_shardings.py,sha256=CD9WkXNJaFmg_ZK55pyhzhBtEJLSKS1X7JGLcL47wQQ,3943
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/__pycache__/__init__.cpython-312.pyc,,
jax/_src/ops/__pycache__/scatter.cpython-312.pyc,,
jax/_src/ops/__pycache__/special.cpython-312.pyc,,
jax/_src/ops/scatter.py,sha256=qte0ZMlsS095MN6IptsmrpOjWOcKBvNDnsHAiAz6poY,18147
jax/_src/ops/special.py,sha256=zYKBMe29eMqKXDeaZ5BK8YehQdpAxAHPFKTmNjwKBa4,4380
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/__pycache__/cost_estimate.cpython-312.pyc,,
jax/_src/pallas/__pycache__/einshape.cpython-312.pyc,,
jax/_src/pallas/__pycache__/helpers.cpython-312.pyc,,
jax/_src/pallas/__pycache__/hlo_interpreter.cpython-312.pyc,,
jax/_src/pallas/__pycache__/mpmd.cpython-312.pyc,,
jax/_src/pallas/__pycache__/pallas_call.cpython-312.pyc,,
jax/_src/pallas/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/__pycache__/utils.cpython-312.pyc,,
jax/_src/pallas/core.py,sha256=Zko0Jz7U--gJeosv-OmwRNT5JlVMtPgSs4urgLHk6WM,61597
jax/_src/pallas/cost_estimate.py,sha256=xH9RKiOw14iirM_gPuchy6k-gk-Z0j7WTrOZ6RxZOho,8351
jax/_src/pallas/einshape.py,sha256=4kwlDW_8KK2L6qUuan2QVrjks7cgQ3XmPhS7zbiR67Y,20558
jax/_src/pallas/fuser/__init__.py,sha256=9vpH6Fvikt8NEg8EA0o3ryFBh1oz37RMm1MudOZ-WxY,1265
jax/_src/pallas/fuser/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/block_spec.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/custom_evaluate.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/custom_fusion_lib.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/fuser_utils.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/fusible.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/fusible_dtype.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/fusion.cpython-312.pyc,,
jax/_src/pallas/fuser/__pycache__/jaxpr_fusion.cpython-312.pyc,,
jax/_src/pallas/fuser/block_spec.py,sha256=tHLAixGTGFPTcnqVI2hfVQMHuAuK6WorRpBszlJANwM,85386
jax/_src/pallas/fuser/custom_evaluate.py,sha256=nVKiLrI7qaXn-uJN4z0BL3o8BGtCO6dMnKmAAoQDfDQ,2804
jax/_src/pallas/fuser/custom_fusion_lib.py,sha256=3W8khCUZILw6VlrnacDHiBolKup3dtVn-SbAKBtXCXM,8985
jax/_src/pallas/fuser/fuser_utils.py,sha256=nu0u_l7Xv4mi0X3rRY2ZfIXvQogesdAAT0QIhBfirU4,1279
jax/_src/pallas/fuser/fusible.py,sha256=oj_iuiaqYS0NsDCwNM2IG3jv9-j6HVLn59firsBXeAk,4558
jax/_src/pallas/fuser/fusible_dtype.py,sha256=ATTs-uqg-A1Ld7yvRmilsYQSApf7CsjGOp6Z33_XXN4,18091
jax/_src/pallas/fuser/fusion.py,sha256=AvfY5PSPzoAT5Vu6NpMHPqobSgNfvv_600VbWIlcslI,1515
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=bqKsZZGlNcHdfBcrby3CUdApoRzMbCkHUsSPrrlIWv8,11087
jax/_src/pallas/helpers.py,sha256=kAXZCKZ0v2UtpJo8hCbxOV6anZ1i_yzbV0karUMlmSc,10026
jax/_src/pallas/hlo_interpreter.py,sha256=Zd766jGeurB6Lpc7NjLm5BTFfMaLFKQYLDgYIuupZZM,18999
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/error_handling.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/helpers.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/pipeline.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/random.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/sc_core.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/sc_lowering.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/sc_primitives.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/tpu_info.cpython-312.pyc,,
jax/_src/pallas/mosaic/core.py,sha256=b9hzWVQu2ROEWF44lLHFt2VVaU3dEyAMX0tRfZE08Zc,19493
jax/_src/pallas/mosaic/error_handling.py,sha256=Czccjp8oL3NMUhQm0yho-EEDWbD2POJQmulvceATRbo,5534
jax/_src/pallas/mosaic/helpers.py,sha256=pf0G5tV09sj7zZLi7zP0L6158qJy4Gz6VBJm7rZIhVY,2693
jax/_src/pallas/mosaic/interpret/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/pallas/mosaic/interpret/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/interpret_pallas_call.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/params.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/race_detection_state.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/shared_memory.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/thread_map.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/utils.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/__pycache__/vector_clock.cpython-312.pyc,,
jax/_src/pallas/mosaic/interpret/interpret_pallas_call.py,sha256=BVDoNqYkR4w0u-nk-MJdj-WJXN9upis6uF0jGmdy1Fc,81570
jax/_src/pallas/mosaic/interpret/params.py,sha256=C4-rREVUI3z92GDzSRDZPNMjNwqiwFmYyZMyUT5dK-I,7610
jax/_src/pallas/mosaic/interpret/race_detection_state.py,sha256=i5t4NwQ4zey0b-rIal_ou9JXh1OBwzRdXp_NEHkyBo4,6454
jax/_src/pallas/mosaic/interpret/shared_memory.py,sha256=OPiI-JKL98nVkGZT_saMVqlSxHutdg81PM7Kq5mwAEE,29994
jax/_src/pallas/mosaic/interpret/thread_map.py,sha256=-e_QC3r1F9A6Gs3LcKKV0nFFmPK7rNzhJ3FbdibAriM,3133
jax/_src/pallas/mosaic/interpret/utils.py,sha256=71QW2O-RhpOz2x_P0oAliAuBhZshTcGdzzk35lNY2e4,10848
jax/_src/pallas/mosaic/interpret/vector_clock.py,sha256=-Z8fzIB8BO0rx4cxzV-QeoOt3w7khYY5_BT1xFj92nk,1363
jax/_src/pallas/mosaic/lowering.py,sha256=li752QgQA_DC3BWHxk12ZPpvy4OZXHUWN6PshhvnBsQ,161685
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=W85nmnrWxERGX1OX60_M7mlGN2XhxJOW5s8Ic4H41bg,20623
jax/_src/pallas/mosaic/pipeline.py,sha256=MTe-x16t9vZERjlMZp_37Ap16YfRgtfgtq9407cUaB4,64067
jax/_src/pallas/mosaic/primitives.py,sha256=5KUFH9TyafhWL5r0LXiTkfkFLr3vDdWBLaHFMXIWezU,44885
jax/_src/pallas/mosaic/random.py,sha256=BkF4uekd6goGSk741Nd8tiKxH9Pv2kjAQMB4Ak42eog,7900
jax/_src/pallas/mosaic/sc_core.py,sha256=lNBsiW5fQbhB3WZU1OKdNOuC4Uz4aB_rYio8fXuhCWI,11344
jax/_src/pallas/mosaic/sc_lowering.py,sha256=uATiSc0LGlSIckMMnT6HAZPX5H2ChmqRvOWTnTkxu-c,45754
jax/_src/pallas/mosaic/sc_primitives.py,sha256=BeRP2_2HPyYuPkBKDRHkKKAuQRlu8UMX_XV8PUd6yoU,44244
jax/_src/pallas/mosaic/tpu_info.py,sha256=Q4sWcyESSx1-xqgI_bo7mGbjVZjfFHFxS6bkzAUQAt0,20735
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/helpers.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/lowering.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/pallas_call_registration.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/pipeline.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/torch.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/core.py,sha256=zK1hmyHIMUAPhB_Egp6Cx_9b88yFR9Hl26MHCPF-JT8,56905
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=tfbGJ2sFQE-5cITW_ITaVDkv_xV3I177kc0Y5C6h7pA,15574
jax/_src/pallas/mosaic_gpu/interpret/__init__.py,sha256=-aCGt5dPNdKLi-ALblRdGErT_9Or12mpWJ52yApKN0I,581
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/gpu_callbacks.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/interpret_pallas_call.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/jaxpr_interpret.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/params.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/__pycache__/shared_memory.cpython-312.pyc,,
jax/_src/pallas/mosaic_gpu/interpret/gpu_callbacks.py,sha256=k6jvOUoOJQCf2hNB6w7dT5nFf2q3yPtKE3TccEXil0k,35124
jax/_src/pallas/mosaic_gpu/interpret/interpret_pallas_call.py,sha256=fwUstyZVET3RM5R8yXn17AkZypjmVUqeazrB3Rc4vh0,18358
jax/_src/pallas/mosaic_gpu/interpret/jaxpr_interpret.py,sha256=21Jjg1fP3xm-Vjqdjo4Wt4gtuy-3L7-It_UAQy9p8SQ,20639
jax/_src/pallas/mosaic_gpu/interpret/params.py,sha256=EjoRdl8ZB53RIhiFtmUQMd1g1BPAQvfefwLJm0ZE1N4,2564
jax/_src/pallas/mosaic_gpu/interpret/shared_memory.py,sha256=ZwK-w2v_6otLuBx0SRtX5_Ms29L5bLPZp5I2uBZyN_M,18350
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=OLhkZjvZycT9OYMNS6f_uSOP6blH9PPoI7ngAGCFTW4,163734
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=_Xqlmq06hseGKzzyBgHcPr-T8MxL3_2TDvYKA3arLwo,5769
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=uUEW7TvY7dELg4-S1vq9Dla_Rg13aRvs7fOyd2gGkDQ,40935
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=NVzJ_NX3DQ-PlM2vUmDCzrNQ9EDdPTy4vvR6YVzZHrc,168897
jax/_src/pallas/mosaic_gpu/torch.py,sha256=2u-WZJ0QecyQmDJabF4DhQYPGlUMC3CzD14O4luTX2Q,11383
jax/_src/pallas/mpmd.py,sha256=mVBR55nUBtOW75ggxZtijy1ZEv-MThBa86akBFWsC1E,13040
jax/_src/pallas/pallas_call.py,sha256=cRHTbry53bdau-4_vMvAW5cWt5tIPbb87d-tzg9qkBQ,62748
jax/_src/pallas/primitives.py,sha256=-hkhIU1Eg9SdNQhuSrM3SWCOAfiPjCg9EIISbogIpAY,49564
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/triton/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/triton/__pycache__/lowering.cpython-312.pyc,,
jax/_src/pallas/triton/__pycache__/pallas_call_registration.cpython-312.pyc,,
jax/_src/pallas/triton/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/triton/core.py,sha256=SGp_aTNJuga6uxio9Tz-AUwEx2_ojwnPyCGuxA2BXBQ,1066
jax/_src/pallas/triton/lowering.py,sha256=dqEvR88Kyfudpyuwd_J3ezmRspVFsZpG0CvboqmWlfQ,98101
jax/_src/pallas/triton/pallas_call_registration.py,sha256=oEjKlSaiEYQmu2rOJ97IXRIZ952CAe4iySif1uP_SGM,7154
jax/_src/pallas/triton/primitives.py,sha256=mt2NmVODQn2sgSVyHdvkgmLqjLsAPOuSHNMLbmWB2wc,21074
jax/_src/pallas/utils.py,sha256=1Uuh9wdaTWR5a0NzeXwUx_rX7pDQQecQOrs800VGSmY,15746
jax/_src/partition_spec.py,sha256=VLhSDAJvqtfzM5-11LVRJTkfFSKYvC-tflAUOZJtaaw,6126
jax/_src/path.py,sha256=kX7Mt_vwwncT-CLWsl3ykZTAb6ACHFDHKF_rE_yWLso,2210
jax/_src/pickle_util.py,sha256=LLHPf639whvC2fHOHJDrUOlLSbq5uYg5NIPfQcN0dSY,2283
jax/_src/pjit.py,sha256=_zoUeP0Z-CB-6vzFNbos9ByK2X9vS5gqmyj5bSEgifA,108462
jax/_src/pmap.py,sha256=xhxWpa6t1GmcOilLv8Lm7ad_0_4EZqJzdkUSq5FOfl8,37208
jax/_src/pretty_printer.py,sha256=hGH7oqFMJFG6pZq1nXxQyllc5MPAsQPl8lOw4Lh-e7s,5879
jax/_src/prng.py,sha256=PkqDK36LCnN7b7vffrzPUR8jy3eos7otxscLhSsK3pc,46812
jax/_src/profiler.py,sha256=jC4x1a9QcMynUR_jkg6qBsy2oid2bt_dGqik4nEUBvk,18358
jax/_src/public_test_util.py,sha256=KDMKncw85neOj-dy5mSzUM5ebOChF8AjoTIsW-uNFTg,13231
jax/_src/random.py,sha256=Uf6JOWBIlUrh8gCtUpUAy4yi1oCPkjqi39ZpfUf4hCQ,117390
jax/_src/ref.py,sha256=M-79BkoTpC-w0Xuu4OeScQwZwIhD5oGeX-N7umvlXVI,1396
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/__pycache__/fft.cpython-312.pyc,,
jax/_src/scipy/__pycache__/integrate.cpython-312.pyc,,
jax/_src/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/scipy/__pycache__/ndimage.cpython-312.pyc,,
jax/_src/scipy/__pycache__/signal.cpython-312.pyc,,
jax/_src/scipy/__pycache__/special.cpython-312.pyc,,
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
jax/_src/scipy/cluster/vq.py,sha256=LPoQUfgmh50I-ooN43p7vBiucOgJ29FIji9ZFbFikOc,3039
jax/_src/scipy/fft.py,sha256=LsUgA1Zxsk4qiAZ2kAnwmNZx2ealZxPazHO8NyPHiDM,17320
jax/_src/scipy/integrate.py,sha256=vHyXnhczC0l-Q7ZUWQOIakc1Knqlne4MMShb1DPIHf4,2334
jax/_src/scipy/linalg.py,sha256=oB9TcD9t-pZHDUQh-5XO6rPAq7izHM2pDXiL1cKSLkg,100842
jax/_src/scipy/ndimage.py,sha256=PW7C4Y01H5R6P2iYWleovCUDFTaIEmh9GY7P_JTH2oE,6967
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/line_search.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/minimize.cpython-312.pyc,,
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/signal.py,sha256=0nOq6syw1N2mhc0_te4l5E3wibd8sLTqg222bZbQv3w,47775
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
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/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
jax/_src/scipy/spatial/transform.py,sha256=a10DkF7vLkS4BMm86swDwoqdV-xDtRjCNfho965M5fk,17634
jax/_src/scipy/special.py,sha256=PkaNw3J2IPjGKjtCdwQO0Rf-Us554dHVzaiJv00ZYcY,103653
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/_core.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/beta.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/binom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/expon.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/geom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gumbel_l.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gumbel_r.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/kde.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/norm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/t.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
jax/_src/scipy/stats/_core.py,sha256=N7aWuKeWmwCoZjV0DPe08ycdmONWI0l6eNKwzxyYpDg,10855
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/shard_alike.py,sha256=vY6VYE9jSN1ZeCyxG7ojmTZerG6ujY6ouvcocW7806I,4001
jax/_src/shard_map.py,sha256=TPL3kVBIJzd4ba7GxmVRjmb1vac0j2xCP3dTaQqjS6o,96277
jax/_src/sharding.py,sha256=o1bqvbJcllWjcyHrXgcDpRUiclF8doW3uniyUr5XIzs,8164
jax/_src/sharding_impls.py,sha256=P5Fx59sGId1ndd6zz8P3up9w2-0CxQ1RPC2_aqSWtT0,41111
jax/_src/source_info_util.py,sha256=L29pyn1UcPsYhLXBwhWtkh_3npjjZqBkFV7L1FYONXo,10479
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=E9iBV_9Wo19qLMUm5bkBpTY_3s10oEe-YhY3T40JRhQ,40512
jax/_src/state/__init__.py,sha256=b52iytwSpPVYmiqoKl0I9yaNmmdjmujHf7mulxO8mSg,1026
jax/_src/state/__pycache__/__init__.cpython-312.pyc,,
jax/_src/state/__pycache__/discharge.cpython-312.pyc,,
jax/_src/state/__pycache__/indexing.cpython-312.pyc,,
jax/_src/state/__pycache__/primitives.cpython-312.pyc,,
jax/_src/state/__pycache__/types.cpython-312.pyc,,
jax/_src/state/__pycache__/utils.cpython-312.pyc,,
jax/_src/state/discharge.py,sha256=zDzD4wy4hF3VNUuDbvOEdOSlnjzgA9UKGpddbI_ccQQ,37410
jax/_src/state/indexing.py,sha256=dxIVwXv6Vl78JMHMSWHmi_ZfI3MZjJgU8ocVpsCEmkw,15274
jax/_src/state/primitives.py,sha256=YuJ4DF4eIoLso2KCCoo2AyEwZjHMSvqLiRtgvG46Wqk,45364
jax/_src/state/types.py,sha256=8dVQQlQ3u7WeIY7Z7-M9hbfFcw_59SJEak-feqlgmME,20052
jax/_src/state/utils.py,sha256=LfeYz8GbPN-CLfrmI0PZJmIqtPcW5zLkbnUgB3zZnKQ,3848
jax/_src/stateful_rng.py,sha256=LFSEefh9jhKncyXgCWlnH79VLmgg5BSTWyM2mGH9q8s,10495
jax/_src/test_loader.py,sha256=d3jLbsoAnxZ75Tj0XYoTCTXZMDYaXqrc9quXbrJlY7s,7867
jax/_src/test_multiprocess.py,sha256=bimzIn-EvjH6axxcZUVAZc9e28_HKVpvFBjHWbt1mfE,16400
jax/_src/test_util.py,sha256=sAmYWRANsUQEC74oGl9UIiFvX5H8MkCHaapUNlyenkg,84376
jax/_src/test_warning_util.py,sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE,4074
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/__pycache__/__init__.cpython-312.pyc,,
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/betaln.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/special.cpython-312.pyc,,
jax/_src/third_party/scipy/betaln.py,sha256=_ZYP3Fo1DltN_YeMmR0CkHtE2fDd2eyOdWM0_2cF_kQ,2290
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=DxAw44hcufCKd5bhOD1jz5TpxPBr6wMhjAzO3p4edwQ,3572
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/__pycache__/__init__.cpython-312.pyc,,
jax/_src/tpu/linalg/__init__.py,sha256=RrUPd-WnsNpraGGzWYI-IYTpNcbtQeGqaotRh0Bwwuw,779
jax/_src/tpu/linalg/__pycache__/__init__.cpython-312.pyc,,
jax/_src/tpu/linalg/__pycache__/eigh.cpython-312.pyc,,
jax/_src/tpu/linalg/__pycache__/qdwh.cpython-312.pyc,,
jax/_src/tpu/linalg/__pycache__/stack.cpython-312.pyc,,
jax/_src/tpu/linalg/__pycache__/svd.cpython-312.pyc,,
jax/_src/tpu/linalg/eigh.py,sha256=cE3qdvcAENFkRoqi6rpnDGS13b1hvG9rH7t1RhhoxtI,26595
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/_src/tpu_custom_call.py,sha256=zEQ8_J9YV8jxQZ4CUtRYaKLhr5B74OlbG5CZbWmlo0w,33408
jax/_src/traceback_util.py,sha256=lihnVurzJJ9A5fBQVR1prtCT--rsl_wCWFafFzjmqDA,9092
jax/_src/tree.py,sha256=VgalOSSSFk9B82NHmhAPzkdpBb8H2TLdfW3TVjsxgaY,17117
jax/_src/tree_util.py,sha256=9QXIIhwDt6MA3yzwV11FEakWjYG-B635ZVaPuzEugQk,59986
jax/_src/typing.py,sha256=jdaddrjvthRKzrF6USr77IS2CyOTb-3nj0X4JDq_DZY,3698
jax/_src/util.py,sha256=kHpUYtoO8NubrTOTKnjs-4YGdDy193bhN9R2S3FaOoU,25193
jax/_src/xla_bridge.py,sha256=_-rDc40mEO2rXilqRTtjwhsjLSiGhEuxW_zQ_YXmqlE,45337
jax/_src/xla_metadata.py,sha256=vY8xOd4kU5WCuzbu7NaAgpV-oTdyZT6z-qhcpzMPRDo,4888
jax/_src/xla_metadata_lib.py,sha256=gyRBCCpMfUG-o51PZQ9dO_rafReE7Py_yZOs6cHSY_Q,1474
jax/ad_checkpoint.py,sha256=HaL8w5o0mW8vaPvAsElY583ad8h5HuORLQnAGpbewgE,1542
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/cloud_tpu_init.py,sha256=MqRADba29KMlTngrnC4tfMyWaF_csfewEo9-nNGHlks,1302
jax/collect_profile.py,sha256=Q5CuPjFBUCdZ7W7aSmYwG6BYyBOaF-tboex9ZZho2SU,6148
jax/core.py,sha256=rFumH-pSaL-K-5pmdMfOO36VqpuyA-4GlYreydkg9ys,9372
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=5ZlbQG085r56qJZkGi1tNxeQYTYufJvJYsGyN6N4Rks,1345
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/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/__pycache__/__init__.cpython-312.pyc,,
jax/example_libraries/__pycache__/optimizers.cpython-312.pyc,,
jax/example_libraries/__pycache__/stax.cpython-312.pyc,,
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=eOLn-UowNy-bEXngwaNUMXbUka5qTqmbQXQUYS8Ou00,13834
jax/experimental/__init__.py,sha256=M9QTPXTaRTKhdkHnXriMboYDct1Ycpe5q7wHnzsBBLY,1294
jax/experimental/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/__pycache__/buffer_callback.cpython-312.pyc,,
jax/experimental/__pycache__/checkify.cpython-312.pyc,,
jax/experimental/__pycache__/compute_on.cpython-312.pyc,,
jax/experimental/__pycache__/custom_partitioning.cpython-312.pyc,,
jax/experimental/__pycache__/fused.cpython-312.pyc,,
jax/experimental/__pycache__/hijax.cpython-312.pyc,,
jax/experimental/__pycache__/jet.cpython-312.pyc,,
jax/experimental/__pycache__/layout.cpython-312.pyc,,
jax/experimental/__pycache__/mesh_utils.cpython-312.pyc,,
jax/experimental/__pycache__/multihost_utils.cpython-312.pyc,,
jax/experimental/__pycache__/ode.cpython-312.pyc,,
jax/experimental/__pycache__/pjit.cpython-312.pyc,,
jax/experimental/__pycache__/profiler.cpython-312.pyc,,
jax/experimental/__pycache__/random.cpython-312.pyc,,
jax/experimental/__pycache__/rnn.cpython-312.pyc,,
jax/experimental/__pycache__/scheduling_groups.cpython-312.pyc,,
jax/experimental/__pycache__/serialize_executable.cpython-312.pyc,,
jax/experimental/__pycache__/shard_alike.cpython-312.pyc,,
jax/experimental/__pycache__/shard_map.cpython-312.pyc,,
jax/experimental/__pycache__/topologies.cpython-312.pyc,,
jax/experimental/__pycache__/transfer.cpython-312.pyc,,
jax/experimental/__pycache__/xla_metadata.cpython-312.pyc,,
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/pytree_serialization.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/pytree_serialization_utils.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/serialization.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/tensorstore_impl.cpython-312.pyc,,
jax/experimental/array_serialization/pytree_serialization.py,sha256=jyKjIU9s8yW3mc8_PMmGOCuNNvcpxRcaXcdp6OymVa4,21671
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=Ab91nxoOKCfYSNNLRAbTNu6RPMx_EK4DFMu1Vlr47DY,2765
jax/experimental/array_serialization/serialization.py,sha256=vtr2lDWxBzY2-UrJCasBdCh75wmufo3a7IvEtu5lAyg,13477
jax/experimental/array_serialization/tensorstore_impl.py,sha256=60ehWSJzbNCba0Wfw2GKW1t6DqsxQ2MNlMcFLsrR0Ic,25304
jax/experimental/buffer_callback.py,sha256=ocKnbOO_9n62E2olM0c2a4ep8AHTan4I_AL-eR5VbrQ,765
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
jax/experimental/colocated_python/__init__.py,sha256=wZwEqaKE8AMWBRS0Xp9BubV0YXIAUWp-FQZv0t8gdZ0,940
jax/experimental/colocated_python/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/api.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/func.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/func_backend.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/obj.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/obj_backend.cpython-312.pyc,,
jax/experimental/colocated_python/__pycache__/serialization.cpython-312.pyc,,
jax/experimental/colocated_python/api.py,sha256=Iw02uQfoFrXcDQUra33-N_WxAHj_scN2_qW4KvgKhvs,6998
jax/experimental/colocated_python/func.py,sha256=qiVeaRKfYgjaiq5PL73oICV1cESkObdBb9jd3DIr9FI,25950
jax/experimental/colocated_python/func_backend.py,sha256=JQdiIUulodBQWSazJqSXKMC11UT9l3BbHVr_SveUcaI,1382
jax/experimental/colocated_python/obj.py,sha256=F8wpz1rMQPucSVWXYMJqUxDbRTfFS054xL2cAFAytjM,6401
jax/experimental/colocated_python/obj_backend.py,sha256=1iAlsxTqTtdQYz9XaaRtyFhj9_Oiwpiup2vayCsxLnU,3585
jax/experimental/colocated_python/serialization.py,sha256=iF8Yz4mHdN3VsmgXvc1SLPOJHmaVWF5kbb9MXMl4Fi8,12332
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-312.pyc,,
jax/experimental/compilation_cache/compilation_cache.py,sha256=0r1UkLBOJBsXql-Zbvwg3SPg7hl4RLt3dWBG0UzE1xI,689
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
jax/experimental/fused.py,sha256=qdC1-smPKmc90mdjOk3K4uUaweSfpQiCYOErBoKw8ig,7259
jax/experimental/hijax.py,sha256=9k9LLPkYeG9KToukmJUkShi3nsqpO9lHSetPP-i0EOc,1430
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/jax2tf/__pycache__/call_tf.cpython-312.pyc,,
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-312.pyc,,
jax/experimental/jax2tf/call_tf.py,sha256=VweAFQhebeR-oiVl9qrC1kETTQFc60jt1LIwNLlq0yk,29062
jax/experimental/jax2tf/jax2tf.py,sha256=KqRNOAO6uavq-mCiaWxT72fvkkcLG3_KuOu_hYglyd0,37977
jax/experimental/jet.py,sha256=IhFDIcrgSnTuHaz-p3Mgx2bwKKbkCDE38lKl3Q45FZU,27856
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/key_reuse/__pycache__/_core.cpython-312.pyc,,
jax/experimental/key_reuse/_core.py,sha256=WYxs-suWnrVmkTWU59dvRgBDZFTpF8Q1gwZjkE8JEUU,23555
jax/experimental/layout.py,sha256=fR0lxc08SqRZ0jv9bh6QACBFt_rFP_4naq5y43pU5GQ,738
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/mosaic/__init__.py,sha256=m9n2VxXEoL9plQRUzwShRTvRYAeNyJKba2qzJQp8vQY,1038
jax/experimental/mosaic/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/mosaic/__pycache__/dialects.cpython-312.pyc,,
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=qQzye6rCA1aYrOD2Fh7B58plzklXiTvlo4JzhfWOw5w,4346
jax/experimental/mosaic/gpu/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/constraints.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/core.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/dialect_lowering.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/fragmented_array.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/inference_utils.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/launch_context.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/layout_inference.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/layouts.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/mma.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/mma_utils.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/profiler.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/tcgen05.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/utils.cpython-312.pyc,,
jax/experimental/mosaic/gpu/__pycache__/wgmma.cpython-312.pyc,,
jax/experimental/mosaic/gpu/constraints.py,sha256=zcnf0P2qwnhLldauT-vO_B113HN92FoyM_KxL2KcsVM,36336
jax/experimental/mosaic/gpu/core.py,sha256=F_Y4IFkpHEUr63RgqR49miNmD4Ka0u0UlxcOjjWwP3o,43484
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=nHIe0vUJYgoCTh9-NIHuocKubLMf3RbBLJaTXin7jyI,96075
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=9TakowMy37sZP7aG1LiPkwNDGx_q6KaFbJP6WZnNhFs,179120
jax/experimental/mosaic/gpu/inference_utils.py,sha256=kwXXxiPcLtCXZvgDAkY7DBpaEOyg1MizkeoeyRRPyEo,7595
jax/experimental/mosaic/gpu/launch_context.py,sha256=biQnxQOoB9jzQg-wqpjrFVt1DxSH0NMyr3sHzVhGtPY,81630
jax/experimental/mosaic/gpu/layout_inference.py,sha256=xTjWNjY5oYfRf-BYfYJYmVP_qixQplrVBhbwk1bXerc,92470
jax/experimental/mosaic/gpu/layouts.py,sha256=ATE4vGSaTOyvvRxxt6tKbhmkizMFve1F0P6Bodeo9RU,6677
jax/experimental/mosaic/gpu/mma.py,sha256=YRHR9bHnzrfJingrfdqgKgSsnlhWjpH0993k0mtcaV8,8756
jax/experimental/mosaic/gpu/mma_utils.py,sha256=w9toJF6M-W2_eXu4STFPKgskgwrJfF09US6oOT2L-qw,10743
jax/experimental/mosaic/gpu/profiler.py,sha256=Z5rDGiN_KG1ASvyXiRv2VQKlq5Mkbm9LAd-u0yedjwQ,14624
jax/experimental/mosaic/gpu/tcgen05.py,sha256=MuHFQbv7LF__jc1flFUvPhW4l_8bT3wv0kzDIFQzw14,75016
jax/experimental/mosaic/gpu/utils.py,sha256=1R7jWNN1BVuXlPXujpmCseP3SzQOUCLhHeYXV-ZTXTU,75244
jax/experimental/mosaic/gpu/wgmma.py,sha256=4J_tKf1oU5cnd6FLvfIQn-BpuW2HYG1XCH_MiS4Z8CE,18920
jax/experimental/multihost_utils.py,sha256=mTPB7tWl3XGBeJkloudlakPzKpk6-FW4kKMjnNO5Z4A,25803
jax/experimental/ode.py,sha256=GUm6YPbYAeikEgIY6PgEW-d_x5m2D8NgYdBqUjn3Xf8,10831
jax/experimental/pallas/__init__.py,sha256=6TNp6hUNasUpcu4uouYUac7k1zwTNPhbPCYCHdaSKB0,4044
jax/experimental/pallas/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/fuser.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/mosaic_gpu.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/tpu.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/tpu_sc.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/triton.cpython-312.pyc,,
jax/experimental/pallas/fuser.py,sha256=RfCFDq_FH2guqwVWyX40nIful8flx_yUWEAgx9XKV1k,1382
jax/experimental/pallas/mosaic_gpu.py,sha256=ViVAyA13wYZKgFIM7AjiMHYnXZhTUghF4yqHWE1JC68,7822
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/all_gather_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/attention.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/attention_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/blackwell_matmul_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/blackwell_ragged_dot_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/collective_matmul_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/decode_attention.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/hopper_matmul_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/hopper_mixed_type_matmul_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/layer_norm.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/paged_attention.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/ragged_dot_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/reduce_scatter_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/rms_norm.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/softmax.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/transposed_ragged_dot_mgpu.cpython-312.pyc,,
jax/experimental/pallas/ops/gpu/all_gather_mgpu.py,sha256=W5hmOIVI1hm2EKmslx4GC17nkackvgQM4LI_V0yqu44,8820
jax/experimental/pallas/ops/gpu/attention.py,sha256=_LWBv9V-ck1DKnnd1P7eeh1-4ppK2m8Cp7NBEy-b5Vw,23260
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=NNPsPMJmQf6JzmowbMWs4hKCyhx-hq8JLLbEbyRTdzw,36643
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=iOJBuij4fRuIqNeP5GfczYTvn8MkB7HrLe3vg-ovYWo,12154
jax/experimental/pallas/ops/gpu/blackwell_ragged_dot_mgpu.py,sha256=wr0aQaWtUWIDaDPniNEWchhnNq7ZaoNqht00RwoTccw,15795
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=OQ4Slx9vfIvYjUreG9MrJPteec4ZCfnKZT_zgqMJD-E,11104
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=ZAIdiWPJYpcS0G3-c5_RSSTRo_WgN6VP6L_5Ob8BJ0M,17094
jax/experimental/pallas/ops/gpu/hopper_matmul_mgpu.py,sha256=Os-HkdJyeR4fxXm_Fb-NxJFCbCvfV3hQwxQZCJPx88M,12389
jax/experimental/pallas/ops/gpu/hopper_mixed_type_matmul_mgpu.py,sha256=UvYeRInBYTQ-Fa9IAkqgFPZVr7hztq4WfEPm-obqMNI,12572
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=ifcUo1WIazMEXijjhGbFVQJ6_jdrh218CtVq69RoU3Y,11271
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=YjvtinfuHPxS53-f2Nid4xIhquBdvo3jOT11cmRvWaw,16156
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=ssO1OWgsySLCNHuBa6SIsRadB81jASYxjU8n_9FHzGU,12187
jax/experimental/pallas/ops/gpu/reduce_scatter_mgpu.py,sha256=lRFziJMlTfOlnzOd7V8ZSk1OXg4zVftKoRD0qFFRuVA,9656
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=zaRHNf71YSffSovVsKaBTQCasE2fRuwO6gS1ETtnGII,9934
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/example_kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/matmul.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=9Pas1EzVAHEICa80cumAGWe3k9SiSGP8-_Gq1odTiOY,5564
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=yGAOUMZQY86qv9nVSkt9yIfGWvM78ez2uxU8cA-Yz-g,49886
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/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/common.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/gmm.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/ops.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=-hkn5qlRxNqi9noij1SI0FAl3lodvHwDRYAt5Da_zDw,2100
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=n-n5gH1z2cY8TydJ5vbuthZF_uJ5PJ290_FfLpks0gk,27197
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/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/paged_attention_kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/quantization_utils.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/util.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=sj1uuChc-m_XPVTwENtLqZzlB53zPMkOnEh6NEpIJ8Q,21624
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/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/tuned_block_sizes.cpython-312.pyc,,
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/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/philox.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/prng_utils.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/threefry.cpython-312.pyc,,
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/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask_info.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=-KCOPD1AyO0pJAM4252Cn73JTWviAjmJjWSIcAhDCIY,79031
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/pallas/tpu.py,sha256=ht0mxY4I5ZGI8zRcxtwPCjJcDf9ANn2DVgn6qFX6kVM,7453
jax/experimental/pallas/tpu_sc.py,sha256=zprBAiz-NKefopOEGrAyIeylr0w0wzNDHiQrKh2_86o,2573
jax/experimental/pallas/triton.py,sha256=J5JZFmxqk2uMdv0KOd5E1TdGjcwh05loPwRa0rlkLiQ,1703
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=MoRIyofDONzEYQULCnoI0m8xSpHZkXLPYF0YEW8MV6c,20126
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/roofline/__pycache__/roofline.cpython-312.pyc,,
jax/experimental/roofline/__pycache__/rooflines.cpython-312.pyc,,
jax/experimental/roofline/roofline.py,sha256=-K3C65sPbrdyQftUtIih4zGczC7EYQwoIS8Zx8Uk620,11832
jax/experimental/roofline/rooflines.py,sha256=C4g-_7iV1VilK257r6plGTcMHNDImAgosZFI2HWVdNw,28455
jax/experimental/scheduling_groups.py,sha256=5uq07rakcl83a86-KvmQSrUB1vIunDLOawGZpacnGJ8,8184
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/source_mapper/__init__.py,sha256=ZGiZxIuIkYEpyCw8cMv5bHSCC5MIBr0Elf_Ig_wArYs,1632
jax/experimental/source_mapper/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/source_mapper/__pycache__/common.cpython-312.pyc,,
jax/experimental/source_mapper/__pycache__/generate_map.cpython-312.pyc,,
jax/experimental/source_mapper/__pycache__/hlo.cpython-312.pyc,,
jax/experimental/source_mapper/__pycache__/jaxpr.cpython-312.pyc,,
jax/experimental/source_mapper/__pycache__/mlir.cpython-312.pyc,,
jax/experimental/source_mapper/common.py,sha256=YIA4R4jYtv3XsNPMnwX2xe_xEgoznCt_odGq3141qv8,2476
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=QvxXsDCB2EXyJCYGea1pfn2tMudfc2by8OrOTVJ9lLg,2526
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
jax/experimental/sparse/__init__.py,sha256=8m4B-xYwUfXp9McGxgMMDiyWagueWwekQz1bRLXUzJU,11419
jax/experimental/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/_base.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/_lowerings.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/ad.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/api.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/bcoo.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/bcsr.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/coo.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/csr.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/linalg.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/random.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/test_util.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/transform.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/util.cpython-312.pyc,,
jax/experimental/sparse/_base.py,sha256=e2FJOBxWAnG7DQcyg09rYUEMs88SS9lcy4T04HQ93j4,3336
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=FIDVlEwTrh-QwagIL_sOjuFDYgjPYiDkNqEKzlF1SFo,129480
jax/experimental/sparse/bcsr.py,sha256=sWBeaOUOyv6OBkpjlaPvL4rm3A3Fjn8sSq0g9LuzqB0,41264
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/random.py,sha256=bolWotpIVCyfu9OAKuCZIuXKmaRu51OUP8wUbphQs98,4306
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=RULrAJhx36w2VrRuZmzFHcSJr4vMCnsiyAgLQ5abPhE,37707
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/experimental/topologies.py,sha256=Zjh9jiMSc3wUh9DONTz4Mv0hnChmFifVEmfPppfiwOM,2043
jax/experimental/transfer.py,sha256=jGPl9y0Qb31h_4LPJmi2FTZVghIN0bmk1sqH02tVRm0,2535
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/extend/__init__.py,sha256=RQiAOsXOJAiGrPLx2lsLsN_UicY5aErTNBk0hJygXZ8,1693
jax/extend/__pycache__/__init__.cpython-312.pyc,,
jax/extend/__pycache__/backend.cpython-312.pyc,,
jax/extend/__pycache__/ifrt_programs.cpython-312.pyc,,
jax/extend/__pycache__/linear_util.cpython-312.pyc,,
jax/extend/__pycache__/lowering.cpython-312.pyc,,
jax/extend/__pycache__/pallas.cpython-312.pyc,,
jax/extend/__pycache__/random.cpython-312.pyc,,
jax/extend/__pycache__/sharding.cpython-312.pyc,,
jax/extend/__pycache__/source_info_util.cpython-312.pyc,,
jax/extend/backend.py,sha256=QOL10ZfJMgYh7HWGYPQ1p8y0RD3KEHR075OnsOb_SP8,1370
jax/extend/core/__init__.py,sha256=zUVDmcv4eI16yQrorvqVDZqweb1P5YaXVK3zgjmoFGU,2208
jax/extend/core/__pycache__/__init__.cpython-312.pyc,,
jax/extend/core/__pycache__/primitives.cpython-312.pyc,,
jax/extend/core/primitives.py,sha256=nUqHh5p9uw5TkT_lwJD5cNpuQiEMQGLDUyC_jFOZ1Rs,6337
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/mlir/__init__.py,sha256=wGke5QNa1sJB6PqTteCbJsSaKVwzwPsb0NUKgWVg12M,998
jax/extend/mlir/__pycache__/__init__.cpython-312.pyc,,
jax/extend/mlir/__pycache__/ir.cpython-312.pyc,,
jax/extend/mlir/__pycache__/passmanager.cpython-312.pyc,,
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/arith.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/builtin.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/chlo.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/func.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/math.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/memref.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/mpmd.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/scf.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/sdy.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/vector.cpython-312.pyc,,
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/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
jax/extend/pallas.py,sha256=676BusN50DcftzgrGqar0G9mCCLvR_kNGTukN85b07U,834
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/ffi.py,sha256=ooQYELAXRz1D7RiwDhUhFYUqfwl3Pr-TV5Gw1KMGhAI,1160
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/image/__init__.py,sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI,1027
jax/image/__pycache__/__init__.cpython-312.pyc,,
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/__pycache__/__init__.cpython-312.pyc,,
jax/interpreters/__pycache__/ad.cpython-312.pyc,,
jax/interpreters/__pycache__/batching.cpython-312.pyc,,
jax/interpreters/__pycache__/mlir.cpython-312.pyc,,
jax/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
jax/interpreters/__pycache__/pxla.cpython-312.pyc,,
jax/interpreters/__pycache__/xla.cpython-312.pyc,,
jax/interpreters/ad.py,sha256=SsJRPFEzErWMo0_Hi3v2XYRBAx3a2Qvn2_tJarcOydw,1987
jax/interpreters/batching.py,sha256=3OBiUlKv_oT5r9XEWh6nWkH_WMzPmlfZzwcHaXCpAYU,1642
jax/interpreters/mlir.py,sha256=3L3JlZaDG3cBFOe-Yd0ozz-0FFCMs2__iGKgyWEApCU,2827
jax/interpreters/partial_eval.py,sha256=kGS7qjH2K_hP0gvYhymyV89mm0nGAUQrcb2aDC_Iuew,1180
jax/interpreters/pxla.py,sha256=22oCxQ8pQv7ck7nkhivXhHO_BzqVkU-joemFJW0fLKw,7246
jax/interpreters/xla.py,sha256=TYm_YJ1oYol39-yei642UjRy2W-tHaxgJ34bw1oD9W0,899
jax/lax/__init__.py,sha256=gbxl9s_d7xEK419UnEEJyJ6TteziRsPIxsJZ94JpnzQ,11857
jax/lax/__pycache__/__init__.cpython-312.pyc,,
jax/lax/__pycache__/linalg.cpython-312.pyc,,
jax/lax/linalg.py,sha256=NVBRgXWUJtmL37gjy13AtcU4nNvE4dIsMUpJnoktCyE,1757
jax/lib/__init__.py,sha256=i2G6ZoULU2uH8HBmxui_CTzii_5bW5ibXbHPlIvkONs,660
jax/lib/__pycache__/__init__.cpython-312.pyc,,
jax/monitoring.py,sha256=ik0PsR4rbn1NswE4IJ-i6bpgmV4ueXuaLfAlagUMdNI,1695
jax/nn/__init__.py,sha256=G9v8E63aJYY0rgGMX_bTRB52uBR8a0qEkDQn92zA_z8,1782
jax/nn/__init__.pyi,sha256=aqHPaDoGq0xSkCrUPqBc9N3zoWTTBRFNKeL9qn8t0T4,5042
jax/nn/__pycache__/__init__.cpython-312.pyc,,
jax/nn/__pycache__/initializers.cpython-312.pyc,,
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=32XkpMfZuFXTz_Q5_LiNQfj4R5rZR9w3MfmMMA_9zQ8,12868
jax/numpy/__init__.pyi,sha256=exTGfkcgEBzBzDwrS_tQE0Rz-DtLt6X70b9p82_kioI,44563
jax/numpy/__pycache__/__init__.cpython-312.pyc,,
jax/numpy/__pycache__/fft.cpython-312.pyc,,
jax/numpy/__pycache__/linalg.cpython-312.pyc,,
jax/numpy/fft.py,sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM,1084
jax/numpy/linalg.py,sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI,1419
jax/ops/__init__.py,sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc,870
jax/ops/__pycache__/__init__.cpython-312.pyc,,
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=77I6A8lbOcLprxCBy7XR2arFcGM08p8JnP45Y53CLU8,980
jax/scipy/__init__.py,sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58,1474
jax/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/__pycache__/fft.cpython-312.pyc,,
jax/scipy/__pycache__/integrate.cpython-312.pyc,,
jax/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/scipy/__pycache__/ndimage.cpython-312.pyc,,
jax/scipy/__pycache__/signal.cpython-312.pyc,,
jax/scipy/__pycache__/special.cpython-312.pyc,,
jax/scipy/cluster/__init__.py,sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E,768
jax/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
jax/scipy/cluster/vq.py,sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0,776
jax/scipy/fft.py,sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE,809
jax/scipy/integrate.py,sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y,777
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/interpolate/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/linalg.py,sha256=w5GtiROdnWnfx3bgU3HwyV9BEmWTha6Ey052tqRpzas,1458
jax/scipy/ndimage.py,sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU,788
jax/scipy/optimize/__init__.py,sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s,822
jax/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/sparse/__init__.py,sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo,757
jax/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
jax/scipy/sparse/linalg.py,sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
jax/scipy/spatial/transform.py,sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o,802
jax/scipy/special.py,sha256=Xw4ThOGn48q7Bm6wYdEN2cKHls-A6dP0EiyDqlBVleo,2698
jax/scipy/stats/__init__.py,sha256=_rg_z3SzFW5UNdLOwvm1RgTHyRWiu7LulOePjNqE0Mo,2093
jax/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
jax/scipy/stats/__pycache__/beta.cpython-312.pyc,,
jax/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/binom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
jax/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
jax/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
jax/scipy/stats/__pycache__/expon.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/geom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gumbel_l.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gumbel_r.cpython-312.pyc,,
jax/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
jax/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
jax/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
jax/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
jax/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/norm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
jax/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
jax/scipy/stats/__pycache__/t.cpython-312.pyc,,
jax/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
jax/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
jax/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
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/sharding.py,sha256=eVkz__VMiWKZTqPHvHb1G-PM_BT7hNAPRoUHShCldX8,1491
jax/stages.py,sha256=KOQqCPRCnLKryQUrTRPVRiNuZqyV4chGi_oY2t3HpQ4,1286
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/__pycache__/__init__.cpython-312.pyc,,
jax/tools/__pycache__/jax_to_ir.cpython-312.pyc,,
jax/tools/__pycache__/pgo_nsys_converter.cpython-312.pyc,,
jax/tools/jax_to_ir.py,sha256=9DOHrGu7OpKwIKU1ViXEsDa33RaN-WDZPrAKAkb16cY,9185
jax/tools/pgo_nsys_converter.py,sha256=C2gz5kMqZRqI8t11HOswanwcGDlrAHF_PTh-p4hkVrI,3407
jax/tree.py,sha256=FfouNfVkwowRidqI_8-zceCtFFxAAuf1JGzCFIZGjdQ,1174
jax/tree_util.py,sha256=WPybt6SfF0S7qjP2zxzd-bZ73wuuddnxUPxAuEONUNg,3223
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/version.py,sha256=YImY7tRfpNSwwvM7aEmEdEq2q8LKMHsIybKpXwKv1L0,6822
