jax-0.4.36.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.36.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.4.36.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.36.dist-info/METADATA,sha256=cocKQLQ1rtTaMEQakWgcjM1fXp8kwhKva5O1-YVfAm4,22763
jax-0.4.36.dist-info/RECORD,,
jax-0.4.36.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
jax-0.4.36.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=jTnOMnMx0fc7hQ7gp5nMzktsV13yuAGAv_PorOFv4aQ,10056
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__/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__/basearray.cpython-311.pyc,,
jax/_src/__pycache__/blocked_sampler.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_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__/errors.cpython-311.pyc,,
jax/_src/__pycache__/flatten_util.cpython-311.pyc,,
jax/_src/__pycache__/hardware_utils.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__/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__/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_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=POBrGGiD0yHcgr1JoSluYQGsZQdnpRXpLvJweOxiC7w,2685
jax/_src/ad_checkpoint.py,sha256=Wwtm0-TLGoq5szq9526KGhL0nZX0nxRjPctOOW7GgyU,38624
jax/_src/ad_util.py,sha256=BiuAD36BB3OxV7xNMrQZZTB5aKELpmNiI0zaVZ8unUU,4220
jax/_src/api.py,sha256=okUrmuFWqeetWztv-ZAd7MxvM7mYXplzuGXqouRueT4,121105
jax/_src/api_util.py,sha256=GRpt9aetFfGUFqh3xj3qRL-vGw99V0GE91NHwSAXtpQ,28272
jax/_src/array.py,sha256=IIdvqC5yK2EGH9waLAjzJBDCkopstgy3OYPlTaeSeE0,46758
jax/_src/basearray.py,sha256=12hYSzFDsq0LVpPX8nQ-gH5mLE253PBWRSnYvQsuVwQ,6503
jax/_src/basearray.pyi,sha256=F8VxJtn9jB6iGzI-2B_kOedk1Pullmmyl21azn_k-CE,13022
jax/_src/blocked_sampler.py,sha256=rYk-BNwg0fIXWqgMXG5v3kjJw66Bod7vjNSQTa8atuw,5666
jax/_src/cache_key.py,sha256=P63nM2zuGwzsiemv32xMlUh6m6RFH0_waEVn2YURLPE,12997
jax/_src/callback.py,sha256=0sbAJzJEOBSOi_QXnjCTl2AQ7P0K07jOXx-YlsfeBPM,23789
jax/_src/checkify.py,sha256=XZj3bWnLVoHZTveXelw0rVqmlPOPtVLqiLSZszUb5zk,56278
jax/_src/cloud_tpu_init.py,sha256=SVyU-xfB2YxNfZtlv2ebpEgoSnp64WUlnyJuutz_oag,3519
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=xpeiJ-pVH6uNypUfIoxwNVaRuATiEtIqGVtNrvYN_lE,8405
jax/_src/clusters/cluster.py,sha256=4_heeOPzwH7NHumJKZspEfCReABCJC9SDfW8z3ORZp0,5856
jax/_src/clusters/k8s_cluster.py,sha256=TzUXU4ksY11X2f6iYjhb5LEGZfr6r5WZM5tc9DQhvBQ,4188
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=m2pr7zVMcYgCY2UsyaajGp5jTEZ1tztrh4YKCiYeZM0,11226
jax/_src/compilation_cache_interface.py,sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo,887
jax/_src/compiler.py,sha256=DsmF_ym7EiFkLJ3_j56gGsSuWj_LKdC_fZRmBdF454s,30375
jax/_src/compute_on.py,sha256=GYS3AbPWjfRgw9SsCFa7R7q2dpTluQeejqTjnQ3iVTk,2074
jax/_src/config.py,sha256=fwC8NOtuPZfJHr2LYQ8_KIQ6Zkzpj0eMajyzsusRZDA,78766
jax/_src/core.py,sha256=MwJCGihglG7jHWa2PpvbTlYbS8U9CxopR_4sw2-hnuI,107129
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/fused_attention_stablehlo.py,sha256=ZXB1i-YC9cavl1TkJ0jQNsh7snuNOBoOk5SNSs6Cj9g,41033
jax/_src/cudnn/fusion.py,sha256=y3uCu2pohDXpND3RoycHtHVv0agyKvlSJPkPjsbydeQ,2803
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=7BXjJzz_rOmFSxtLZ7EMwq7KB_RZWEV9EHb7CM8iiF0,10738
jax/_src/custom_derivatives.py,sha256=A3jAlDHSixATpAv0A106Twc-O0C6efCMarY1OzX2IrU,68868
jax/_src/custom_partitioning.py,sha256=S73d81hUd01NbDq9JXUbFiamdfgGAFssQKqPhB-71MM,22746
jax/_src/custom_partitioning_sharding_rule.py,sha256=Qcic-tHeS6LxkAldWX_cxs-YDepnOeSIgd-DpUcevCQ,14057
jax/_src/custom_transpose.py,sha256=8UOwMc39nsG90p1jDBw50_QZ8YX5a_qFP1CsFYFoV0c,7788
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=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew,4780
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=_XY-rTEjo_-dARgWAj6R_iE4WPM6wlzYlfEyOwQDpZc,25904
jax/_src/deprecations.py,sha256=_Jrs-t1EUGo5rJFfRexYGAbjuEqK7U4dONCrRPUm4rY,4670
jax/_src/dispatch.py,sha256=IDFgQSpHYkJ46szFAwKw0vPUuTSdnzSdCwflfEqb_00,23435
jax/_src/distributed.py,sha256=MDRG-sqa2zYnxqpGo7ShBFmNgwWXVXvSgyGHiHh5bEU,11878
jax/_src/dlpack.py,sha256=pzQdxNalm0JLOOZJyHaLYSAT5DfP_UTPHFl5lBSX9SU,11544
jax/_src/dtypes.py,sha256=2BAQDodqh9RhKNEbqAskWoVM0c0Rm2xp-FmW_HMATWk,33852
jax/_src/earray.py,sha256=ziY31_lJX_TmgpC87I1pq0xFZAv2BqZsEt3Kbe6XDBc,4500
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=TLwjyscnAh8rrQWePe1b14-IeSc-z-XP51Y92dMQeGI,1873
jax/_src/errors.py,sha256=s3zDLKcfvcCVSrJgNON0YF7bb7RB8Gp-xWinVHcvsNU,24741
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=KcZuGs5LMMFXLNsFZ7wtlJYypxjQVk1P0aJNjT8xBwQ,64765
jax/_src/export/serialization.py,sha256=WLEwoP7dnrQSBnz73KzXQXPs88Cl3jdrtjnpLiY4s9w,19203
jax/_src/export/serialization_generated.py,sha256=hd6uuEJUksQMgq5_SMVN06lfVaC41PPnMR0HVhngqVM,27772
jax/_src/export/shape_poly.py,sha256=lBf0BXfA0133vQM8M0v5hF2m_ZUOlpgpoFlpgXxHvYU,84926
jax/_src/export/shape_poly_decision.py,sha256=7cEAydAto4UiXlOYdq7oHqp4eXwp4lc3OAY-KKgy9Bo,20904
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-311.pyc,,
jax/_src/extend/__pycache__/ffi.cpython-311.pyc,,
jax/_src/extend/__pycache__/random.cpython-311.pyc,,
jax/_src/extend/ffi.py,sha256=_FiABpKZx-eqeWega5A4DkXOlf3RdMemXbI3grGRxLY,24380
jax/_src/extend/random.py,sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw,1258
jax/_src/flatten_util.py,sha256=T8p7oExtun3ddhh8fn4XL5_TzZEh6ChM_qiZhmcPaOk,3554
jax/_src/hardware_utils.py,sha256=L03Dapnam8MdnSyBGq2d8nio5gPF51xd_Ufxw2pVjCE,1728
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=_Hezs_YE3kCj6_cTn6XvMuU26sPiedsiJR8-VVhKEDA,13740
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_util.py,sha256=S8wDJb98vYYeIq-omRo_b8jErOBa2DJAoX9mO1V-cI8,15336
jax/_src/internal_test_util/lax_test_util.py,sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo,12140
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=d3BdtWFyEKoB8v8vWTbp01UfhQkzHagQkpSptra28zc,126010
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=41U83lw9BEDKwvjpQvbRshJJ1CeqrukWYJmOsM1-bbE,37281
jax/_src/interpreters/batching.py,sha256=esSUzj-TcBvjMSQbVxhx56fQEf8nfMoALVKR8NAUH6c,47526
jax/_src/interpreters/mlir.py,sha256=U7TkGg4DIM96oX4dT1DWR4OrmquvWAPvkde2jXWVwMg,131205
jax/_src/interpreters/partial_eval.py,sha256=jkcBaF2WocVRqMXsuWyYmGOrao2L4J9zFmZIWFAcaXc,114238
jax/_src/interpreters/pxla.py,sha256=lLZUbsQ4sbbxiLVTzPYVG2UJ3-7JbpgBo8Xub-mw-ec,130046
jax/_src/interpreters/xla.py,sha256=honVedIyTBp9bbf80DvYFmaSV0MQBrCeyX1eCnTJbBU,7002
jax/_src/jaxpr_util.py,sha256=Jv3fcRlWFNUt-WQwHYfmLeXOSKAYka9RKEwrkYl-BkQ,6439
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__/eigh.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__/qdwh.cpython-311.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-311.pyc,,
jax/_src/lax/__pycache__/special.cpython-311.pyc,,
jax/_src/lax/__pycache__/stack.cpython-311.pyc,,
jax/_src/lax/__pycache__/svd.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=53ap20vxBD-hV6L1JZsLNfzyACtK8qkE2fmPqsJivvk,16761
jax/_src/lax/control_flow/__init__.py,sha256=If-Hh196w69VMIk8l2iCXfcRL4C5Dpi3clMO-qdbtmI,2158
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=Wpr7T_Mm9I3WHLCO_aH-LxjyVWK6h0DtuO_BpMzLlMc,11041
jax/_src/lax/control_flow/conditionals.py,sha256=fmt1S5eXDloQl6RLUtj5AGxmzmxy9AEA4b8IrIjA50E,42552
jax/_src/lax/control_flow/for_loop.py,sha256=lCeaMSeJq8E4KkrOoqCo5jmmXMqzp1JzvDvahvZGUZo,35776
jax/_src/lax/control_flow/loops.py,sha256=lMjr9YrirnX5znltO3APSwZptnMCG1ELnA9sxfz0W20,110001
jax/_src/lax/control_flow/solves.py,sha256=AuQGNm2c_Btx9IhYApjq-VeRO8tb9dPrSFxBaPhc9yo,18399
jax/_src/lax/convolution.py,sha256=kxGQ5J18irVz3S1sxqoqlTW6otJO_5g3zPCHi09FRhY,44551
jax/_src/lax/eigh.py,sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ,22235
jax/_src/lax/fft.py,sha256=fIrp5WrJCZM0FYZpZnmC67Y7rsWCthbOwK3GXryoEqQ,7258
jax/_src/lax/lax.py,sha256=8y9Bpjk2ST5Pbwc2ECQ7CMk0DanIW58QjUrGnSpg6s0,263587
jax/_src/lax/linalg.py,sha256=0Dl_RJQ5tNNtBmBrtNzi2vAtVki4CqKApFF1zftFR_k,105570
jax/_src/lax/other.py,sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc,12497
jax/_src/lax/parallel.py,sha256=mMlcCdDLG-Ii0lrtNjAnnoWot1vjMpWqZ9qsL9UO5l4,66226
jax/_src/lax/qdwh.py,sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo,9391
jax/_src/lax/slicing.py,sha256=VbGnkcBuJQS1J5Tovh6u9oMNxf6yMRuvPqqEaLxsX5o,128717
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=iVY3iFwt0ug1ow_SkFIJd0lNYdn9eXrCKhNbZ0Dh4Bc,8794
jax/_src/lax/utils.py,sha256=_riaCd94rIBTYv_2P_OIAP08ADdZ5b5rssO5BT6QgGM,4193
jax/_src/lax/windowed_reductions.py,sha256=4QHpAFd5_FYroVr5WM5KI4nsQk_v6hBFDTvXstYwggU,42352
jax/_src/lax_reference.py,sha256=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM,18163
jax/_src/layout.py,sha256=Km3wPwPjWyBE37S4V0prIvE0FqI56Nq1u0TCVhuL3tE,5375
jax/_src/lazy_loader.py,sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw,1842
jax/_src/lib/__init__.py,sha256=bUYdjvw8d2XkBpIYLe2i31-v8yu5BEMlM7MpdY4uBa4,6807
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=6KUUW3fBmKZn0eM7VlIFh5yiSLRQ9PNC1Cl9tAxoF2o,2032
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=6zzcpqutI1cZW4sG9tLwPFyPDw35xhpPoNH8wkG-VbE,662
jax/_src/linear_util.py,sha256=BE8f0Ni748OUXGS2R3kAJsK2e_pyzRW-YRgOp8ElvxQ,13284
jax/_src/logging_config.py,sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk,4452
jax/_src/lru_cache.py,sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s,6661
jax/_src/mesh.py,sha256=ZRj8RFC8j_fZUVIutsRH1bHSwwTcj5FURaukxb3uJ58,16400
jax/_src/mesh_utils.py,sha256=JRcbw16UH94co4A6eSlZ_7iupL_JuYdyNiCSO6tzPgM,32307
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
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=S0TYEuHZ1X4yk_0JCKdKtwB5e97sWbuXvyoKHsuYbe8,36330
jax/_src/nn/initializers.py,sha256=TOGKwbExnXQvMai1amFpEH2PFaZxiVz95KGBVaopnmA,25450
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/__pycache__/__init__.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_api_metadata.cpython-311.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-311.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-311.pyc,,
jax/_src/numpy/__pycache__/index_tricks.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__/setops.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/array_api_metadata.py,sha256=je15NAJj_CFlgqeDqo5iSg9EzmN9IwCkZ_E8TtStKcg,3347
jax/_src/numpy/array_methods.py,sha256=qBPXDtPKTZRchwMuXSDV_zEzRUpG92OTYYaTcZIm2Jw,45665
jax/_src/numpy/fft.py,sha256=ZZih9b0NDQYrD82RwdR4UwqrbXvIKv-px18hyANG9wE,51814
jax/_src/numpy/index_tricks.py,sha256=aRFc9So43wmgzk-eVGBeN16BI4Ny7eXYFr7yEtrgDqM,9992
jax/_src/numpy/lax_numpy.py,sha256=oYcJc9TD1O7P0roArZpYL5W5CWYjsfBynG-kAt61ziU,483794
jax/_src/numpy/linalg.py,sha256=O3cQthEdsbumSrAge-vqtrVWvJf89oQeXjysYvr2G1E,77607
jax/_src/numpy/polynomial.py,sha256=mjOS7TXk5eWDkKPXr5iXNkwX7T39VnWtf8GHBcEspjw,27906
jax/_src/numpy/reductions.py,sha256=khOHS0fiXYX4A9aLUX5_m2zd0nuJ9WA5t4QTYx4KFTw,109085
jax/_src/numpy/setops.py,sha256=Uptcxj3KK04zB5P4wTrr0Qoj0INwDl44d9zEQmvXl8E,47405
jax/_src/numpy/ufunc_api.py,sha256=VXdn5cXraEXPjBXUTTU_jhWFGrdOF5NlrDEXHBhyn0w,24204
jax/_src/numpy/ufuncs.py,sha256=2AYPa4CZWiX_w7KM4XusPTbMRC3FZzAcvaOR-dAV82s,116122
jax/_src/numpy/util.py,sha256=C3nXUYWYhb__hGNx8y3aX7YkKp8Uh5wtkoE6ag4k07k,11666
jax/_src/numpy/vectorize.py,sha256=lugYJ7HrEpG-L0yNd9UD9mEa6ASQlaaACdb72PFtat4,14018
jax/_src/op_shardings.py,sha256=ZYx-ASvqGPXifwEBC1utdbAJA-3ZCgWM037y6l4Wp4I,3865
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=wsV-jnH9epkp-GjIy8AG9xgpVoBUrPEmaj6h0eXhBis,17625
jax/_src/ops/special.py,sha256=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA,4151
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__/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=k2rdm8pQIqbejiDQnTAZNpJ7tvrpA3KslQ-BdCXMS3Y,37982
jax/_src/pallas/cost_estimate.py,sha256=6x4ApjjBmd0IwDUuWCRIH42UE7u3hAXp8mYdziL_ZUg,7881
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__/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=KP2vNaGAMJOiMVYicaSIvTR1Z7Mz0CbJP7NHkab9Bg8,8069
jax/_src/pallas/mosaic/error_handling.py,sha256=nmjLsqhnE9-foi7lwMAWdWNQMdHd2jOMqJtDxhBhr1o,5480
jax/_src/pallas/mosaic/lowering.py,sha256=QstdKZu69sCplB2DZPQTGuXfpkFXgdtSPZL_O7msxGo,115395
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=_KWTj-D2OJLoX9BKesTVmpeSV1LtEMDWx7uUOBglgDM,9008
jax/_src/pallas/mosaic/pipeline.py,sha256=8ShG9SBvCySphlB-GC2glH8nDyU3euVwXt9_rH4H9Ag,44060
jax/_src/pallas/mosaic/primitives.py,sha256=qmN9pfBGaDcq6P-bYURHVvodU9XhVBFOvA518jNV0n4,30444
jax/_src/pallas/mosaic/random.py,sha256=yBP4tvNPUodwWMSssHDUJkwW92uU3K2wtvziRg00CpM,7989
jax/_src/pallas/mosaic/verification.py,sha256=fQAVzbETThqBWA_yp2jLbeX_5meoVShh3flYgQWiF9g,24540
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__/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=QD9jq4MRHz97JwMdzNmUoea3BIfwobIS5YrlCfm5XYI,17451
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=QiZSIDbvJgcgPnDqYze6pGe9dRln6VO0qa_ek19GPkA,62623
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=KZjV8d6zSprGhLDSEOTX8RI5q8jGJAyMiqmI5HzZYm4,3513
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=QiwvxB4cuKNqiWNBJJ4Xc7He0H_BOYAzvBUUiMqHUrw,11190
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=uAJqdDGn02nECaM1jmpEAsLbhbIODpuynI82-t5g74s,24221
jax/_src/pallas/pallas_call.py,sha256=_rt3__g5_t8dy9rtepk6st_9yrc8ik9NcjBpfl8quwA,74947
jax/_src/pallas/primitives.py,sha256=rB0fzchNW6qhDurO373EWcw5XXhkRDvydrFECXp-WUQ,32411
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=5u6hubCM5N7aSYK5iExZpffEGrmlpqR6Op1xd3dOhDk,1374
jax/_src/pallas/triton/lowering.py,sha256=9Kyx-SS2JsqWJCp12DL1f3mkHF5JZGg-LodRIuwlOXk,88821
jax/_src/pallas/triton/pallas_call_registration.py,sha256=lX-Z9cj2H6imyOYMaxA2UmWZeLybltVNBOTKv_LfK6c,4395
jax/_src/pallas/triton/primitives.py,sha256=on8o0zgU7H19hChzji_twx3QAORFLwOEDxj4Zgd52V4,4223
jax/_src/pallas/utils.py,sha256=tNjoP20oRWC4A-ns5aSUB5LjRCS11fy4E1SH4rw-8-M,14067
jax/_src/partition_spec.py,sha256=8ic-b6aezlLLHV1PqwajxzDAdMMdg3WXpspj70OvfCo,2206
jax/_src/path.py,sha256=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A,1452
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=YcqtCrLLq2a-yzqcetetjdijUpeoVQSC5-19zIiN6nc,121176
jax/_src/pretty_printer.py,sha256=mnGHSMkb8EaXh_1_9dsKOGh2XI_9dSaaAUvFJP-XdMM,15500
jax/_src/prng.py,sha256=yRUtepFx84EMc3M0mIAx_70wld2XvQWCm76YA9ncGAk,44346
jax/_src/profiler.py,sha256=278hQdUi8B7FVXo-YKhY9jf1qXSaBNem50ZAeB6F4DI,16195
jax/_src/public_test_util.py,sha256=SZxqWnOEobpLnMpXA18FAFrhdU9YG0y-JvTZDTE2p_g,11152
jax/_src/random.py,sha256=Uie38yies6ZSHN_C6nHEUZTNHhERIJ0YzRv2myIAF38,100231
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=rEOuNrkRnMOofq0i8XunDVONpgMyUws_vn7HgGCjHAs,15266
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=uNk5a101MNAn-bnvLaKnV7khyhgnur3_1Be1rLW2LLU,75465
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=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs,47453
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=hz2kXBC7VlrJfs_phRbioCiO2xWDt6N0Fo-KtiLfST0,17513
jax/_src/scipy/special.py,sha256=oQaxHPa3lTbqdOqpYGZEb-IlhF7FgJf5bWCn2kDXePU,80012
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=Gq8H86uI_wWodhyvdEy5GiNulBfW3kJZvm8Z0Qw6Vdc,10653
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=8q5nUo_N8Rl8BjMVfFKv2ydB-x7-TkKtIdZi5FVrbl4,2268
jax/_src/scipy/stats/gamma.py,sha256=W5B8z6LPD3Zi3cbcWHDB-RU_OuKjXopJIspf_VKZMw0,6987
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=Mi9nMB6q3eueSbyNMfs3DokLhsjte0lYsSIdAZ9Birc,10141
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=qjvCzQLe_ef39MlT98hUv9WPtXkO5ctX8pntSYRpqfo,4028
jax/_src/sharding.py,sha256=frnT8rcK_erouiHIyySG20lFYlZXjfzmh3B9WtXEzMw,8102
jax/_src/sharding_impls.py,sha256=NB2BFUeKS0bs-4zcb_74BP5tvGGasxZ_Ge-5wPsd3qw,65009
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE,10140
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=GbizwdOf97W_wRrseL8RIXn_bHgXr1KLMUbNmslPwac,29088
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=h3gUeMrkGJU6jttRV9NZYlXjgmw0EDdHnHSogLP8TvM,44284
jax/_src/state/indexing.py,sha256=p8oEIfj8oIv1M3VhQBoQI2gkG4AJ19MiPayqlDZC6xw,9264
jax/_src/state/primitives.py,sha256=I25CEEqRDss6pmLgz1yXwFgw2ABruVCm0yQELa7SlTE,25041
jax/_src/state/types.py,sha256=Thgdqutsqix4dc5eZDY-kWiLP0bb9aBEOwEb_kPV5y0,12484
jax/_src/state/utils.py,sha256=fv7oalhAmGZhOmhdvGcZo8Z5vbtlda-fx8mPfXx07p8,4100
jax/_src/test_util.py,sha256=VDYFgLrZwuBDQJsxAGR9C6EWFi8Tzz0T7GIdthdgT_k,77930
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=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU,3495
jax/_src/third_party/scipy/special.py,sha256=ojE5A5mClaexr_qca69NYpBTgvBEdvVaVMRmptVh0yE,9155
jax/_src/tpu_custom_call.py,sha256=u-x96aPJ7Jkuioxu2By3u2zljkchNYhq5f_WVNwst5Y,25508
jax/_src/traceback_util.py,sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo,9103
jax/_src/tree.py,sha256=pQY3bwJ4puT4TZqPcAmmeAdoeHsoG4EkMwFPN-0iwqQ,9310
jax/_src/tree_util.py,sha256=Qg8_vK5_EqbLMEPZ6hm-71ulNMS5cmLKHr8Y8W7UyXs,51585
jax/_src/typing.py,sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw,3287
jax/_src/util.py,sha256=nfAoVFR6RXLHXsmhL3FYLg4pYynWCPFAHyzCkgAamSQ,19806
jax/_src/xla_bridge.py,sha256=XZ1P2wpMB455ZtNfmxKNEQiHOwhZMNFLI_UitkaWODw,46381
jax/_src/xla_metadata.py,sha256=IvJL8QejbnCGogGcaTCMMcseAWsKW0JouZrCRzUxV9c,1519
jax/ad_checkpoint.py,sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U,932
jax/api_util.py,sha256=Jd8iVlf7i8crDpyvCg7OTaumZjtGvTIO9pUHXlXmQsU,921
jax/cloud_tpu_init.py,sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs,651
jax/collect_profile.py,sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk,4730
jax/core.py,sha256=i3TzKo9Xw6TiY20dWlGDXggiQ1rwbEGREtsiCGhe4uk,6457
jax/custom_batching.py,sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8,691
jax/custom_derivatives.py,sha256=-yrA5kDE7gDErJFdartoj43ELYeogp6PI8kjP8p_TLk,1453
jax/custom_transpose.py,sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc,664
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=D9-SeP4MWzq9jw2itQrlm10cn5BtR-9Zjn2YGzEXBDU,673
jax/dlpack.py,sha256=TWzMqAIedhARYUrnreJ35tEYt0SepzrpvTQtmnSdhEw,710
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=OQS_hX2SykwFBXHl4t0VHv5WKOsqpLjuVL5wXg0CZik,1012
jax/experimental/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/__pycache__/attrs.cpython-311.pyc,,
jax/experimental/__pycache__/checkify.cpython-311.pyc,,
jax/experimental/__pycache__/compute_on.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_alike.cpython-311.pyc,,
jax/experimental/__pycache__/shard_map.cpython-311.pyc,,
jax/experimental/__pycache__/topologies.cpython-311.pyc,,
jax/experimental/__pycache__/x64_context.cpython-311.pyc,,
jax/experimental/__pycache__/xla_metadata.cpython-311.pyc,,
jax/experimental/array_api/__init__.py,sha256=JYhDscKw62F_3Y0uaTh2SrJADrKBOOQwwavlvXo0KKA,1090
jax/experimental/array_api/__pycache__/__init__.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__/serialization.cpython-311.pyc,,
jax/experimental/array_serialization/__pycache__/serialization_test.cpython-311.pyc,,
jax/experimental/array_serialization/serialization.py,sha256=VS2pBc2z5E7QPQ5-Xk8HmNwI5_kR7kOjvyMdguQyb8k,26055
jax/experimental/array_serialization/serialization_test.py,sha256=BJ-FXawRqDfl6G6SXrDpl_4XaZy0IP2B2SodDrve6n8,23222
jax/experimental/attrs.py,sha256=6WiYIKhAd-awTy59aQWeEQZzkAOjgVpHCoDYmn_PCug,9601
jax/experimental/checkify.py,sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0,1213
jax/experimental/colocated_python/__init__.py,sha256=bpuYP80GTKaGnrDoSTwjYRb_Faoi2Kz7E0il-40CVvw,926
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__/serialization.cpython-311.pyc,,
jax/experimental/colocated_python/api.py,sha256=iei8HCFVPlzvcB8vm5_yNyRceF9l-QoxX0822ymB5Ug,2171
jax/experimental/colocated_python/func.py,sha256=eyCZpQZrAXCJe09_VnrxzYQV9RpJ8g5Nm-mfUobBzuk,17245
jax/experimental/colocated_python/func_backend.py,sha256=lXN-FRbWIWezsICyIc3QYwjpF9-tYgZV1Ut7TBN_dpE,1373
jax/experimental/colocated_python/serialization.py,sha256=w8pKcncrWWKZtezgWhVb7BzLuSeUAm67b-8PPqbBU0s,8635
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_partitioning.py,sha256=E0IrgU4LX1P-khUlG69nf9d4SPi9wbaBmyTiyvE6PbU,856
jax/experimental/host_callback.py,sha256=cuZhGDILVkhFgEFjgwSRH4C__ySPwxeJjxvPVNXf-b8,1207
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=5_UtMy3cx45NHQ2OIVXRZTJCaFKStDZoYGA1UfPcC0Y,28243
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-311.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-311.pyc,,
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=AN1R_wzL4NyCgMYEb6ab8fos2vruCL0PGj_GYAEWNEs,2936
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=0nnk1mDlssNRhOIG4u1_WSx6KXF9C7F5Eih6f5hFWjw,1670
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=c-Bno2RVxfY4aVgMRK9sjrCInqWVpTIBIQH3C9nPyIQ,11512
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8,7031
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U,8003
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g,2467
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-311.pyc,,
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=De9TWLSyC5EbkQRg00S9bvVOo17HJuS_M4cztRFjOa4,4912
jax/experimental/jax2tf/impl_no_xla.py,sha256=0qzCj3AFxvRIm-iCFykCT2tFgo5ghlJd8BXcK-HDzrs,54474
jax/experimental/jax2tf/jax2tf.py,sha256=kZTDchp9rMjHXO4IVqTHxtipTKldOf9OZWcxxyk7a1M,145498
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/converters.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-311.pyc,,
jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-311.pyc,,
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=xOP6CFs9sD3yKiI2vkk0W4rPKdB5T5C_4IA1E5-H09g,5360
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=Dox3xHLclyOzyWXTZm2zbY8jJPaprcI5LdEbG3nz2Dw,67020
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg,10108
jax/experimental/jax2tf/tests/converters.py,sha256=uYSiDOPUNaxcgVwLExAgquiK6M0VFH7j55OnbruOyaM,1998
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=holjJ2qso1R1VD_QSmKj-HnSoh_dkJgkOVLVqZqPIcA,7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=FWkCsZeRHZq93XP2No2CtiO4uLq24AcLbh6kLns1Frw,60812
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=NiPZyHCLBQwRKkAYSCX51lAKo7zqfGsQtmxlFz1NeXw,67530
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=2ouDpIkKBvLLomEsmYU8jMKIm4iy4xKjONdlEG1Fy-s,6645
jax/experimental/jax2tf/tests/model_harness.py,sha256=zhPlq2VnostLmvDfkPdE74A6UPec7wLRS6c_0oFwaLA,13934
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=DUaRRB6pQ7ObD-wDAVXlVafYYAuvX9OaxFLX4CbBfeM,14278
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0,13067
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=WmFsKMQOzlFVsLR7O2i5o_M_duh0FIydetFq_8mEUVQ,126535
jax/experimental/jax2tf/tests/sharding_test.py,sha256=cGnkFXOsvqP20Q5IRaht_znOJJmVMQUQMeTZyvddkOc,25883
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg,19962
jax/experimental/jet.py,sha256=YxlrKwLlhfGWrTZfJrOksB7H9-CxDi1LlGIVSgkU2CM,27023
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=NVFNMCQZg0UOZhHD0RpiDqlXum6A48s2z79ETOP9rb0,22781
jax/experimental/layout.py,sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY,675
jax/experimental/mesh_utils.py,sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY,930
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=GxZYu4vjLePXP6DYuekzYN6VAQgM5tLg50kch3FxUIA,2665
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__/profiler.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=c2svpoODGyyMwwUuvuU23XT0UznqRfsXixUked_JUQ4,40500
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=fZ6BTfCfw431UtzeawX-BzKgBRcvf_4l0s1ReM2fn-c,4108
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=-9C_an5JXRtyuD7LNsacDNHeYZN3_1k64iG1Ux1eKEM,76157
jax/experimental/mosaic/gpu/profiler.py,sha256=ToyAg9odFPmIE7pbTU9m-KvaDABj7x8KiBOVKN35kjQ,11336
jax/experimental/mosaic/gpu/utils.py,sha256=ZDzIEWLaCYbtChPS1sc23mBvNDjmiZVnoOHYyeJO8oY,35211
jax/experimental/mosaic/gpu/wgmma.py,sha256=kDsIBYlSViDyP8TsNbKP3vYoPXq4Px1S081c-OJtciU,16152
jax/experimental/multihost_utils.py,sha256=RgRWj6wxNSyuTWloxpJ_tTaZV_sqqxVXevnu_GoCaeA,18372
jax/experimental/ode.py,sha256=zvQAZHbxgGHA9fAhhq7KVxGefqO9-E0kB726eXc18Pc,10646
jax/experimental/pallas/__init__.py,sha256=RaOXR9hWOTxrIUOZHum3hmwLzNclK3XZejmENiKd7Wk,3295
jax/experimental/pallas/__pycache__/__init__.cpython-311.pyc,,
jax/experimental/pallas/__pycache__/gpu.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/gpu.py,sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I,865
jax/experimental/pallas/mosaic_gpu.py,sha256=FbA96Z4GmD06pcAjhDiGxiow5AFGObNFEf9eFiyNYCw,2823
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__/decode_attention.cpython-311.pyc,,
jax/experimental/pallas/ops/gpu/__pycache__/layer_norm.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=u7C3fYnWEVr8Df1L7cHFK8CEFYiHoV3mdi_Q2_e6fz0,19616
jax/experimental/pallas/ops/gpu/attention_mgpu.py,sha256=a_htFI_JeczAHvDmdumHA-B4WKFbrM9tFaHPPvPHgME,12232
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=-J86zYUnqRKU4YI6FzzqQ45xmWa5Hw3iAAiSohOLYvE,16321
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=OAdBY-Ijtai5IOUfXafJAtudlOUrli0g-_Qhc-t9jQU,11142
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=hNREA4EYeqPbF3RAijhEvtFk4jP8IiNy5LhPGcQldwU,10033
jax/experimental/pallas/ops/gpu/softmax.py,sha256=ydajdpaW_oAOwwdGJUx-t2TmIx9QfHhnXX_lNdtmVL4,2774
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=oLZefmIUQ1Ag_C3kz5h114E2l4OiW9-Nn6DkpB4AC8I,5581
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=2g9jECYvE3S-HnVPmABhsQgFsiw6JU3wacYhYIGFAWs,50787
jax/experimental/pallas/ops/tpu/matmul.py,sha256=jfpCbL3yQ37So4yOtb9pF6K7PtA0WmV65yrzQ8yp4cs,2726
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=vNNgQIirDK7pyDViP2foec7GC8EhJuxNjR1mp3kHwjU,27249
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/paged_attention_kernel.py,sha256=f6z0ukhHjObUOYoWsLc8Wxvhv8g3UVpjGxUixdnf_UE,21960
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
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=RoYvEAClmyg5mhWDNY6q-KRUfNgygvIn_cEo8rBzT6Y,73483
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=o-9hXrMgy6vq7nq8SiUFPceTGUdvvAsXNHR1bnjPa5A,14554
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=epW1K_v5OPLTqMr9dFDIjh89sxb3BX7ReDNTi47gDB4,33385
jax/experimental/pallas/tpu.py,sha256=Dmx7XjC55WtLzn3ZFBBwAYCoK0c7IpFTdoBXwRCQC44,3891
jax/experimental/pallas/triton.py,sha256=8FEMurTT9d-X8tipH-706ETuSLjFkT0LGWiOHjp8zfM,948
jax/experimental/pjit.py,sha256=2Qi4A4tKn1p6pjxysiCY5UAsfQq_KVZ7QpMX3gRJw30,749
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=kxuSVk5trxvOkWqPLUJ6iZqROD3lSKxcsMA7Uq5IRok,18777
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=mg9vpinasUF8c57r7IjniQDkD1JKkAQDXKtzTPaQ3V4,10004
jax/experimental/roofline/rooflines.py,sha256=_1MGFjZaRvcZlN-E_C6nLFVfq6oEPN4slTQ1Dm-K8DQ,8169
jax/experimental/serialize_executable.py,sha256=DYfargD1RSL8Uc65iFWOjm1MU1WArjt4Q4t5AZTTz_0,3006
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=JZLlYvNCvoos4qP3Gq1B_u7Xat-23kVO2HFZt9gL4gU,90292
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=NW3WFfrvsf5O9sxB4vN6fMj6b7F0NxtQEuj3L1oO-NA,7661
jax/experimental/sparse/ad.py,sha256=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE,7784
jax/experimental/sparse/api.py,sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g,6596
jax/experimental/sparse/bcoo.py,sha256=Z5XBQ1dSS52xSKQoDSSnl7dcpgsq2GIZJcpBr6etDYI,128994
jax/experimental/sparse/bcsr.py,sha256=gs4LWREQFXYokwqNSA5_zmZrYAwGT6NKfpLIWTBjROc,38290
jax/experimental/sparse/coo.py,sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg,24143
jax/experimental/sparse/csr.py,sha256=bBNOl1KR8WwAW0MtiBNKfebV3W2qTW58mA7lsH6Kotg,24396
jax/experimental/sparse/linalg.py,sha256=G7ndX6rm2nds_e3DglFPogdcNq4vHOc1kmpZd3RHl0A,23435
jax/experimental/sparse/nm.py,sha256=_yfZ2HKjZmZUicFG5BOC_sqlKfo79K7nCvw9sU4Ih3M,8563
jax/experimental/sparse/random.py,sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA,3882
jax/experimental/sparse/test_util.py,sha256=4JlLtN_KEl_bzF-ts9KRSbMNbzUsjEB_6IyQAhw7WQA,9428
jax/experimental/sparse/transform.py,sha256=iTPd6BBQEZ0Hrx-XtVn6_xesTKIi4KPo9_Ee6yxVfHs,36898
jax/experimental/sparse/util.py,sha256=TcMp_1-74yfTUU1q35lXL5ufjSBZ12xFof3-Vpnbd8E,4328
jax/experimental/topologies.py,sha256=Lxxb2W6nT6AEhM2ekkNUKm6W1tgh5i-_49454NuxCkQ,2101
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/export.py,sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA,1537
jax/extend/__init__.py,sha256=cL0QjoXSmMpM_j6_QBlKN6dtl30zMbvHqYA3eoHsd4g,1630
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=ODjEVf5_5U37PZk5WVp8wI8thqOZO8XAdRV6Rortr8E,1054
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=seKQ2n-x781qmStZiw9ifJyqbl0QBItrCsMimf5n9FI,6376
jax/extend/ffi.py,sha256=Bo8zJigDbyClf75aMNeDPKKJoWQTNdJsp2fvctaVH2g,915
jax/extend/ifrt_programs.py,sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc,801
jax/extend/linear_util.py,sha256=Ie3ii7eVvtHFWsbtlOVu3o_T4jzP_JiXcKDnl-QKm6E,1080
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=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc,746
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/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=9Kg6EyCFqLqVIxvAo6euAo_o3UYVmZxJcfWPgcFSrBk,2924
jax/interpreters/partial_eval.py,sha256=Qjptpf1zUvUt6mUK-RF3JqqY3Aq9uBIB4uH2JZwBb2I,4163
jax/interpreters/pxla.py,sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE,1801
jax/interpreters/xla.py,sha256=NLL4fnTDuhZMHIiAUu5Nk1R9zlo7q3AlNd7ym4flf7U,1613
jax/lax/__init__.py,sha256=Au4q1_n2KzICN24ZOW1Ut7K8SBMMUsVcYRXHPToY6Vg,10686
jax/lax/__pycache__/__init__.cpython-311.pyc,,
jax/lax/__pycache__/linalg.cpython-311.pyc,,
jax/lax/linalg.py,sha256=xvbvXnAOPZ_z97hjoaYzqlyM9verSaMqaL5ZrEsulK0,1397
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=RlgLzihbRKJNPKVumdx1QojpZ-xf9XScRmJO708ZjRc,1767
jax/lib/xla_client.py,sha256=-wgVItB6gDGce-6iyIepyPw3vaQccDNviQInp6T884U,3888
jax/lib/xla_extension.py,sha256=xuSR42eY6codcchsSZx8rJZ2MdokxbJDNZwW9Xldz6o,1942
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/nn/__init__.py,sha256=4LyqLzosWMtCv_zCd-9Gt9XmfCohbduCQrN_cRPqvVg,1562
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=S_tWfBH-1fVlo_FhD-uC_i_Sb6gDn_15w3fr67Ohzvw,12360
jax/numpy/__init__.pyi,sha256=aezQ-8ffKVDiJgZoPCtzLbUoQx2EOEOwOreeoe6s5Zg,43330
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=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=N3MzWsmgxQtps5khNgCzqGY5Ix7u4jfXLfgzWCteSp4,9420
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=jLoDaaxGZTcRu94dbVgk7qa4Nm-CHu_L8rMsvEIwzWI,1350
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=hT-5s-5ki7yXeQG3ZO-GMTjSrx3JHla41UNnQ0Njwsk,1718
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=D4a9KT9p2u0vRjpBS6sQgZ95vuzyZLCh-Mon8HiNq4U,788
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=2Id3LLKs6Ms3Ou8xzUZ1ZbWqkfSMv4S_bAqxfDOxz60,1624
jax/stages.py,sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218,1314
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__/build_utils.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/build_utils.py,sha256=nu6hxCk0o1Z06Tydm8msAqSmoZnjMOq6kY58qpCfHwk,4393
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=Z8LkCMaXfqjYFSHPSTS3L5imebmYtV3NPLy7V1WMLPU,8776
jax/tools/pgo_nsys_converter.py,sha256=BCxXmHnvcorM62adEbOFNYcsnTyv-Hkqnf4K7lo_wt4,3413
jax/tree.py,sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA,956
jax/tree_util.py,sha256=ZCuOkjx2V6A0EVV3e9sp7JtvhI5sxHF7u1-Vqpak-es,3125
jax/typing.py,sha256=CrQpi9UTB9MYfg-zEsqP3n2qe28hlY_QEsgY-9mrtAs,3331
jax/util.py,sha256=OXU0bu3n2-y901BwEhxd1CvOeGoEBaXPdvYaVNHOT3E,1135
jax/version.py,sha256=Aprz6IE6z3hyZIqREWgzSiLALUIeyEHfxQ3g-BV11lE,6013
