jax-0.6.2.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.6.2.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.6.2.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.6.2.dist-info/METADATA,sha256=tXu99kxksiCOsT9r2djs6BW2zbvyYhU_BZbidhpJHx0,13123
jax-0.6.2.dist-info/RECORD,,
jax-0.6.2.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
jax-0.6.2.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=LD82Acv1D4o9YeBUcnz7unFDFzjLRVkDToNoifV2bvQ,9291
jax/__pycache__/__init__.cpython-311.pyc,,
jax/__pycache__/ad_checkpoint.cpython-311.pyc,,
jax/__pycache__/api_util.cpython-311.pyc,,
jax/__pycache__/cloud_tpu_init.cpython-311.pyc,,
jax/__pycache__/collect_profile.cpython-311.pyc,,
jax/__pycache__/core.cpython-311.pyc,,
jax/__pycache__/custom_batching.cpython-311.pyc,,
jax/__pycache__/custom_derivatives.cpython-311.pyc,,
jax/__pycache__/custom_transpose.cpython-311.pyc,,
jax/__pycache__/debug.cpython-311.pyc,,
jax/__pycache__/distributed.cpython-311.pyc,,
jax/__pycache__/dlpack.cpython-311.pyc,,
jax/__pycache__/dtypes.cpython-311.pyc,,
jax/__pycache__/errors.cpython-311.pyc,,
jax/__pycache__/export.cpython-311.pyc,,
jax/__pycache__/ffi.cpython-311.pyc,,
jax/__pycache__/flatten_util.cpython-311.pyc,,
jax/__pycache__/monitoring.cpython-311.pyc,,
jax/__pycache__/profiler.cpython-311.pyc,,
jax/__pycache__/random.cpython-311.pyc,,
jax/__pycache__/sharding.cpython-311.pyc,,
jax/__pycache__/stages.cpython-311.pyc,,
jax/__pycache__/test_util.cpython-311.pyc,,
jax/__pycache__/tree.cpython-311.pyc,,
jax/__pycache__/tree_util.cpython-311.pyc,,
jax/__pycache__/typing.cpython-311.pyc,,
jax/__pycache__/util.cpython-311.pyc,,
jax/__pycache__/version.cpython-311.pyc,,
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/__pycache__/__init__.cpython-311.pyc,,
jax/_src/__pycache__/abstract_arrays.cpython-311.pyc,,
jax/_src/__pycache__/ad_checkpoint.cpython-311.pyc,,
jax/_src/__pycache__/ad_util.cpython-311.pyc,,
jax/_src/__pycache__/api.cpython-311.pyc,,
jax/_src/__pycache__/api_util.cpython-311.pyc,,
jax/_src/__pycache__/array.cpython-311.pyc,,
jax/_src/__pycache__/attrs.cpython-311.pyc,,
jax/_src/__pycache__/basearray.cpython-311.pyc,,
jax/_src/__pycache__/blocked_sampler.cpython-311.pyc,,
jax/_src/__pycache__/buffer_callback.cpython-311.pyc,,
jax/_src/__pycache__/cache_key.cpython-311.pyc,,
jax/_src/__pycache__/callback.cpython-311.pyc,,
jax/_src/__pycache__/checkify.cpython-311.pyc,,
jax/_src/__pycache__/cloud_tpu_init.cpython-311.pyc,,
jax/_src/__pycache__/compilation_cache.cpython-311.pyc,,
jax/_src/__pycache__/compilation_cache_interface.cpython-311.pyc,,
jax/_src/__pycache__/compiler.cpython-311.pyc,,
jax/_src/__pycache__/compute_on.cpython-311.pyc,,
jax/_src/__pycache__/config.cpython-311.pyc,,
jax/_src/__pycache__/core.cpython-311.pyc,,
jax/_src/__pycache__/custom_api_util.cpython-311.pyc,,
jax/_src/__pycache__/custom_batching.cpython-311.pyc,,
jax/_src/__pycache__/custom_dce.cpython-311.pyc,,
jax/_src/__pycache__/custom_derivatives.cpython-311.pyc,,
jax/_src/__pycache__/custom_partitioning.cpython-311.pyc,,
jax/_src/__pycache__/custom_partitioning_sharding_rule.cpython-311.pyc,,
jax/_src/__pycache__/custom_transpose.cpython-311.pyc,,
jax/_src/__pycache__/debugging.cpython-311.pyc,,
jax/_src/__pycache__/deprecations.cpython-311.pyc,,
jax/_src/__pycache__/dispatch.cpython-311.pyc,,
jax/_src/__pycache__/distributed.cpython-311.pyc,,
jax/_src/__pycache__/dlpack.cpython-311.pyc,,
jax/_src/__pycache__/dtypes.cpython-311.pyc,,
jax/_src/__pycache__/earray.cpython-311.pyc,,
jax/_src/__pycache__/effects.cpython-311.pyc,,
jax/_src/__pycache__/environment_info.cpython-311.pyc,,
jax/_src/__pycache__/error_check.cpython-311.pyc,,
jax/_src/__pycache__/errors.cpython-311.pyc,,
jax/_src/__pycache__/ffi.cpython-311.pyc,,
jax/_src/__pycache__/flatten_util.cpython-311.pyc,,
jax/_src/__pycache__/frozen_dict.cpython-311.pyc,,
jax/_src/__pycache__/hardware_utils.cpython-311.pyc,,
jax/_src/__pycache__/hashable_array.cpython-311.pyc,,
jax/_src/__pycache__/jaxpr_util.cpython-311.pyc,,
jax/_src/__pycache__/lax_reference.cpython-311.pyc,,
jax/_src/__pycache__/layout.cpython-311.pyc,,
jax/_src/__pycache__/lazy_loader.cpython-311.pyc,,
jax/_src/__pycache__/linear_util.cpython-311.pyc,,
jax/_src/__pycache__/logging_config.cpython-311.pyc,,
jax/_src/__pycache__/lru_cache.cpython-311.pyc,,
jax/_src/__pycache__/mesh.cpython-311.pyc,,
jax/_src/__pycache__/mesh_utils.cpython-311.pyc,,
jax/_src/__pycache__/monitoring.cpython-311.pyc,,
jax/_src/__pycache__/named_sharding.cpython-311.pyc,,
jax/_src/__pycache__/op_shardings.cpython-311.pyc,,
jax/_src/__pycache__/partition_spec.cpython-311.pyc,,
jax/_src/__pycache__/path.cpython-311.pyc,,
jax/_src/__pycache__/pickle_util.cpython-311.pyc,,
jax/_src/__pycache__/pjit.cpython-311.pyc,,
jax/_src/__pycache__/pretty_printer.cpython-311.pyc,,
jax/_src/__pycache__/prng.cpython-311.pyc,,
jax/_src/__pycache__/profiler.cpython-311.pyc,,
jax/_src/__pycache__/public_test_util.cpython-311.pyc,,
jax/_src/__pycache__/random.cpython-311.pyc,,
jax/_src/__pycache__/shard_alike.cpython-311.pyc,,
jax/_src/__pycache__/shard_map.cpython-311.pyc,,
jax/_src/__pycache__/sharding.cpython-311.pyc,,
jax/_src/__pycache__/sharding_impls.cpython-311.pyc,,
jax/_src/__pycache__/sharding_specs.cpython-311.pyc,,
jax/_src/__pycache__/source_info_util.cpython-311.pyc,,
jax/_src/__pycache__/sourcemap.cpython-311.pyc,,
jax/_src/__pycache__/stages.cpython-311.pyc,,
jax/_src/__pycache__/test_loader.cpython-311.pyc,,
jax/_src/__pycache__/test_multiprocess.cpython-311.pyc,,
jax/_src/__pycache__/test_util.cpython-311.pyc,,
jax/_src/__pycache__/test_warning_util.cpython-311.pyc,,
jax/_src/__pycache__/tpu_custom_call.cpython-311.pyc,,
jax/_src/__pycache__/traceback_util.cpython-311.pyc,,
jax/_src/__pycache__/tree.cpython-311.pyc,,
jax/_src/__pycache__/tree_util.cpython-311.pyc,,
jax/_src/__pycache__/typing.cpython-311.pyc,,
jax/_src/__pycache__/util.cpython-311.pyc,,
jax/_src/__pycache__/xla_bridge.cpython-311.pyc,,
jax/_src/__pycache__/xla_metadata.cpython-311.pyc,,
jax/_src/abstract_arrays.py,sha256=1kxQA1qnwmri9_GeZHerQmghGZ-zHOy-etSyGHIhTIE,2889
jax/_src/ad_checkpoint.py,sha256=yRK5mpBVlDjivqmVgBp_sfVP8KfFvm-u0Bg5N0n01Hw,37058
jax/_src/ad_util.py,sha256=i1Oj_yGWZXcPb33GVukSYkLskVN3L4Kt0AUZ2F-i8wc,4446
jax/_src/api.py,sha256=QSSywLTQv0pN_RyNZGkECdyHrQ0ImuWAB4uRdZJDnU0,138017
jax/_src/api_util.py,sha256=j8VmM3_7ggUf6dPJYP2gATAQi3dWk0ks01XJLpwWCHE,30733
jax/_src/array.py,sha256=jTqMr6kPZvd3uFpF01U4c-4jUYHSZbIqMKjCtj_cAc8,49562
jax/_src/attrs.py,sha256=guRVvHWjaiyCauxtHOJY3gjIHrp4ts6U5JinaBpCYbo,15498
jax/_src/basearray.py,sha256=FQOcRwUhh1w4byHijEzCZAVqa8Tukr3bSr4fEqZz1q4,6904
jax/_src/basearray.pyi,sha256=2AAKVkmJ6c4fLr2N-SA0AiLNUuCKqOmO_XMRf6jsuJI,13467
jax/_src/blocked_sampler.py,sha256=OtV8uaYBfdGsWOpYD6LG9JPK1yknKkrJuLPAJ_LJPeM,5915
jax/_src/buffer_callback.py,sha256=1qpEUw7Lku1JrNIU-vKW36oYO47aPkAxmJPyt7YmJog,10533
jax/_src/cache_key.py,sha256=Yon7NspNxWKIm0yBJso5j6I4Z6mnKFLmAIg_g0rc228,12923
jax/_src/callback.py,sha256=cRLRf6LqKtnXIYZ3WONrZrzf0c6ZtFCvxWvAu44vXss,33140
jax/_src/checkify.py,sha256=hZHK1ixcQroaaKq9By7QhMn5NWVENscCNe8pVsepu48,57782
jax/_src/cloud_tpu_init.py,sha256=Ks9-0g8s45dOGPs3cHVa7Ddjy-Kqlbag9JAFBWyPlr0,4875
jax/_src/clusters/__init__.py,sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94,1283
jax/_src/clusters/__pycache__/__init__.cpython-311.pyc,,
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/k8s_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/mpi4py_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/ompi_cluster.cpython-311.pyc,,
jax/_src/clusters/__pycache__/slurm_cluster.cpython-311.pyc,,
jax/_src/clusters/cloud_tpu_cluster.py,sha256=SuB7FuOWChVC2dpLeKmNOYYpd49Ke1c0UgLoMJe4c3E,8522
jax/_src/clusters/cluster.py,sha256=raW-A1TKgHDyqZeMfv3lgdm4ImDHxrzoMnFAOfgiK4M,5858
jax/_src/clusters/k8s_cluster.py,sha256=uo4QCCwpqixEB3Mo_28fPmPpCkze-fQb1PK8I_WIGy4,9279
jax/_src/clusters/mpi4py_cluster.py,sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg,2608
jax/_src/clusters/ompi_cluster.py,sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ,2317
jax/_src/clusters/slurm_cluster.py,sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0,2321
jax/_src/compilation_cache.py,sha256=WfrNBOZSpIyxrKFIzdgq7pkzekYy_Nxa_eFY6CttL5w,12006
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=yvNFzMoaJgd7chPd0kaQmQ4Dwkaq694nB8QGjr_huXY,33211
jax/_src/compute_on.py,sha256=0b74TSV1R31kJTR53thafdtB0sYiFGqmRoAbZwAFcps,1911
jax/_src/config.py,sha256=TrdCH1B_WTUeYqbfVEFQN_zwyra_C9-qdvuFmKz23YU,70329
jax/_src/core.py,sha256=TM_E3VA3EmmnX76_eoN3bSwyQaCxddP_Q0NVyJ565g4,128945
jax/_src/cudnn/__init__.py,sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A,631
jax/_src/cudnn/__pycache__/__init__.cpython-311.pyc,,
jax/_src/cudnn/__pycache__/fused_attention_stablehlo.cpython-311.pyc,,
jax/_src/cudnn/__pycache__/fusion.cpython-311.pyc,,
jax/_src/cudnn/__pycache__/scaled_matmul_stablehlo.cpython-311.pyc,,
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=83_OlPbZ2vX57Ut2tMcPplbSESDxNtlbG48Kf222ezs,78802
jax/_src/cudnn/fusion.py,sha256=KO7fo_thf5-sy1i_9HUrbTpt5cyOlS6zbu6k2RPonk8,2809
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=AKIxjX6NvXFp4YPQhenBjUTDpt0L5IbYBYdqW3xICTs,23901
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=pulnaqQfqfGA9YnwGvzyaZEhSJC8ktYZZmlyKAqshyc,15240
jax/_src/custom_dce.py,sha256=cz9n_6n8JuKtur3a_EryePp6bN9aJqf8wrRN0QlTs70,17478
jax/_src/custom_derivatives.py,sha256=thae08fHF2oRoeOxr_CbYOXve9PQ2D0edUk_bQbMFcA,79950
jax/_src/custom_partitioning.py,sha256=Q8HTkO6QjECgc_RSpp75x4nRieCBqWm3BYX47wXJgn4,26170
jax/_src/custom_partitioning_sharding_rule.py,sha256=qijmSIw4pKpyqyTb_mzTMYCnDHV3eMb-yzixrLUi7XY,18026
jax/_src/custom_transpose.py,sha256=UpDGexoY0QOuwJvQS5Sk3ZCJQET1CnQKCJba8XlZ5qI,9195
jax/_src/debugger/__init__.py,sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q,899
jax/_src/debugger/__pycache__/__init__.cpython-311.pyc,,
jax/_src/debugger/__pycache__/cli_debugger.cpython-311.pyc,,
jax/_src/debugger/__pycache__/colab_debugger.cpython-311.pyc,,
jax/_src/debugger/__pycache__/colab_lib.cpython-311.pyc,,
jax/_src/debugger/__pycache__/core.cpython-311.pyc,,
jax/_src/debugger/__pycache__/web_debugger.cpython-311.pyc,,
jax/_src/debugger/cli_debugger.py,sha256=bzn-5Rd4C9WwGsVret8gbFNaLpXAf0SzvgltXNuZrcU,4778
jax/_src/debugger/colab_debugger.py,sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA,7823
jax/_src/debugger/colab_lib.py,sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U,4298
jax/_src/debugger/core.py,sha256=WilcRMqpzssKa5YL_P7SFtKaUVUaXqDutxO_N-ZRnp0,8593
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
jax/_src/debugging.py,sha256=qgEF63pSywSBU_fbpgHl9PR4V3E8MwncCopY7T5ibJ0,28819
jax/_src/deprecations.py,sha256=vBQeRe1h_4Okm3lJV5Xpu5yFSb9Z-nLX8Ieh5c7Yypg,4795
jax/_src/dispatch.py,sha256=k27zGW7WZBaFWUXjxOKLT7ach2BxmpgaMhM6L4abA4M,26907
jax/_src/distributed.py,sha256=8rAhhjxfe-1G4ZW0oKEqkt2RZYS2o1RaqUZeV4Q9c0E,13693
jax/_src/dlpack.py,sha256=S8bG2Xjgxq_xrbE4GLnhi2DIoYZtBymRPiU5_qr1yl0,11411
jax/_src/dtypes.py,sha256=yCM5YOoFke7jkSLF9JwK5qWAwt9YxpIThF-nQW7RnYw,35725
jax/_src/earray.py,sha256=6KsYjNLzWaT1S0IKDwmZMAsa94E8yg9TU2Yb3-2QN70,4412
jax/_src/effects.py,sha256=Jg3hDB313Vl1CN_khELWMI1mE2kJ8MzLXRHf2b31ArM,4736
jax/_src/environment_info.py,sha256=IoRky2nPXfdS_VPKvpPhzW0mTShyfUJiAl6SBleErh4,1986
jax/_src/error_check.py,sha256=X-FFObANtgQB_hZqKrjVCDvJ1hjZZDgSuT9fThPXtwQ,12727
jax/_src/errors.py,sha256=Vp2gp5kQxtoss9aby-vpGwO5wBxRndvddVD_3-UWyZ4,24710
jax/_src/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/__pycache__/__init__.cpython-311.pyc,,
jax/_src/export/__pycache__/_export.cpython-311.pyc,,
jax/_src/export/__pycache__/serialization.cpython-311.pyc,,
jax/_src/export/__pycache__/serialization_generated.cpython-311.pyc,,
jax/_src/export/__pycache__/shape_poly.cpython-311.pyc,,
jax/_src/export/__pycache__/shape_poly_decision.cpython-311.pyc,,
jax/_src/export/_export.py,sha256=VR8o8Xg380sJOHS4vUmK96ZMNyDDTJ4vFGgcyeGgsyU,69828
jax/_src/export/serialization.py,sha256=0fYNO50OjU8TWRC1_wHJm_z1pFbLnaGQai3Tif5KluU,19466
jax/_src/export/serialization_generated.py,sha256=pCU6x5XxV-2IUyRkA036MrhMY1RvbKWdefXwOySy7Cw,27869
jax/_src/export/shape_poly.py,sha256=4NpQwchghuErOJLwFDU6jXlN1oL-7fLSxDOpsrghD7I,85943
jax/_src/export/shape_poly_decision.py,sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc,20476
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-311.pyc,,
jax/_src/extend/__pycache__/random.cpython-311.pyc,,
jax/_src/extend/random.py,sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw,1258
jax/_src/ffi.py,sha256=m7JbkKFe3yCv6onjv-NwfyofRWkouw1klxk1em2ERVM,28810
jax/_src/flatten_util.py,sha256=WSVHRWlDiJm4gdnEIK3csN9aOxLuvWdLe8HXJfuQPMA,3550
jax/_src/frozen_dict.py,sha256=22m-h6SepCRV95hyPZpKfGyQigoE3lG7HHvum77Eu0E,1342
jax/_src/hardware_utils.py,sha256=0-8OyFvMYu5l-C0e3VObpQMkLGg-BBtzP56CAdFEIxE,2295
jax/_src/hashable_array.py,sha256=4ff9eVDYR7DFK3l9nNF0btGecguXonn1AzeWviiJQcQ,1036
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/__pycache__/__init__.cpython-311.pyc,,
jax/_src/image/__pycache__/scale.cpython-311.pyc,,
jax/_src/image/scale.py,sha256=XL1VSkCM4gtOOFurIElk8nXPfUB802nw3P8wlqquP4Q,13774
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/__pycache__/__init__.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-311.pyc,,
jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-311.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-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/annotate_data_placement.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_cholesky_lapack_potrf.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eig_lapack_geev.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_eigh_lapack_syev.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_hessenberg_lapack_gehrd.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_lu_lapack_getrf.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_qr_lapack_geqrf.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_schur_lapack_gees.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_svd_lapack_gesdd.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_triangular_solve_blas_trsm.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_lapack_sytrd_hetrd.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cpu_tridiagonal_solve_lapack_gtsv.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_eigh_cusolver_syev.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_cusolver_getrf.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_lu_pivots_to_permutation.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_qr_cusolver_geqrf.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_svd_cusolver_gesvd.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_threefry2x32.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_cusolver_sytrd.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/cuda_tridiagonal_solve.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/rocm_eigh_hipsolver_syev.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/shardy_sharding_ops_with_different_meshes.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_approx_top_k.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_rng_bit_generator.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/stablehlo_dynamic_top_k.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_ApproxTopK.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Eigh.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Lu.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Qr.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_Sharding.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/__pycache__/tpu_stablehlo_dynamic_reduce_window.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/annotate_data_placement.py,sha256=PZiDa_599R0BfRekXxtgYygrfGXKWEsTHrzb73Yr5is,14496
jax/_src/internal_test_util/export_back_compat_test_data/cpu_cholesky_lapack_potrf.py,sha256=h78iCf_nVYFnPTpAJByDclfoQMJv9GBxQrBd1slnq9o,45926
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eig_lapack_geev.py,sha256=zIJnIA_e3r4-xrpYTQMtscVEjnCypDuJVBWeSMRT1ms,35773
jax/_src/internal_test_util/export_back_compat_test_data/cpu_eigh_lapack_syev.py,sha256=WvhpYwzpKTxHsx4TBJjlYm9XXlT_u6w1DyYh8OzcpQ4,58702
jax/_src/internal_test_util/export_back_compat_test_data/cpu_hessenberg_lapack_gehrd.py,sha256=Sav1CtBDRZ4SPzvbxS9LdXShoGBVgUmvyOb4dg8p15I,37237
jax/_src/internal_test_util/export_back_compat_test_data/cpu_lu_lapack_getrf.py,sha256=2G9cFdWm6Ppz-p0v5XSW-XvBgtA1o42ajE_ysyW3wro,80469
jax/_src/internal_test_util/export_back_compat_test_data/cpu_qr_lapack_geqrf.py,sha256=ht61muzEXcrW-835kxdIhwPiK60AfY2ZisgzE0WOjLk,28549
jax/_src/internal_test_util/export_back_compat_test_data/cpu_schur_lapack_gees.py,sha256=-yBUW2hATIqD6Q4fRyOyu1yFjYOVTfHkupyeKAyn_i4,25011
jax/_src/internal_test_util/export_back_compat_test_data/cpu_svd_lapack_gesdd.py,sha256=pbGLtB8oUnS1yUVLHfCURVySu2I0ozAJaXOw3bPWWK8,44237
jax/_src/internal_test_util/export_back_compat_test_data/cpu_triangular_solve_blas_trsm.py,sha256=2jc16YKe1KNaQ68KsJbRtB1M1hpqdvvMcHOfN6PmKLA,19047
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_lapack_sytrd_hetrd.py,sha256=4tpXxuEDCrDK5SHtZ8Z-gFGJFHtAew6TJ6Wfa5tyPk0,58836
jax/_src/internal_test_util/export_back_compat_test_data/cpu_tridiagonal_solve_lapack_gtsv.py,sha256=o8jisAqvk02Uyd9ExxujqvEELIwHHm7ngtvwqmWtXRs,31747
jax/_src/internal_test_util/export_back_compat_test_data/cuda_eigh_cusolver_syev.py,sha256=Ggvl-nVWZvHLVbQXhoJGyijq4MHw0lu9GfvnBM-BklA,43004
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_cusolver_getrf.py,sha256=BKwHumIdGqT2yADKuI6MmaC9Lzyr5R1T1UUgJlq6CIY,28679
jax/_src/internal_test_util/export_back_compat_test_data/cuda_lu_pivots_to_permutation.py,sha256=9VBfh6UuV5OPxxHkRrSnUBHNm0XSElBtoqncKlwUCf0,4136
jax/_src/internal_test_util/export_back_compat_test_data/cuda_qr_cusolver_geqrf.py,sha256=eLLv7JISa9c1h2FSbxLJEb5mVLf90QaXW1c93spk5SU,31545
jax/_src/internal_test_util/export_back_compat_test_data/cuda_svd_cusolver_gesvd.py,sha256=R97n-YBNb8jGCGRkPkzoLe96Sk0OmyM7ykPpcq4_tpk,80009
jax/_src/internal_test_util/export_back_compat_test_data/cuda_threefry2x32.py,sha256=1XTgzOBzbiaxi_MeAKtww5FpZ8DgoQ4lZ58w_XeeZoI,14884
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_cusolver_sytrd.py,sha256=M7LMg3Gcj7PMt336U7KvsPrJ1BG8UGevYhb7dDeD954,39010
jax/_src/internal_test_util/export_back_compat_test_data/cuda_tridiagonal_solve.py,sha256=sAipfYHp6DuXPRrt9sz9MtLV80lDRU3LLjuddHFU0Gk,7731
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-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_gpu_add_one.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_matmul.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/mosaic_semaphore_dma.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/__pycache__/triton_add_one.cpython-311.pyc,,
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_gpu_add_one.py,sha256=BdloBuwb9-98TIDs7IORnhG-CSyITzRynkeNn26ipEg,102043
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_matmul.py,sha256=EZfWq9HQ8HHkz7CUg3i9ps0B4Kw1YmSnGsBYXxpLCAY,31178
jax/_src/internal_test_util/export_back_compat_test_data/pallas/mosaic_semaphore_dma.py,sha256=ogBS9Lpa2MqBju6EX2I9p1K_PQemWOQEh4t5_ugUWCQ,16882
jax/_src/internal_test_util/export_back_compat_test_data/pallas/triton_add_one.py,sha256=t-7TInuIX49zOqHTa9HZXEhnbPY8Ae8v1Dy-kYspRO0,10018
jax/_src/internal_test_util/export_back_compat_test_data/rocm_eigh_hipsolver_syev.py,sha256=_tei01kakwhNzxyUD57wtXX6SZ7J4MqIRPwyM79Xi6k,114989
jax/_src/internal_test_util/export_back_compat_test_data/shardy_sharding_ops_with_different_meshes.py,sha256=0EVxa7Z3_TfR-LWO_cUT4qjGgb48vOz53UGGny7iMfE,14786
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_approx_top_k.py,sha256=oekOSBpWgOtEYFdMS8FBkncdKzkQUYQ7V_QiY-zJjcg,13027
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_rng_bit_generator.py,sha256=jQPA3m1YSVp6PE78bXkckyBNqgYzdWUH5yl-wtXEEyE,11386
jax/_src/internal_test_util/export_back_compat_test_data/stablehlo_dynamic_top_k.py,sha256=iinBbhX0A06ODm2gWUGY6ZxOkXdO-1y_ss1XeG74a6E,14132
jax/_src/internal_test_util/export_back_compat_test_data/tpu_ApproxTopK.py,sha256=M200M40lgHqXk58UWeKeFzAOb4NSmbhmLo8ehbijAQE,12524
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Eigh.py,sha256=Imsy4t5CKQR4Zy1bFCyvmXiCdVSRgpnS64SMwY7Na3E,10121
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Lu.py,sha256=mSP47CuGZ1dBQviBSSFZD8yk2tXwcmVC2gC3rNQj_Eo,6150
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Qr.py,sha256=UekaagWJ4d_fWfrzFpXPHyRQPlGQeDEtk__gcmGwNpg,10203
jax/_src/internal_test_util/export_back_compat_test_data/tpu_Sharding.py,sha256=F_QjAPzA2FuQXhJUHIwTvBDZl8n0qj2_Il_QBEa-kas,11052
jax/_src/internal_test_util/export_back_compat_test_data/tpu_stablehlo_dynamic_reduce_window.py,sha256=JnDt0u66UpQNcMAzZthYKcctBFtyRvhdY7pYhH0KIo4,22961
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=ztpe2MofjCMxiUk_1T0KhHAqYOk0qrkcFUU-5ou9JLc,15676
jax/_src/internal_test_util/lax_test_util.py,sha256=_83LJKAcbyDkSrT2D2k4x1waaBHa37JMNISX6veyGHU,12998
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-311.pyc,,
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-311.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=U7PkgvUN_JDeyaw5T66M-9GgWStCRFjMacGIQEakOSU,126282
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/__pycache__/__init__.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/ad.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/batching.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/mlir.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/partial_eval.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/pxla.cpython-311.pyc,,
jax/_src/interpreters/__pycache__/xla.cpython-311.pyc,,
jax/_src/interpreters/ad.py,sha256=V5stFH6IbCzeoP5Ad0ZVGTpUGf2d9GoezODSdTtMEb4,58581
jax/_src/interpreters/batching.py,sha256=6IPzUSG0I2c34Qm2q2zyWQcu1Fo0GSU8tPHAu4KI-NU,51724
jax/_src/interpreters/mlir.py,sha256=XGOvZVyqWmOvwnHaSweMdQxvBJsimBkt0u2gofBXd-s,129090
jax/_src/interpreters/partial_eval.py,sha256=Jb5qY5sJRdA3urA9KVneK-ZYbpSLmAcFWWMXN-EHh_c,122533
jax/_src/interpreters/pxla.py,sha256=Tq-EXkLMKzKKZrCF2AVq66WRtsuZ8Vj4jK-dV6u4PA8,135973
jax/_src/interpreters/xla.py,sha256=Me2RiFbqW_hF-RUOBNxQEGWkPhqYfYjH-12eAQbJHb0,4981
jax/_src/jaxpr_util.py,sha256=igIrtHU_7BtJVfLwEkfslVSV43S9bnvmljKneAhOvQA,8532
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lax/__pycache__/ann.cpython-311.pyc,,
jax/_src/lax/__pycache__/convolution.cpython-311.pyc,,
jax/_src/lax/__pycache__/fft.cpython-311.pyc,,
jax/_src/lax/__pycache__/lax.cpython-311.pyc,,
jax/_src/lax/__pycache__/linalg.cpython-311.pyc,,
jax/_src/lax/__pycache__/other.cpython-311.pyc,,
jax/_src/lax/__pycache__/parallel.cpython-311.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-311.pyc,,
jax/_src/lax/__pycache__/special.cpython-311.pyc,,
jax/_src/lax/__pycache__/utils.cpython-311.pyc,,
jax/_src/lax/__pycache__/windowed_reductions.cpython-311.pyc,,
jax/_src/lax/ann.py,sha256=drKiFUIxy8wodSZxGqpYd91Lq_M-WFyqM4NSSn1DqGc,16786
jax/_src/lax/control_flow/__init__.py,sha256=WRnqzy4ec591tqipY4GcyXFzniF0njoIVkP0HhKGbiI,2244
jax/_src/lax/control_flow/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/common.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/for_loop.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/loops.cpython-311.pyc,,
jax/_src/lax/control_flow/__pycache__/solves.cpython-311.pyc,,
jax/_src/lax/control_flow/common.py,sha256=8rq9e1Pp87oAIdepv9f0sp8HlLK2akStBWNQ9Ik9WQA,11534
jax/_src/lax/control_flow/conditionals.py,sha256=pRECf355U24pFi9Z7NpA8S5m8v8IjCXzd5f3l-HEY1Y,50891
jax/_src/lax/control_flow/for_loop.py,sha256=XtqDTv8OmHFHVKlNW8qWqgyyyJk11s9mjT3BboCrNwY,36377
jax/_src/lax/control_flow/loops.py,sha256=CRRWGTjfDoH8JYwLdYUIFAFJe65tS-JrKcVGEUEekO4,132594
jax/_src/lax/control_flow/solves.py,sha256=9C-rBd_2PaVdefP7FHeGRmIEZ8e7hlf5JJg-rlaWpv0,20140
jax/_src/lax/convolution.py,sha256=rL-mM5hd7g7ounRZLItM2zZqE4BlOOJDSMhSpMmqooc,45958
jax/_src/lax/fft.py,sha256=3jloTWfLPoPvHbUfHde9Dqt-E96D5UXDmqgBxPEAsOg,7277
jax/_src/lax/lax.py,sha256=EakRTxiKvgYTcA7Awz7s-HMdSGrM25NrLj8EHtSGk4I,362044
jax/_src/lax/linalg.py,sha256=L1Lo2ufyXmT0q0IC3a-E2hdTpyaUmaSEcov1Ch0oCPs,101863
jax/_src/lax/other.py,sha256=CzM6WbfsOKVlEDkGEd6SMdhftIx3GsIk4Wc_Zjh4n3E,13880
jax/_src/lax/parallel.py,sha256=P9fN30E3hQEMVWptIArPBjc0ngnWl24hvrTW78xRr1Y,92373
jax/_src/lax/slicing.py,sha256=JA7mHLFTslJFetcZROBvYGXmEJBlsPjhNWcdQcvjOkI,134815
jax/_src/lax/special.py,sha256=yIjzWwwZ6IetGqkpQbEvyLtbcKj09HBI0VUS19yi7fc,28856
jax/_src/lax/utils.py,sha256=6g-yPv2f8m04wOWxqYrct2zT-S3wwLNUD0W4XfZJgYg,8184
jax/_src/lax/windowed_reductions.py,sha256=sJAt0GtGqMroUbUgabrFLWgMF0Y2l7GFAfvopbBgGQw,45255
jax/_src/lax_reference.py,sha256=YPZTCI52PssuXAEnMn4r44ZZ2jDHB0wYj8d_fJDrCnw,19282
jax/_src/layout.py,sha256=hh8CdqYW1CAF0JJ2i9g2D1LnK3HPNFK0Wveb5R21_us,5473
jax/_src/lazy_loader.py,sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw,1842
jax/_src/lib/__init__.py,sha256=5diQ2imse9hzyc_lttydXH4-2vLxcRkqEnjuMatDtPg,7414
jax/_src/lib/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lib/__pycache__/mosaic_gpu.cpython-311.pyc,,
jax/_src/lib/__pycache__/triton.cpython-311.pyc,,
jax/_src/lib/mlir/__init__.py,sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM,733
jax/_src/lib/mlir/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lib/mlir/dialects/__init__.py,sha256=AmPoWPMMC7cOKxh68U9nEoIbL7FuiDh-5jOsf6_w57o,1965
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-311.pyc,,
jax/_src/lib/mosaic_gpu.py,sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4,898
jax/_src/lib/triton.py,sha256=lMwAguUTQxYs9HkdLqeracg8sz5RL7gm1W6PpzSrzdk,2202
jax/_src/linear_util.py,sha256=BywNkv9YiT5Aj9EBDFCKKSwlGpLV53PKQ1s4NOGSdBA,18959
jax/_src/logging_config.py,sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk,4452
jax/_src/lru_cache.py,sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs,6741
jax/_src/mesh.py,sha256=T-PwGNcaVI628_R8HB5aYjzYx2qCUAUqWK-4EzkEaho,19327
jax/_src/mesh_utils.py,sha256=JJWHor6Gjp4PL3FHeAybDrQmVLuKolK7Sja2tjMWT8M,32401
jax/_src/monitoring.py,sha256=PSOb1js-21Hb4INsIdiJpm0s483xeljnoRr0Lha4nzM,6241
jax/_src/named_sharding.py,sha256=spppAwVBVRD8oAIZfe1zS_wyqQrNZ8hjZ8wcj1TpjBc,20936
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/__pycache__/__init__.cpython-311.pyc,,
jax/_src/nn/__pycache__/functions.cpython-311.pyc,,
jax/_src/nn/__pycache__/initializers.cpython-311.pyc,,
jax/_src/nn/functions.py,sha256=Pw6MrlzB3mZwWE5WzNKY2ATEs3PQPstKkLPvFDEIHQU,46816
jax/_src/nn/initializers.py,sha256=Cxy-y3C9m9ZW0OBdYpPQZFmOUx8rdFLy78dao4dhH8M,26245
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_api_metadata.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_creation.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-311.pyc,,
jax/_src/numpy/__pycache__/einsum.cpython-311.pyc,,
jax/_src/numpy/__pycache__/error.cpython-311.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-311.pyc,,
jax/_src/numpy/__pycache__/index_tricks.cpython-311.pyc,,
jax/_src/numpy/__pycache__/indexing.cpython-311.pyc,,
jax/_src/numpy/__pycache__/lax_numpy.cpython-311.pyc,,
jax/_src/numpy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/numpy/__pycache__/polynomial.cpython-311.pyc,,
jax/_src/numpy/__pycache__/reductions.cpython-311.pyc,,
jax/_src/numpy/__pycache__/scalar_types.cpython-311.pyc,,
jax/_src/numpy/__pycache__/setops.cpython-311.pyc,,
jax/_src/numpy/__pycache__/sorting.cpython-311.pyc,,
jax/_src/numpy/__pycache__/tensor_contractions.cpython-311.pyc,,
jax/_src/numpy/__pycache__/ufunc_api.cpython-311.pyc,,
jax/_src/numpy/__pycache__/ufuncs.cpython-311.pyc,,
jax/_src/numpy/__pycache__/util.cpython-311.pyc,,
jax/_src/numpy/__pycache__/vectorize.cpython-311.pyc,,
jax/_src/numpy/__pycache__/window_functions.cpython-311.pyc,,
jax/_src/numpy/array.py,sha256=y28YH4w3g4_kuRrARM4XmKn0Go50UmFjlTclT5mukLE,14802
jax/_src/numpy/array_api_metadata.py,sha256=GWwtKgbVQqOlmOsUqAqfE6A5p93ggVn4XBJNGXNwyx4,3655
jax/_src/numpy/array_creation.py,sha256=uqLG63eWmYjTQvB324ioPteBHwhkg54SABQXzoiybOw,28396
jax/_src/numpy/array_methods.py,sha256=Uu8IKkCzFnBCszxpjwxOAMOFQs4nAh545XGcHwXU-xA,47587
jax/_src/numpy/einsum.py,sha256=OIO5axi7ltf8rljLpsfg_XavPtDHC6sNpVVt1mMVW-g,23717
jax/_src/numpy/error.py,sha256=CYwkbfCTOCA-FNOIZSWiAHuUEx_9wa-Z7MW4XYv9Ss8,6625
jax/_src/numpy/fft.py,sha256=UCYHowz7dIYjMas2VfMY16ePUuZWo3JgDKq6e5yUslY,52211
jax/_src/numpy/index_tricks.py,sha256=Ug5oq2VY_XifIakhA6gibLbliXv0-1mz7SGptixN6jk,10047
jax/_src/numpy/indexing.py,sha256=EjCQwYUyNiifwqfkh5osLqBePa687AlZARye2RTKyB0,51527
jax/_src/numpy/lax_numpy.py,sha256=8NEU9-TvN8blY-oCwZJCvit45VgY8QfncGHzJfBFoq8,334446
jax/_src/numpy/linalg.py,sha256=WJrCDPlGok-2XulamhQI4q2HF1l8C9aXyvn-lU63EXw,79128
jax/_src/numpy/polynomial.py,sha256=GFJBmVeesY-fPwaz2CuNAOocODyBi6lYmCRu7jy20Jg,28429
jax/_src/numpy/reductions.py,sha256=W3mb6r3xO7woO19XenARGUU9q6ucX1486zaLJI2wbFw,109440
jax/_src/numpy/scalar_types.py,sha256=zd0YLOSUGus6TpytEUkENBMCz6g_pbuif0MYaLHqD1c,3823
jax/_src/numpy/setops.py,sha256=HIRlTvUNoH7QdmpiGrnAH_86tO1_oi4gf8QSfQeqQvA,47910
jax/_src/numpy/sorting.py,sha256=SbC_zj9z3IUUFnuXGJJ12kWSpjpl2oLpbuoFl2ndPts,16232
jax/_src/numpy/tensor_contractions.py,sha256=uGQd_mRqfQcSJIndZKAaQt--qVOdlR72aKGo6bubeAU,24311
jax/_src/numpy/ufunc_api.py,sha256=77SkEC8gDmMFpX9twGJfWAU-BsNZymy3C6ZUIqqAThw,24245
jax/_src/numpy/ufuncs.py,sha256=L4nWL2e3f7kYWautK29Q6LHRw4WGN9hLq-kbiioqTgA,118733
jax/_src/numpy/util.py,sha256=DazxM1zHR_fRTvdH3S0kUt-xKho8P8KNXzEATGG5HKM,17394
jax/_src/numpy/vectorize.py,sha256=thLB2fBHVQBas7oaVWSsaOhtLgVWRwQekq_L5F7IjAA,14077
jax/_src/numpy/window_functions.py,sha256=-AlJAgIqxJXMiXbIupk9XFC3DJljUF_BYVYIygR0k9I,5533
jax/_src/op_shardings.py,sha256=nRnGgfaKz7mIB7sHxh_QkiCa1oPMA9VsDpPyQkivtlU,3857
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/__pycache__/__init__.cpython-311.pyc,,
jax/_src/ops/__pycache__/scatter.cpython-311.pyc,,
jax/_src/ops/__pycache__/special.cpython-311.pyc,,
jax/_src/ops/scatter.py,sha256=8QUaAIS9mgB28RfAL-L_qocl6WW8I6pit1rPQJqe2ac,18181
jax/_src/ops/special.py,sha256=p1hRiOonzxVdhfcDGjVxLkL2nU0iU3EtgtmnchLIFgo,4212
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/__pycache__/cost_estimate.cpython-311.pyc,,
jax/_src/pallas/__pycache__/helpers.cpython-311.pyc,,
jax/_src/pallas/__pycache__/hlo_interpreter.cpython-311.pyc,,
jax/_src/pallas/__pycache__/pallas_call.cpython-311.pyc,,
jax/_src/pallas/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/__pycache__/utils.cpython-311.pyc,,
jax/_src/pallas/core.py,sha256=2CCu88ci6N9xytDmT_8tERaH6uAaE5Zo4e5LVMINOiI,48329
jax/_src/pallas/cost_estimate.py,sha256=inTDMJ3N7qBsRFf7MIBnkgK72wN5qyaTDp5Qg5jt7WE,8316
jax/_src/pallas/fuser/__init__.py,sha256=cHEWioEa-43e-714tP3fFN9hdyZg_ebBFbFJur7bXwg,1182
jax/_src/pallas/fuser/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/block_spec.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/custom_evaluate.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/fuser_utils.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/fusible.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/fusible_dtype.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/fusion.cpython-311.pyc,,
jax/_src/pallas/fuser/__pycache__/jaxpr_fusion.cpython-311.pyc,,
jax/_src/pallas/fuser/block_spec.py,sha256=mbOyMpm3r61LFNSxS8BDMDmhlg1lVk4yYzQE-TIyuSQ,67290
jax/_src/pallas/fuser/custom_evaluate.py,sha256=jv5JjL5BlUUK89ZKx9sufff0bVyqJA3s5ag6Bpus5ys,2823
jax/_src/pallas/fuser/fuser_utils.py,sha256=UY_9Ly9Z863mHX3R_EdLw4H1jLsTCeaeTDMV1Dg9l-s,1273
jax/_src/pallas/fuser/fusible.py,sha256=afsksW8n7JyKqC3Ggolu4GLWYU1ontw-czS0xkwPmjU,2692
jax/_src/pallas/fuser/fusible_dtype.py,sha256=dDhqowkKZ7uePKSSsCq3toBctGXC8bBRAfm5yMO0MGM,15648
jax/_src/pallas/fuser/fusion.py,sha256=W1NBIzSqOkBk8ADty9HfjnmePZXKBsPxFUiM_k3qtC8,1488
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=02209zSpwEsPCrR8lsj1EXkleKG5ylZFmcKhHLclHsU,10808
jax/_src/pallas/helpers.py,sha256=UpIfs-yNk4IrTXNVzmGlSOOXQriEzGwNAlCLofX7CMg,2806
jax/_src/pallas/hlo_interpreter.py,sha256=zeOxcMsbGApyMHQcB4ktTdNlvU4rQIMf6qzbh9HR_mE,19049
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/error_handling.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/helpers.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/interpret.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/pipeline.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/random.cpython-311.pyc,,
jax/_src/pallas/mosaic/__pycache__/verification.cpython-311.pyc,,
jax/_src/pallas/mosaic/core.py,sha256=pD2hETgHNjp_tr4wJa5eT7hY50Rp4XKnN5Hx8mqzwlA,9908
jax/_src/pallas/mosaic/error_handling.py,sha256=FCbB-b5Dl75erMmr-FMvc4QJwCXqn2z7ioNC2Eh0vQc,5542
jax/_src/pallas/mosaic/helpers.py,sha256=DFBL7RuFtfBQc-aUaC4bILa5Xzsbb91Wu3VCn7b3v4M,2767
jax/_src/pallas/mosaic/interpret.py,sha256=EhtWDOp8hSSzO130OQKcG9GgYbUBGcLg5IaipJOY6CU,89639
jax/_src/pallas/mosaic/lowering.py,sha256=TjUpBJjBFPZhCUpXRGSdasnrMm2A67QXM7a9Uv3zHkY,137751
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=N46ZF8Wv2aJleLyDWfBf88rDWSyD8HmtNlaDQ1yo_iE,10372
jax/_src/pallas/mosaic/pipeline.py,sha256=QM23p3Bi3mOMe47mTISP0oVor3LfNOySImFlW0O-8do,58211
jax/_src/pallas/mosaic/primitives.py,sha256=C5YGYasXoiBa-YRZGRlPN5q6ykZQT3w4YpSi2uezRP4,27566
jax/_src/pallas/mosaic/random.py,sha256=l402QtsGJkvPQ9UY_o9XNWsL58_YXVDBxaGxNhFTLWo,7845
jax/_src/pallas/mosaic/verification.py,sha256=1QWYOwAAOvuCgB8gXWCJUlP1mPLvegzeUoFgJ_xC4X4,24485
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/helpers.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/lowering.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/pallas_call_registration.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/pipeline.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/mosaic_gpu/core.py,sha256=TrrfnxJBJaDBqxXTzoyuvyn9v3o1tvpTf5MQtiizJRM,36464
jax/_src/pallas/mosaic_gpu/helpers.py,sha256=DOkDJ7shcKrNau9KgJ8Kt_ceUGE4lkFCEEpqaZjUE-k,2854
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=Dah7uTWIUk0ik1XKnSbrNPar78vxqSi63dr5nyOXU8c,116786
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=THgSdULeaycMfFwzVPu6IoMoC8TFIvEIIWAL55DmJlc,5589
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=xYlDhR0S2ZE0CpHRDwG11coljWacwYssiSGsE3-AdAE,29039
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=pw7pqjT9RSYP61a3v9YdBPVjyXjCDNQ2--p63M-jTYU,71008
jax/_src/pallas/pallas_call.py,sha256=msCZDel3ss4Slrj3EhwZmpNQI5fG_L9DurOBn5nd284,69253
jax/_src/pallas/primitives.py,sha256=sS4qmP9oV5WIxA0GCjnpyjJeNGFPZ7YZMt4GIG2hV_Q,43036
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/__pycache__/__init__.cpython-311.pyc,,
jax/_src/pallas/triton/__pycache__/core.cpython-311.pyc,,
jax/_src/pallas/triton/__pycache__/lowering.cpython-311.pyc,,
jax/_src/pallas/triton/__pycache__/pallas_call_registration.cpython-311.pyc,,
jax/_src/pallas/triton/__pycache__/primitives.cpython-311.pyc,,
jax/_src/pallas/triton/core.py,sha256=pd0AQ5oEU2oljvCJfkJmUuyjEBtxEn6Ml6dtChdCTE4,1383
jax/_src/pallas/triton/lowering.py,sha256=jLBLVEQ-mHGT4C1lNHDFYFbq87xoMxR4JtSRkMzuNCA,99180
jax/_src/pallas/triton/pallas_call_registration.py,sha256=5D-I2Y4c-xr4AcFNU7z2NwhpVffrwMZxBzro7STZQng,6699
jax/_src/pallas/triton/primitives.py,sha256=h_nAb5_WTDgtna0a5VDKql4hDx8I2gt9ryuY5LsQo9A,4235
jax/_src/pallas/utils.py,sha256=LvCcGVDxrA08HbQ3cngF8D2hZKq2lz7-_Opye8wSccA,13889
jax/_src/partition_spec.py,sha256=zvr3RWjMJDxpq_l9APIxwyeLPJU2zmtGtpzTdR0NQUg,8618
jax/_src/path.py,sha256=VkrwcBMWagZZQBBQYhdSFr7pQzoxwBnNUEfKfEbN-nU,1596
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=4AygjlqWJSbkFOolz1P1h7ThT_8wXSz3auPOnxJ1j4M,147338
jax/_src/pretty_printer.py,sha256=1tSAtbJj6N9a0ZjwqphDD-kPPolsYxRp8JM_WYkrUXI,17578
jax/_src/prng.py,sha256=jjeeUE463LUWFfviEXqIrdauTTBzY8e_xvqf9NHrqpg,46495
jax/_src/profiler.py,sha256=zPgtTULdZBFGT629usNmQq82mtAfRjd77Hssq2Xsf3Q,17282
jax/_src/public_test_util.py,sha256=wXB1H7lgFuDvZVRyDOs-yPMm9C18VKVPKUfKlIdF7gI,12676
jax/_src/random.py,sha256=lWxw3thgcBMIiZM-dgEBUEWQ9wfl7LBGYPJPti015q0,108171
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/__pycache__/fft.cpython-311.pyc,,
jax/_src/scipy/__pycache__/integrate.cpython-311.pyc,,
jax/_src/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/scipy/__pycache__/ndimage.cpython-311.pyc,,
jax/_src/scipy/__pycache__/signal.cpython-311.pyc,,
jax/_src/scipy/__pycache__/special.cpython-311.pyc,,
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/cluster/__pycache__/vq.cpython-311.pyc,,
jax/_src/scipy/cluster/vq.py,sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc,3043
jax/_src/scipy/fft.py,sha256=NFy_bDIzx44Lpkw1-74fBm2jxeXmzIUp8LTVh3-6Z5s,15281
jax/_src/scipy/integrate.py,sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0,2346
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/interpolate/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/linalg.py,sha256=jLR8xl1wfluw0vr0rEdA1gCGfpKhYuFRsmD7cbGaX3k,79500
jax/_src/scipy/ndimage.py,sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc,7092
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/line_search.cpython-311.pyc,,
jax/_src/scipy/optimize/__pycache__/minimize.cpython-311.pyc,,
jax/_src/scipy/optimize/_lbfgs.py,sha256=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw,7769
jax/_src/scipy/optimize/bfgs.py,sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE,5609
jax/_src/scipy/optimize/line_search.py,sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc,13235
jax/_src/scipy/optimize/minimize.py,sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs,4979
jax/_src/scipy/signal.py,sha256=mwzup6IeV3hlG5npG2Q92imD-v74w9AtENhvxpcABQw,47396
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/sparse/__pycache__/linalg.cpython-311.pyc,,
jax/_src/scipy/sparse/linalg.py,sha256=IGX8lZ2il77S3WA50Lz7MdBiw7lqdD8mHgWRaCUHX6k,27245
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/spatial/__pycache__/transform.cpython-311.pyc,,
jax/_src/scipy/spatial/transform.py,sha256=yD9HOfgQpQO3-eACTh5zW8qzV-WLcrTHY_5ZlsnBoSk,17525
jax/_src/scipy/special.py,sha256=tR7nHwwEUHKCWjEXOwRoTXvWdnLaNTJKeyrkK3jzDo0,90272
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/__pycache__/__init__.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/_core.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/beta.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/betabinom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/binom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/cauchy.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/chi2.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/expon.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/gamma.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/gennorm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/geom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/kde.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/laplace.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/logistic.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/multinomial.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/nbinom.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/norm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/pareto.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/poisson.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/t.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/uniform.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/vonmises.cpython-311.pyc,,
jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-311.pyc,,
jax/_src/scipy/stats/_core.py,sha256=khdXG4LFVlXGQ2derIQIfL7TN1SXfdNZ59uBq3gtBn0,10652
jax/_src/scipy/stats/bernoulli.py,sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo,4541
jax/_src/scipy/stats/beta.py,sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME,7988
jax/_src/scipy/stats/betabinom.py,sha256=dmhOAUk6py5Fnu-jR95KnwJ6suvvfwNF9CXNHz73CSc,3447
jax/_src/scipy/stats/binom.py,sha256=E7a3go8bCM5bSiC72l1pDQEuuzSpXLmgfG40iPF90wA,2825
jax/_src/scipy/stats/cauchy.py,sha256=26eHaNIfxwh1og3kTqnPYgivH0GTIuWVwElai4VJwv8,8884
jax/_src/scipy/stats/chi2.py,sha256=vPFrhpbLRPDheYenmHQ7QvZrRt8cAyRtOpU9VmemNIY,8008
jax/_src/scipy/stats/dirichlet.py,sha256=_wWyx07z62_EH25i3J4jS5q53oj0UH5A3-XgQzapwbY,3198
jax/_src/scipy/stats/expon.py,sha256=W0tt3HtC5jB5mEM-bQbAC4U9FKVJFspsqnPpOAN6_FE,7813
jax/_src/scipy/stats/gamma.py,sha256=Nf0qgFD7K-2H4R581g1Kb0RQ_WHZrl81A1vgyS_PdxM,7038
jax/_src/scipy/stats/gennorm.py,sha256=kp22L5H-rR5n7sn9rpeWsWta-F76yZaXkO0Ovvf9zc8,2989
jax/_src/scipy/stats/geom.py,sha256=2hQ4LoaqNHFBahdRTGS9oyWMYdFQufJb2ByqOw9709s,2243
jax/_src/scipy/stats/kde.py,sha256=qY2ZLH8HVOU50DhQBDrhH2XTLT1eUHAFx7qGuvhgSmc,10139
jax/_src/scipy/stats/laplace.py,sha256=9PlK08r9wRHelqSIuKkv8KGO0oPV2AArI0Fs6rICFj0,3269
jax/_src/scipy/stats/logistic.py,sha256=HB3Brjwwl-yyGB_9K1lpdiQlmitsHwNLfgl0g4owFSM,6271
jax/_src/scipy/stats/multinomial.py,sha256=CVgpOPCVcr_wTNuF1vXcUjgsENe2mVO7gYBYTSdCOwI,2469
jax/_src/scipy/stats/multivariate_normal.py,sha256=Dei7nhj3j5gB1lBi9q_ZpdMHi1UY7XCr3GTzNuaWF1E,3433
jax/_src/scipy/stats/nbinom.py,sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw,2603
jax/_src/scipy/stats/norm.py,sha256=pUS-JhrjrFtdbfO178IeOJizZBU_WjXU6aUr8W67tQg,8401
jax/_src/scipy/stats/pareto.py,sha256=WcRAchdj198RF_W3egR8Bteok1dXE0PHv9ygw-pQYtM,2596
jax/_src/scipy/stats/poisson.py,sha256=Y43e2hY6L89go0Vc1HJCDrf61-vDZcQVIC6zVJdaYyQ,3348
jax/_src/scipy/stats/t.py,sha256=I32sQ9TVaes4keK5AIK6I-vX--nvl4qtZctuc0zIhG0,3221
jax/_src/scipy/stats/truncnorm.py,sha256=V4_Y2OEMYPepPyWVqMZKO0hRV8HwwnI6_djK9xYwdDA,9239
jax/_src/scipy/stats/uniform.py,sha256=pR3NYr6ZHTs6xhw-LECsRJqp5l_iBbsw689QHxr9Puc,4361
jax/_src/scipy/stats/vonmises.py,sha256=yxvmgBI_HWQn8U3u7wtLv9PjqOk7hRX8I4gWKfqRZKg,2451
jax/_src/scipy/stats/wrapcauchy.py,sha256=SpHIjzS9j3TqgAMTHXxtDz73aBfezqYUaNSsGYCky5I,2371
jax/_src/shard_alike.py,sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg,3989
jax/_src/shard_map.py,sha256=XEN3qJm1jiEoKU6MEVzISJuDsZyenfIgVIeePV1dd5g,83143
jax/_src/sharding.py,sha256=pkrmy_r6huTjqDmNsQaFmWIFdxZukjp5NyeWhMbC-OE,7704
jax/_src/sharding_impls.py,sha256=cz5OCPAFt9xdpg2KknC1cgeHehaIr2NosluMTb0mSSQ,54459
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=O8Ff0QonDyYKecTH8a2KdqBe7HuQHrIDTkb-OShHnpk,10852
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=ZhJCGlgFqiYyIvYEiCLojdrwanNULwhorSAb_6N7SJA,31569
jax/_src/state/__init__.py,sha256=TU0qQETOXKzVAd1oVaIEtOJTC2OMj1PdFhsExMHFx2g,988
jax/_src/state/__pycache__/__init__.cpython-311.pyc,,
jax/_src/state/__pycache__/discharge.cpython-311.pyc,,
jax/_src/state/__pycache__/indexing.cpython-311.pyc,,
jax/_src/state/__pycache__/primitives.cpython-311.pyc,,
jax/_src/state/__pycache__/types.cpython-311.pyc,,
jax/_src/state/__pycache__/utils.cpython-311.pyc,,
jax/_src/state/discharge.py,sha256=6tvlsF0-yXrdFuIKj7VgzWIkdbsvGsuM9TpsQoo25rA,50958
jax/_src/state/indexing.py,sha256=_ZVN84Jo5Z39SVAJR8WJvXlnDMnGKKMNncdUc1azI9A,12013
jax/_src/state/primitives.py,sha256=IVxrV6bJcEFfyzo3qGbaeHlZJER70hoG539OvqWzd7c,30799
jax/_src/state/types.py,sha256=BePu2rY2_iqistfF42FcxQCYqmyFm1NMDqfFt4fKZk4,14923
jax/_src/state/utils.py,sha256=BBCCdjdCyaZmnB1Y6FSfSZ9JyWiAbTX8c1ZaXfR-lt4,4107
jax/_src/test_loader.py,sha256=l7LDHdnqD5g44nUo4Hdwtc8IplXjCcn-L9-RO4Q_qhs,7260
jax/_src/test_multiprocess.py,sha256=ZOXfV-PAbNdhyJpbaMywexNTfM1pnm_xpvLbRo3ub8U,8602
jax/_src/test_util.py,sha256=xUMWeuXxjtnTGIBa10HHMzU4McYKjl613I-gAjcUf-8,81442
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-311.pyc,,
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/betaln.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-311.pyc,,
jax/_src/third_party/scipy/__pycache__/special.cpython-311.pyc,,
jax/_src/third_party/scipy/betaln.py,sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=5C9-SwIUrwlpBUgAKYskaSLgTbOtpwESJRPv-6cjj0Y,6307
jax/_src/third_party/scipy/linalg.py,sha256=SzmpCoi4kSnLTflDsRa5QdoBcijKHuwmtN9xac2TWlw,3786
jax/_src/third_party/scipy/signal_helper.py,sha256=3IF3mEJ09uSuPeDVM1ntW5dYSoOi-LQDZJkXr38Rv-Q,3508
jax/_src/third_party/scipy/special.py,sha256=weGxx01LOfJDklgdX8vKJVUpTy1_FV_9PWEX2hgNtB0,9136
jax/_src/tpu/__init__.py,sha256=4S3pYNuiw6yzPmNNm2iRi40_Jf9xlHj-oIRb9Gp4sxk,581
jax/_src/tpu/__pycache__/__init__.cpython-311.pyc,,
jax/_src/tpu/linalg/__init__.py,sha256=RrUPd-WnsNpraGGzWYI-IYTpNcbtQeGqaotRh0Bwwuw,779
jax/_src/tpu/linalg/__pycache__/__init__.cpython-311.pyc,,
jax/_src/tpu/linalg/__pycache__/eigh.cpython-311.pyc,,
jax/_src/tpu/linalg/__pycache__/qdwh.cpython-311.pyc,,
jax/_src/tpu/linalg/__pycache__/stack.cpython-311.pyc,,
jax/_src/tpu/linalg/__pycache__/svd.cpython-311.pyc,,
jax/_src/tpu/linalg/eigh.py,sha256=0MfHzHgLzvEshFYTdO3p_J-lj9zNyI8EcGGLJ4mCCog,24588
jax/_src/tpu/linalg/qdwh.py,sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo,9391
jax/_src/tpu/linalg/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/tpu/linalg/svd.py,sha256=jg30dXixuXfvLX21YCl21rl2lwwxS2OE2-QQp0KraXA,10158
jax/_src/tpu_custom_call.py,sha256=0KcEmuUf0rohGQfFRSUsdwC8fYzbJ-C93FpoXSXCmiQ,26097
jax/_src/traceback_util.py,sha256=QZinVxddgYKdO3p8OFo2JITpZMJO-wttlxe4L7GcTqM,9141
jax/_src/tree.py,sha256=UAJzeQwkIRHqVwtQlM3OSfn9lvb1pdnDxnfA6HCYOSo,13975
jax/_src/tree_util.py,sha256=03qiPML25BoA4zEcg3Ew4DAlXATcS9UCihXntnw2Xew,47308
jax/_src/typing.py,sha256=Ed5ujNHsKktTk0Z2nhZojS_77K1INnsPvj4S7o0KQ5Q,3529
jax/_src/util.py,sha256=E9YnFhRSRbuaVQSN4e5_3VckAfaV4K6hWEF0qqrtCak,20945
jax/_src/xla_bridge.py,sha256=VeZ-NCvtITNFrxpuu3IByAhUaZqAeotMsbipmdaOxpo,41461
jax/_src/xla_metadata.py,sha256=ZQUQ4c8Sn11_3LW1WcOl8umxV3NerwhE5NLYKaUUh9I,2103
jax/ad_checkpoint.py,sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U,932
jax/api_util.py,sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc,979
jax/cloud_tpu_init.py,sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs,651
jax/collect_profile.py,sha256=vLzsQEhtxsHlLbSocpoY4-9IIza8gEZ22NhLUo1Es6Q,4511
jax/core.py,sha256=ArEQGXJE2KwLLZKXaLAkDM13beCiDP4XdiNAxCKHhxk,8619
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=CVhRiRgnWk5hBsc3ZX7mFD0Yst2S9VDAirtwF8rSZcY,1967
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE,710
jax/dlpack.py,sha256=UBWbObzJ-N-nFAPvoSTamtXH2uaqSoG2-NnUDYuZuEo,1411
jax/dtypes.py,sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I,1188
jax/errors.py,sha256=MaZbeSxeRyWRMfq4cj6P0GA9UFykLgX3aURPAKhqbts,1363
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/__pycache__/__init__.cpython-311.pyc,,
jax/example_libraries/__pycache__/optimizers.cpython-311.pyc,,
jax/example_libraries/__pycache__/stax.cpython-311.pyc,,
jax/example_libraries/optimizers.py,sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU,20426
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=yoE5_c5xRagzLRgNBgo2jRWg31h8Sm-dpKW58vhgjh0,1198
jax/experimental/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/__pycache__/attrs.cpython-311.pyc,,
jax/experimental/__pycache__/buffer_callback.cpython-311.pyc,,
jax/experimental/__pycache__/checkify.cpython-311.pyc,,
jax/experimental/__pycache__/compute_on.cpython-311.pyc,,
jax/experimental/__pycache__/custom_dce.cpython-311.pyc,,
jax/experimental/__pycache__/custom_partitioning.cpython-311.pyc,,
jax/experimental/__pycache__/host_callback.cpython-311.pyc,,
jax/experimental/__pycache__/jet.cpython-311.pyc,,
jax/experimental/__pycache__/layout.cpython-311.pyc,,
jax/experimental/__pycache__/mesh_utils.cpython-311.pyc,,
jax/experimental/__pycache__/multihost_utils.cpython-311.pyc,,
jax/experimental/__pycache__/ode.cpython-311.pyc,,
jax/experimental/__pycache__/pjit.cpython-311.pyc,,
jax/experimental/__pycache__/profiler.cpython-311.pyc,,
jax/experimental/__pycache__/rnn.cpython-311.pyc,,
jax/experimental/__pycache__/serialize_executable.cpython-311.pyc,,
jax/experimental/__pycache__/shard.cpython-311.pyc,,
jax/experimental/__pycache__/shard_alike.cpython-311.pyc,,
jax/experimental/__pycache__/shard_map.cpython-311.pyc,,
jax/experimental/__pycache__/topologies.cpython-311.pyc,,
jax/experimental/__pycache__/transfer.cpython-311.pyc,,
jax/experimental/__pycache__/x64_context.cpython-311.pyc,,
jax/experimental/__pycache__/xla_metadata.cpython-311.pyc,,
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/pytree_serialization.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/pytree_serialization_utils.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/serialization.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/tensorstore_impl.cpython-311.pyc,,
jax/experimental/array_serialization/pytree_serialization.py,sha256=L5zgqU3-STe4LsDJywjOCwoketfhMHn7FMdSMz6enl4,21220
jax/experimental/array_serialization/pytree_serialization_utils.py,sha256=rBZP00cuR2uWhIgdralpm4HgVrFjJ-nTMeNU2zqP7zA,2776
jax/experimental/array_serialization/serialization.py,sha256=3wiRVOMdD3odJbxFPwK3gj1ttUhkrWWTEutDF_XXPBI,13434
jax/experimental/array_serialization/tensorstore_impl.py,sha256=sqUObJJ-40GIJ_1CXDtSfe2GQK-T5ZuseAHO68oQvX0,24666
jax/experimental/attrs.py,sha256=FRU-Wm4BsARc07DAZ_GyL765Rhuusz4OLtKHmwRPDL0,749
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=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
jax/experimental/colocated_python/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/api.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/func.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/func_backend.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/obj.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/obj_backend.cpython-311.pyc,,
jax/experimental/colocated_python/__pycache__/serialization.cpython-311.pyc,,
jax/experimental/colocated_python/api.py,sha256=wVcPWNFmwKL-NdmtzrK2vzhCndf90VmME-gkqnbnm_A,4455
jax/experimental/colocated_python/func.py,sha256=rUwSmbLuA_lVE3mB-KI-kWU0t1zuNXHzgcu6TI0cJ_A,17353
jax/experimental/colocated_python/func_backend.py,sha256=lXN-FRbWIWezsICyIc3QYwjpF9-tYgZV1Ut7TBN_dpE,1373
jax/experimental/colocated_python/obj.py,sha256=N_KEQff3d9vzuX5EGkM4Z8rPn6lGEDPNFrTIqcesQ70,5636
jax/experimental/colocated_python/obj_backend.py,sha256=50I4Ttb17qRk7IUrtufYMNlXqyl4KY6bBc61lGQBWfM,2283
jax/experimental/colocated_python/serialization.py,sha256=DObZGJ-in5JHA4YKsBpshXhAcbtN3iLZzSlxQ-kgzNE,8987
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-311.pyc,,
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_dce.py,sha256=ORRk_kBjbseQ06T_LmLPxYCO7-Rby5U8Erro5U5Gi4A,682
jax/experimental/custom_partitioning.py,sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g,1054
jax/experimental/host_callback.py,sha256=Msk8gUwv_UtJckKwsGU0lYTYGpPG2isWCJrxJ8mzb6g,1201
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/call_tf.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-311.pyc,,
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-311.pyc,,
jax/experimental/jax2tf/call_tf.py,sha256=pE4kpi_9JhPO64j3HwVVSIDp1HJLtzKlDdTEb43xV8w,28939
jax/experimental/jax2tf/impl_no_xla.py,sha256=zi-U4UaZ7xVJ_Y9gmhHn8evKRX4_6SKJM33INEDucLY,54479
jax/experimental/jax2tf/jax2tf.py,sha256=AXQlfMiqCRu8c2Jm163q3qVl9LOVBoecZSwWzUILfoY,145513
jax/experimental/jet.py,sha256=qCQiv4k2XoRKOoLWgRGZYgwskBAWuTPqMSGx8zKhgXI,27322
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/key_reuse/__pycache__/_core.cpython-311.pyc,,
jax/experimental/key_reuse/_core.py,sha256=10LBbkl-_vUqYSEI7a2s-su86fIS7ORnA9c8foagr2o,23103
jax/experimental/layout.py,sha256=ICtPXKjsJcDSmfF7YD8Gr3PSIAIJQJAj57hb76Ik71o,777
jax/experimental/mesh_utils.py,sha256=Sf5ivjPuZ_u0ufQ38MLmhbEFnlIMs4SHd6lGXWQJgQY,923
jax/experimental/mosaic/__init__.py,sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY,883
jax/experimental/mosaic/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/mosaic/__pycache__/dialects.cpython-311.pyc,,
jax/experimental/mosaic/dialects.py,sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs,769
jax/experimental/mosaic/gpu/__init__.py,sha256=YP7xRHpKa_ueEtHN6I3VjBeyW7kwfRc0NsuRDVNDCn0,3467
jax/experimental/mosaic/gpu/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/core.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/dialect_lowering.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/fragmented_array.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/inference_utils.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/launch_context.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/layout_inference.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/layouts.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/mma_utils.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/profiler.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/tcgen05.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/transform_inference.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/utils.cpython-311.pyc,,
jax/experimental/mosaic/gpu/__pycache__/wgmma.cpython-311.pyc,,
jax/experimental/mosaic/gpu/core.py,sha256=RPtRuxKPfvbF0iekpCT-H6yPV64NnMjHQ402lfrjC8s,33119
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=EqtT14rIW0aWJ6cIupGBBKuW1hLLEMVPztFiLNQ8MoU,62202
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=j-lPCaRMl4DML5KDG9uo9lTdXU6MKr7vf7k5wDce71g,114293
jax/experimental/mosaic/gpu/inference_utils.py,sha256=a5qX6vpSp8FVLju7QCFypo_Z7Ny5cxnX5MrMW0lKZe0,7716
jax/experimental/mosaic/gpu/launch_context.py,sha256=lDpM165JZtX4p_VbBXFOtHvegxHbnmU-4nhsWreE1KE,37873
jax/experimental/mosaic/gpu/layout_inference.py,sha256=jLktgaPC0gjCT2O74qKSWPnarN_V5ggRDXnW3W6GzDY,27589
jax/experimental/mosaic/gpu/layouts.py,sha256=FAojzJO91tQ_CP_f_WQmmTiozQOMJUKxb_ND7Vb-PCA,6820
jax/experimental/mosaic/gpu/mma_utils.py,sha256=_Aoyrzdo3FWifCBUsSrlKMGyJ5NfxhLL01o3jlF1WGc,8664
jax/experimental/mosaic/gpu/profiler.py,sha256=Y5I7eaONisW4DmRGOIlD-nZP6ebtPGmzDOq_SqYnRvw,15532
jax/experimental/mosaic/gpu/tcgen05.py,sha256=I5x_mRVuOU8bTDR3_2FGYdadQJjfyyPBkb8lzrhWqGE,38907
jax/experimental/mosaic/gpu/transform_inference.py,sha256=4U61EAMidFdRHuQoCR8exPZzZElOiMjsd4Rz7mwQiiE,17677
jax/experimental/mosaic/gpu/utils.py,sha256=CHVvgAO23sDE32g88aZvffI39uFwpgsaGpTSv2kRMKA,53470
jax/experimental/mosaic/gpu/wgmma.py,sha256=9tFPcdZJDfWumGHVHI3hZaBcsE1bH37MuDDyL8lFRG8,16961
jax/experimental/multihost_utils.py,sha256=oCbJ2BfjfccFPM82eZbC58sMgw_W9zIhPYBpSG7KYdU,21886
jax/experimental/ode.py,sha256=VmaWgwXaD1N9WspkCJ7pvdbUtYRuGtp4cpQaqJkf__8,10859
jax/experimental/pallas/__init__.py,sha256=4dP1KYWvPGObgLxtP89scLukBD7DUMsiiGPnhVDUTTA,4303
jax/experimental/pallas/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/mosaic_gpu.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/tpu.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/triton.cpython-311.pyc,,
jax/experimental/pallas/mosaic_gpu.py,sha256=Gr37TOTqXGhBc4zurUsWfSDVGVtbTGGQAKvJZcnP3m4,4402
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/attention.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/attention_mgpu.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/blackwell_matmul_mgpu.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/collective_matmul_mgpu.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/decode_attention.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/layer_norm.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/paged_attention.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/ragged_dot_mgpu.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/rms_norm.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/softmax.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/attention.py,sha256=9jJNoBTnzr_yPM5N5Q4DV-J9EYJtqilxG4vjgHH2r7s,23222
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=2p2WCtiDrMeHN7cn80ui1kdQ2tGVj-cGSRJTxCJgVvc,36296
jax/experimental/pallas/ops/gpu/blackwell_matmul_mgpu.py,sha256=62e5y4tysd0jOd4UJICqIyEiqUyJ6ENMBaZEHkYi6v4,8453
jax/experimental/pallas/ops/gpu/collective_matmul_mgpu.py,sha256=AKorrKmQexqSEdNSUKyT1Elu21CVed9QO1Qv7NXh3X4,7646
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=1BvVgUAtMZni4kM_pBTjYMl7Lns8Axg4u5-jc7sUen4,17080
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=3LL6x4gxSVlMhob33M1-0XUrLaYEo3E4sZe0neY0vE4,11127
jax/experimental/pallas/ops/gpu/paged_attention.py,sha256=yviTcP15OFIWOu4UtYgT7q9r3BbsuhVbjOJIm546ChQ,16221
jax/experimental/pallas/ops/gpu/ragged_dot_mgpu.py,sha256=g1KB4Y4aRVuCMd4ZG89ZwWYx_hDmxCNLDQNVJx99V1Q,11406
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=LdiTCNLeuTYPu15TFM1VmJypLBeqLitTc-ITa_PwbzY,10027
jax/experimental/pallas/ops/gpu/softmax.py,sha256=sKeVjcAwN4va31DeFhMO_-ZLkLkGbgOtXucbG7G2GUE,2768
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/example_kernel.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/matmul.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=9XXcBpb_pXZGn8EOa-daiO_EJecOLdBtx9QCxCowEL8,5573
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=faIwM9av0WKRRW3Q1yOmhg2_DDvtMKtsW9LztFIiNPM,49874
jax/experimental/pallas/ops/tpu/matmul.py,sha256=x-2MCTChhr6BwQMQXzKcRT8PDVQ0ZvPjGxYOCaAgb7g,2717
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w,650
jax/experimental/pallas/ops/tpu/megablox/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/common.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/gmm.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/megablox/__pycache__/ops.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=PnH6tp1yv31d5QPtedi7oI7sBgqu0iQ2Pm2lJ4EzqTA,27243
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY,700
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/paged_attention_kernel.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/quantization_utils.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/__pycache__/util.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=MRVFNdCdETdHNS1SAAXWV_fufPJHpMsMfSQhiEHKxFQ,22026
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
jax/experimental/pallas/ops/tpu/paged_attention/util.py,sha256=t1_5K06BjSyQKaa-zqQnJDpW4dX1iQ-URfb_hEUJ53g,3254
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__init__.py,sha256=XuQq_I5Pn_5EDon__0hWN9NOrOJ3grU3y0qqc8qKSsg,1055
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/kernel.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/ragged_paged_attention/__pycache__/tuned_block_sizes.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/ragged_paged_attention/kernel.py,sha256=t4sYYuwb6-P6TZ3PZsgDYUXrmjC23iVVrmgXHJE4-WA,31994
jax/experimental/pallas/ops/tpu/ragged_paged_attention/tuned_block_sizes.py,sha256=_QZBjMU41ln_i9Fo6QSHjclzxCd0f4xAdiTTy-39iXU,25959
jax/experimental/pallas/ops/tpu/random/__init__.py,sha256=gEMCMS8LKTvoFm0t7khiAnQg78EDUuMI-fvTvw250eo,682
jax/experimental/pallas/ops/tpu/random/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/philox.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/prng_utils.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/random/__pycache__/threefry.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/random/philox.py,sha256=ySUyKx2lromSM1Wez6nRYEuzy2srOzAGb8KbiMoeuUo,7485
jax/experimental/pallas/ops/tpu/random/prng_utils.py,sha256=CH1W4SuTJtdKNVA-7G64qf_RwtJM7zteTuQsfsRieMk,1771
jax/experimental/pallas/ops/tpu/random/threefry.py,sha256=0HyU9Y_PUpAe_4ypcaqmGgDHeZRmEwsPVOnFhNho18I,4330
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs,2738
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_kernel.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask_info.cpython-311.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=2par2bNjSjBCTHl8LExI1T1yuYt9VAYu74nbfcBYKfY,75020
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=ncsRlbb-oHneOkZqkSpOB6r7V0p_lZY60zBEsar364c,16307
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=UTG7D_TWynzs4oQS82DQE9ydLrQ-z372PvrfMZ2h5cQ,40286
jax/experimental/pallas/tpu.py,sha256=dQZUek8Z-FyMXteRONkj-KvGW-k-SAQz3IlfBk0FsN8,5157
jax/experimental/pallas/triton.py,sha256=HWFJ51-oRjRLY5fiMmlX2DwvE2hT6--7ghBzIMv1Fgo,1459
jax/experimental/pjit.py,sha256=2Qi4A4tKn1p6pjxysiCY5UAsfQq_KVZ7QpMX3gRJw30,749
jax/experimental/profiler.py,sha256=Rxlgbb4TPJsfeWXwLYG58gW0PCqBjDVWswlCCLr7fBg,1413
jax/experimental/rnn.py,sha256=q__NeQnA8uZspzC5zFU5a-593i3VOEAbrRmsiJwYEfU,20203
jax/experimental/roofline/__init__.py,sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw,1260
jax/experimental/roofline/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/roofline/__pycache__/roofline.cpython-311.pyc,,
jax/experimental/roofline/__pycache__/rooflines.cpython-311.pyc,,
jax/experimental/roofline/roofline.py,sha256=cMzOBEADpaY4jDtivkAGDCTIgfsdPvm4hdQYkKMGjfY,11167
jax/experimental/roofline/rooflines.py,sha256=aTQzp55zF-mpfB682LsrpbN0a88Y-35NIRKzNNcv8eI,20404
jax/experimental/serialize_executable.py,sha256=hiR59Fn_4PhECvMXftzwT5lHKx2_xSS9__OAl_RQQu4,4228
jax/experimental/shard.py,sha256=xTXar7eZIDvVAXyXUdgCPaqHEUmbl8zPh6iF5oLt380,696
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=zkBPC3rKW787yjzwqz0vjc_2GHpRI18CuK_ocmo5rlg,4427
jax/experimental/source_mapper/__init__.py,sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q,1698
jax/experimental/source_mapper/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/source_mapper/__pycache__/common.cpython-311.pyc,,
jax/experimental/source_mapper/__pycache__/generate_map.cpython-311.pyc,,
jax/experimental/source_mapper/__pycache__/hlo.cpython-311.pyc,,
jax/experimental/source_mapper/__pycache__/jaxpr.cpython-311.pyc,,
jax/experimental/source_mapper/__pycache__/mlir.cpython-311.pyc,,
jax/experimental/source_mapper/common.py,sha256=2wh5nZMvCdhjDBQKw539R5WTnmFJWNzMBRgaZxqI__s,2481
jax/experimental/source_mapper/generate_map.py,sha256=rCr8YPEZI1OAjw_GZFUndUkhgVVS7gcv6dDVszTey3w,2102
jax/experimental/source_mapper/hlo.py,sha256=ItS2ytK1D4cBQOZsap0UzyWA-saoI_T_fWH2DVoMSFA,4112
jax/experimental/source_mapper/jaxpr.py,sha256=PXa9zMzP2_fD_0vxo8Za-mw21pXp2Aao6IQkQIqmoJU,2595
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
jax/experimental/sparse/__init__.py,sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs,11186
jax/experimental/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/_base.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/_lowerings.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/ad.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/api.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/bcoo.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/bcsr.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/coo.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/csr.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/linalg.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/nm.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/random.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/test_util.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/transform.cpython-311.pyc,,
jax/experimental/sparse/__pycache__/util.cpython-311.pyc,,
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=nNviA0Ed0Dq-J_mRF1us4DQ4GUP69Gu3WBFzuSbLlWI,13334
jax/experimental/sparse/ad.py,sha256=zJLXiDufoAVcLTHdFitEXdbQUM_XtDiJoIktwDVuNZU,7789
jax/experimental/sparse/api.py,sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g,6596
jax/experimental/sparse/bcoo.py,sha256=D2n0P_IQOqOhyF0f2jWIpZeDA9EgFpK3G-V7YICHm-g,129450
jax/experimental/sparse/bcsr.py,sha256=qSvCId7ZjWHNZNDVpEPdc_dTnB9446V5xsMiIFdpAxs,41314
jax/experimental/sparse/coo.py,sha256=_R22aml359eVXXBA_3BP4mQfrJAfmdCtD_BzCDLy2vw,24165
jax/experimental/sparse/csr.py,sha256=dhOQNlU57XwMZom4n8VVfhc8t4TP1OxPcwtDurK0il0,24332
jax/experimental/sparse/linalg.py,sha256=Zk0pMWAwHMExcC2VaAu8Pelr1rRcVW7K2G1io_XKh0Q,23411
jax/experimental/sparse/nm.py,sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I,8568
jax/experimental/sparse/random.py,sha256=o_g5_F8_EN4vmncmIkFKChU5uX4L3MVM6z-_qdoYUKI,3887
jax/experimental/sparse/test_util.py,sha256=hBw1SDgjVL2KkadjSdvIs5DcHqmdrtQHTHUCwj-MBfw,9433
jax/experimental/sparse/transform.py,sha256=HzxzEfAwUqwrpFXPx8ahQlARBZLOxWR4XBf8_9L_cuk,37484
jax/experimental/sparse/util.py,sha256=zb8BrK9aG5DC9ubS1ZmjSKfLhnnGVKA5ydy9i0hKzA8,4337
jax/experimental/topologies.py,sha256=YnSqioiXZqwsX-WzVhHVUO44lAib22-TkfIzC0KRfLA,2083
jax/experimental/transfer.py,sha256=E8dFkOjF0jnpEYk6--VEdnozMEcC-ntdSG-GkKln9eI,2327
jax/experimental/x64_context.py,sha256=XXTsGH7SavrMuT1hXzThT8f04SAFWdrhjZiyfEAaqVU,2394
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/extend/__init__.py,sha256=tFbMTUdo-HwWjrmedD-df0hndkPkqGWPuZT_Tb-Fspc,1600
jax/extend/__pycache__/__init__.cpython-311.pyc,,
jax/extend/__pycache__/backend.cpython-311.pyc,,
jax/extend/__pycache__/ffi.cpython-311.pyc,,
jax/extend/__pycache__/ifrt_programs.cpython-311.pyc,,
jax/extend/__pycache__/linear_util.cpython-311.pyc,,
jax/extend/__pycache__/random.cpython-311.pyc,,
jax/extend/__pycache__/source_info_util.cpython-311.pyc,,
jax/extend/backend.py,sha256=WOpw4RNsuCR_EMkNrikaWrtyUSp200m09z7QnXr2wgQ,1183
jax/extend/core/__init__.py,sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM,1036
jax/extend/core/__pycache__/__init__.cpython-311.pyc,,
jax/extend/core/__pycache__/primitives.cpython-311.pyc,,
jax/extend/core/primitives.py,sha256=GaZ_MirytqtZR9haKV_xv44I3TDS4Da9ZXf7a9OhZME,6299
jax/extend/ffi.py,sha256=31OhFUi5t8Msov6VnqpTj8U8bpDe9V746tMb1enGh9w,1783
jax/extend/ifrt_programs.py,sha256=TCWvR-K8nHQ9PuUNoMWVAfEuYsutTNrefKwLg4AIRao,787
jax/extend/linear_util.py,sha256=xag9NUqPndR7mh3NDQ4TluFpc6hFFd4r7Q5LarAFqvY,1592
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/__pycache__/__init__.cpython-311.pyc,,
jax/extend/mlir/__pycache__/ir.cpython-311.pyc,,
jax/extend/mlir/__pycache__/passmanager.cpython-311.pyc,,
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/__pycache__/__init__.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/arith.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/builtin.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/chlo.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/func.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/math.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/memref.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/scf.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/sdy.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-311.pyc,,
jax/extend/mlir/dialects/__pycache__/vector.cpython-311.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/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/random.py,sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ,1083
jax/extend/source_info_util.py,sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU,1215
jax/ffi.py,sha256=S4EjUxeohrl-Kmmnu8rykPCTxMl2LAxGdaerWLhd5ac,1052
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/image/__init__.py,sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI,1027
jax/image/__pycache__/__init__.cpython-311.pyc,,
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/__pycache__/__init__.cpython-311.pyc,,
jax/interpreters/__pycache__/ad.cpython-311.pyc,,
jax/interpreters/__pycache__/batching.cpython-311.pyc,,
jax/interpreters/__pycache__/mlir.cpython-311.pyc,,
jax/interpreters/__pycache__/partial_eval.cpython-311.pyc,,
jax/interpreters/__pycache__/pxla.cpython-311.pyc,,
jax/interpreters/__pycache__/xla.cpython-311.pyc,,
jax/interpreters/ad.py,sha256=p0gJFj9VvrsZbZavI-xg00Y-PzEu5Rw_6d3upq5u92Y,2759
jax/interpreters/batching.py,sha256=QqKOP4uX4F1PSf96ZrXi6x3DOIrU8wyWqXGTlt1nC2g,2574
jax/interpreters/mlir.py,sha256=H_cKoKAmfXBH7FEmMRthJ7M2Hjs5HeD4c-FkLFHLb1o,3411
jax/interpreters/partial_eval.py,sha256=DJBwBbcrQ7Wrj9EA_TyFNsiq6ZSymxqYpGgPZscdvvI,3924
jax/interpreters/pxla.py,sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE,1801
jax/interpreters/xla.py,sha256=JD6CBHHS9gTmO83age6Xwe9CxuoHLlYYBoD3RxyL41U,1584
jax/lax/__init__.py,sha256=0fm8wqN5TQHHNqzs5vlyEV77RplzZdp-JL4Km0tPr_0,12449
jax/lax/__pycache__/__init__.cpython-311.pyc,,
jax/lax/__pycache__/linalg.cpython-311.pyc,,
jax/lax/linalg.py,sha256=I9ZzEhenoEzcBnfnU-jCjIspi-aOXHc5KJY3t_45uWs,1631
jax/lib/__init__.py,sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M,774
jax/lib/__pycache__/__init__.cpython-311.pyc,,
jax/lib/__pycache__/xla_bridge.cpython-311.pyc,,
jax/lib/__pycache__/xla_client.cpython-311.pyc,,
jax/lib/__pycache__/xla_extension.cpython-311.pyc,,
jax/lib/xla_bridge.py,sha256=NtW-LD_Dh_TkdCT-19CZ91nvJPWy27aUzCon8ij3CWU,1246
jax/lib/xla_client.py,sha256=UEViLrEA2wi-MUO6ZYNF22r4yboy9TKekN645jEBkrM,5094
jax/lib/xla_extension.py,sha256=J5ldZNiB9BxU8wGkl1rWNUVTAGZP4jHSkZiMuVuOXf8,4428
jax/monitoring.py,sha256=xWZF_CyFg7Vv7MYmIyb3JWNA-AwsvIm9oc-MbGX4Bbw,1415
jax/nn/__init__.py,sha256=vC3v2Eg_WOW8kJWDzbpKS35jDaabCP9iwnOyV1HQEJ8,1730
jax/nn/__init__.pyi,sha256=GwG0FR6g3QtjFcmSM9jsARRM2QwXBNv2hRgaKkMPi9I,3833
jax/nn/__pycache__/__init__.cpython-311.pyc,,
jax/nn/__pycache__/initializers.cpython-311.pyc,,
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=-OT1qZuRyMi0dhKSfQ7DC1zkRfng2cNBh4AfqVnkuFY,12228
jax/numpy/__init__.pyi,sha256=sF36SyNjqOZNDV0KuB9RdJl_05VgP4p9mYU5xTYTY-M,44304
jax/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/numpy/__pycache__/fft.cpython-311.pyc,,
jax/numpy/__pycache__/linalg.cpython-311.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-311.pyc,,
jax/profiler.py,sha256=_aALYj1kkwBxGCHmiIzcdafpIBoJV2M5Kqw7VBNXQME,1387
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=-1oFQ0v1HFzwvFTipdl1JQVqsk8dXRJe2ZH4xTyU51g,9434
jax/scipy/__init__.py,sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58,1474
jax/scipy/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/__pycache__/fft.cpython-311.pyc,,
jax/scipy/__pycache__/integrate.cpython-311.pyc,,
jax/scipy/__pycache__/linalg.cpython-311.pyc,,
jax/scipy/__pycache__/ndimage.cpython-311.pyc,,
jax/scipy/__pycache__/signal.cpython-311.pyc,,
jax/scipy/__pycache__/special.cpython-311.pyc,,
jax/scipy/cluster/__init__.py,sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E,768
jax/scipy/cluster/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/cluster/__pycache__/vq.cpython-311.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-311.pyc,,
jax/scipy/linalg.py,sha256=GXC1BgLMztvwRMxddJhvJS3obG81HLQ0OJFVt_Iv4V8,1370
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-311.pyc,,
jax/scipy/signal.py,sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710,975
jax/scipy/sparse/__init__.py,sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo,757
jax/scipy/sparse/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/sparse/__pycache__/linalg.cpython-311.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-311.pyc,,
jax/scipy/spatial/__pycache__/transform.cpython-311.pyc,,
jax/scipy/spatial/transform.py,sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o,802
jax/scipy/special.py,sha256=s-0P9YKfZmmTze9BVkrr125COCf6H17G-GFvsmzvCOc,2406
jax/scipy/stats/__init__.py,sha256=ssiwAm2AdZ81MPMM11015yMNogj4DGsNfWfq507yhlo,1995
jax/scipy/stats/__pycache__/__init__.cpython-311.pyc,,
jax/scipy/stats/__pycache__/bernoulli.cpython-311.pyc,,
jax/scipy/stats/__pycache__/beta.cpython-311.pyc,,
jax/scipy/stats/__pycache__/betabinom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/binom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/cauchy.cpython-311.pyc,,
jax/scipy/stats/__pycache__/chi2.cpython-311.pyc,,
jax/scipy/stats/__pycache__/dirichlet.cpython-311.pyc,,
jax/scipy/stats/__pycache__/expon.cpython-311.pyc,,
jax/scipy/stats/__pycache__/gamma.cpython-311.pyc,,
jax/scipy/stats/__pycache__/gennorm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/geom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/laplace.cpython-311.pyc,,
jax/scipy/stats/__pycache__/logistic.cpython-311.pyc,,
jax/scipy/stats/__pycache__/multinomial.cpython-311.pyc,,
jax/scipy/stats/__pycache__/multivariate_normal.cpython-311.pyc,,
jax/scipy/stats/__pycache__/nbinom.cpython-311.pyc,,
jax/scipy/stats/__pycache__/norm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/pareto.cpython-311.pyc,,
jax/scipy/stats/__pycache__/poisson.cpython-311.pyc,,
jax/scipy/stats/__pycache__/t.cpython-311.pyc,,
jax/scipy/stats/__pycache__/truncnorm.cpython-311.pyc,,
jax/scipy/stats/__pycache__/uniform.cpython-311.pyc,,
jax/scipy/stats/__pycache__/vonmises.cpython-311.pyc,,
jax/scipy/stats/__pycache__/wrapcauchy.cpython-311.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/laplace.py,sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA,804
jax/scipy/stats/logistic.py,sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g,845
jax/scipy/stats/multinomial.py,sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU,794
jax/scipy/stats/multivariate_normal.py,sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ,802
jax/scipy/stats/nbinom.py,sha256=hWMMt7D_rLRtiGjlpHcbOXJF03BfA3v9Cu972twYKCY,660
jax/scipy/stats/norm.py,sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4,879
jax/scipy/stats/pareto.py,sha256=xOqP9JMLAYdVXMFpCfCoRqxJNXVeWfJXkEqpOsrUjHc,789
jax/scipy/stats/poisson.py,sha256=Guu0WOaGAwtujTTWVlBZiBg2fLu4JOjx0fumX4m4Ems,804
jax/scipy/stats/t.py,sha256=kPmOWYVodYyD6cTxzX98qNxiFeWpDHBM1WwrdK5Ztoc,784
jax/scipy/stats/truncnorm.py,sha256=q5aSLcidwhZqFi9sNjqvFF9RxT_YLcYBMW2DofDoE5M,855
jax/scipy/stats/uniform.py,sha256=KCmUJttnFPbr-_ECG7O5xd-fy8CqvysIhepltRj_6SU,818
jax/scipy/stats/vonmises.py,sha256=NYGQbIKES93vjVumCkF6IihydEhrhH7JMaJFmGDfCwM,791
jax/scipy/stats/wrapcauchy.py,sha256=pcs6O0cdJTyhsN6LGTOxBm4TyuKkm3HRIdxfZVoz8r4,793
jax/sharding.py,sha256=PHXvo9WjCtsXIzX5wAdpPCX8031W91yoiZjMto4VJs4,2072
jax/stages.py,sha256=M5yM7y1ubMHqBm6KJor_kJGnJDH8UXVoudgZ15cvaJI,1308
jax/test_util.py,sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs,835
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/__pycache__/__init__.cpython-311.pyc,,
jax/tools/__pycache__/colab_tpu.cpython-311.pyc,,
jax/tools/__pycache__/jax_to_ir.cpython-311.pyc,,
jax/tools/__pycache__/pgo_nsys_converter.cpython-311.pyc,,
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=fFvvhbWBsHJel1cBJJOBhLeacpk-OJVJzRE-hZHxK90,8711
jax/tools/pgo_nsys_converter.py,sha256=BCxXmHnvcorM62adEbOFNYcsnTyv-Hkqnf4K7lo_wt4,3413
jax/tree.py,sha256=cqFvisf_oyq5BWE18iwtiJc_PQpZ6O1FbZqAOiVfB8M,1106
jax/tree_util.py,sha256=XPwdmupZXBqEdw6EP2oM8Er6uyGv6Hi4ueV_W99g9CM,3676
jax/typing.py,sha256=zijMsLWzFOqn-u8__DbGxxI7MyjePxpUwU5nFE_P78Y,3319
jax/util.py,sha256=Ny5dah5vtOF0NJtr2UMJlxSJKl9AvdIX_vsbxo-zbho,4486
jax/version.py,sha256=q0iokoxglukg_Be9ST0SfrrRJipJZXBlt8AYrTuR4DI,6725