jax-0.4.25.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.25.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.4.25.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.25.dist-info/METADATA,sha256=WX1pEAIxEfXMLrAqIKJLeJrIu2x0OrF7qG6D351odew,24214
jax-0.4.25.dist-info/RECORD,,
jax-0.4.25.dist-info/REQUESTED,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax-0.4.25.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
jax-0.4.25.dist-info/direct_url.json,sha256=AdkKe44-ZKTCc7pZhjDzRXbovIE5ZKvr8I1r7qRhaKE,63
jax-0.4.25.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=Jx81F2rX5j03zcVZCphOl-YOpQyNH4M9e1cBXbR-vFE,9610
jax/__pycache__/__init__.cpython-312.pyc,,
jax/__pycache__/ad_checkpoint.cpython-312.pyc,,
jax/__pycache__/api_util.cpython-312.pyc,,
jax/__pycache__/cloud_tpu_init.cpython-312.pyc,,
jax/__pycache__/collect_profile.cpython-312.pyc,,
jax/__pycache__/config.cpython-312.pyc,,
jax/__pycache__/core.cpython-312.pyc,,
jax/__pycache__/custom_batching.cpython-312.pyc,,
jax/__pycache__/custom_derivatives.cpython-312.pyc,,
jax/__pycache__/custom_transpose.cpython-312.pyc,,
jax/__pycache__/debug.cpython-312.pyc,,
jax/__pycache__/distributed.cpython-312.pyc,,
jax/__pycache__/dlpack.cpython-312.pyc,,
jax/__pycache__/dtypes.cpython-312.pyc,,
jax/__pycache__/errors.cpython-312.pyc,,
jax/__pycache__/flatten_util.cpython-312.pyc,,
jax/__pycache__/monitoring.cpython-312.pyc,,
jax/__pycache__/profiler.cpython-312.pyc,,
jax/__pycache__/random.cpython-312.pyc,,
jax/__pycache__/sharding.cpython-312.pyc,,
jax/__pycache__/stages.cpython-312.pyc,,
jax/__pycache__/test_util.cpython-312.pyc,,
jax/__pycache__/tree.cpython-312.pyc,,
jax/__pycache__/tree_util.cpython-312.pyc,,
jax/__pycache__/typing.cpython-312.pyc,,
jax/__pycache__/util.cpython-312.pyc,,
jax/__pycache__/version.cpython-312.pyc,,
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/__pycache__/__init__.cpython-312.pyc,,
jax/_src/__pycache__/abstract_arrays.cpython-312.pyc,,
jax/_src/__pycache__/ad_checkpoint.cpython-312.pyc,,
jax/_src/__pycache__/ad_util.cpython-312.pyc,,
jax/_src/__pycache__/api.cpython-312.pyc,,
jax/_src/__pycache__/api_util.cpython-312.pyc,,
jax/_src/__pycache__/array.cpython-312.pyc,,
jax/_src/__pycache__/basearray.cpython-312.pyc,,
jax/_src/__pycache__/cache_key.cpython-312.pyc,,
jax/_src/__pycache__/callback.cpython-312.pyc,,
jax/_src/__pycache__/checkify.cpython-312.pyc,,
jax/_src/__pycache__/cloud_tpu_init.cpython-312.pyc,,
jax/_src/__pycache__/compilation_cache.cpython-312.pyc,,
jax/_src/__pycache__/compilation_cache_interface.cpython-312.pyc,,
jax/_src/__pycache__/compiler.cpython-312.pyc,,
jax/_src/__pycache__/config.cpython-312.pyc,,
jax/_src/__pycache__/core.cpython-312.pyc,,
jax/_src/__pycache__/custom_api_util.cpython-312.pyc,,
jax/_src/__pycache__/custom_batching.cpython-312.pyc,,
jax/_src/__pycache__/custom_derivatives.cpython-312.pyc,,
jax/_src/__pycache__/custom_transpose.cpython-312.pyc,,
jax/_src/__pycache__/debugging.cpython-312.pyc,,
jax/_src/__pycache__/deprecations.cpython-312.pyc,,
jax/_src/__pycache__/dispatch.cpython-312.pyc,,
jax/_src/__pycache__/distributed.cpython-312.pyc,,
jax/_src/__pycache__/dlpack.cpython-312.pyc,,
jax/_src/__pycache__/dtypes.cpython-312.pyc,,
jax/_src/__pycache__/effects.cpython-312.pyc,,
jax/_src/__pycache__/environment_info.cpython-312.pyc,,
jax/_src/__pycache__/errors.cpython-312.pyc,,
jax/_src/__pycache__/flatten_util.cpython-312.pyc,,
jax/_src/__pycache__/gfile_cache.cpython-312.pyc,,
jax/_src/__pycache__/hardware_utils.cpython-312.pyc,,
jax/_src/__pycache__/jaxpr_util.cpython-312.pyc,,
jax/_src/__pycache__/lax_reference.cpython-312.pyc,,
jax/_src/__pycache__/layout.cpython-312.pyc,,
jax/_src/__pycache__/lazy_loader.cpython-312.pyc,,
jax/_src/__pycache__/linear_util.cpython-312.pyc,,
jax/_src/__pycache__/logging_config.cpython-312.pyc,,
jax/_src/__pycache__/maps.cpython-312.pyc,,
jax/_src/__pycache__/mesh.cpython-312.pyc,,
jax/_src/__pycache__/monitoring.cpython-312.pyc,,
jax/_src/__pycache__/op_shardings.cpython-312.pyc,,
jax/_src/__pycache__/partition_spec.cpython-312.pyc,,
jax/_src/__pycache__/path.cpython-312.pyc,,
jax/_src/__pycache__/pickle_util.cpython-312.pyc,,
jax/_src/__pycache__/pjit.cpython-312.pyc,,
jax/_src/__pycache__/pretty_printer.cpython-312.pyc,,
jax/_src/__pycache__/prng.cpython-312.pyc,,
jax/_src/__pycache__/profiler.cpython-312.pyc,,
jax/_src/__pycache__/public_test_util.cpython-312.pyc,,
jax/_src/__pycache__/random.cpython-312.pyc,,
jax/_src/__pycache__/shard_alike.cpython-312.pyc,,
jax/_src/__pycache__/sharding.cpython-312.pyc,,
jax/_src/__pycache__/sharding_impls.cpython-312.pyc,,
jax/_src/__pycache__/sharding_specs.cpython-312.pyc,,
jax/_src/__pycache__/source_info_util.cpython-312.pyc,,
jax/_src/__pycache__/stages.cpython-312.pyc,,
jax/_src/__pycache__/test_util.cpython-312.pyc,,
jax/_src/__pycache__/tpu_custom_call.cpython-312.pyc,,
jax/_src/__pycache__/traceback_util.cpython-312.pyc,,
jax/_src/__pycache__/tree.cpython-312.pyc,,
jax/_src/__pycache__/tree_util.cpython-312.pyc,,
jax/_src/__pycache__/typing.cpython-312.pyc,,
jax/_src/__pycache__/util.cpython-312.pyc,,
jax/_src/__pycache__/xla_bridge.cpython-312.pyc,,
jax/_src/abstract_arrays.py,sha256=r5HI4QdiNeLvvysAx4HoMR7MqrEzEbypFIzc8549OKQ,2680
jax/_src/ad_checkpoint.py,sha256=b7C0oGGAlqi6xiqSFHft0q6qffNUjaADv_sjY2aUxU8,35979
jax/_src/ad_util.py,sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY,3773
jax/_src/api.py,sha256=I72-6kNh-sb9j4ltnQZVD2vwulK6oH4Hf3S2rlyJG7U,132722
jax/_src/api_util.py,sha256=2H9hKAi2FhQMtw8sutVRvUpm6aCGRM3naFcpDxkyquM,27333
jax/_src/array.py,sha256=0tjdbSSzl9pxliD6mb5Bu9e67Hg3Nm9WzKD1xlvJ1eI,36670
jax/_src/basearray.py,sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A,4296
jax/_src/basearray.pyi,sha256=mf8HC7mdr5lD1N6g0MAZmW8jyqh_oDIrumizAxbPMhg,10655
jax/_src/cache_key.py,sha256=ocnBWVmx4gOtF_2gX-_7JDbU8wNHncL2f7N2NEGfwrY,9531
jax/_src/callback.py,sha256=y6GhYzA0H6PaZzVn83O-SEfKbdpTtNK4XvSUzAYx3rA,18282
jax/_src/checkify.py,sha256=JN7HbRfmbiG5a1fYHPxdvAcDV3DztGIku_gbfz4wA1E,52865
jax/_src/cloud_tpu_init.py,sha256=0IY467hAWhiTXeL4PXM49RCrci2298bSurPiQ55paXM,2371
jax/_src/clusters/__init__.py,sha256=zGB6xLPiYJsM-zWSgDaofQTdP1Ot1vPXYk-JNzYXZ6w,1161
jax/_src/clusters/__pycache__/__init__.cpython-312.pyc,,
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/ompi_cluster.cpython-312.pyc,,
jax/_src/clusters/__pycache__/slurm_cluster.cpython-312.pyc,,
jax/_src/clusters/cloud_tpu_cluster.py,sha256=o97GoH5FSbLXdtBTKYfgBbDWV-02Rd7f93VpGNMpU24,5967
jax/_src/clusters/cluster.py,sha256=8fcnWnZVV6msrljrN8UmH-JZ9-7zZvhN5GQvUIZh3AQ,4760
jax/_src/clusters/ompi_cluster.py,sha256=FVzhwuactnX3un7bMtLggl8LaiFF2QHisjqVQq5WPDg,2268
jax/_src/clusters/slurm_cluster.py,sha256=V2J-mQj6Rf_bV4Q50-yc4EjPW66CABpFbdRdjs653r0,2271
jax/_src/compilation_cache.py,sha256=45YR_4oGGUkvFm5-vDt-Vm1dHxAEb6R68_YwfhwsQoE,8842
jax/_src/compilation_cache_interface.py,sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c,863
jax/_src/compiler.py,sha256=5OKp-0CUrWmo5WEmlBfgBYINncnX9ETUyI0I_6rzbow,19753
jax/_src/config.py,sha256=2iJ1t5HPj3Dvg7_6qYnG0GWplOVzgwlE3pWxgOQgBtc,61417
jax/_src/core.py,sha256=HP5lcjwY_lmyoWy1utZZIsbmpBcJH-OJfi-Br6lgdz8,116761
jax/_src/cudnn/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/cudnn/__pycache__/__init__.cpython-312.pyc,,
jax/_src/cudnn/__pycache__/fused_attention_stablehlo.cpython-312.pyc,,
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=NVwq0GZ5rY1O9miU2t0lVWhGIxBe6PzWOpGXjN9OUiE,34959
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=x0zm1OkE1g-bBo6w_W4ul3zGnDx7ouUMTLRLz09-6ss,10502
jax/_src/custom_derivatives.py,sha256=xeY9uQL1sMTF69j5PLaKR_1vFP1Sq3pwwf-nm-OXXWo,58984
jax/_src/custom_transpose.py,sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E,8310
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
jax/_src/debugger/__pycache__/__init__.cpython-312.pyc,,
jax/_src/debugger/__pycache__/cli_debugger.cpython-312.pyc,,
jax/_src/debugger/__pycache__/colab_debugger.cpython-312.pyc,,
jax/_src/debugger/__pycache__/colab_lib.cpython-312.pyc,,
jax/_src/debugger/__pycache__/core.cpython-312.pyc,,
jax/_src/debugger/__pycache__/web_debugger.cpython-312.pyc,,
jax/_src/debugger/cli_debugger.py,sha256=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=0rEPhWjcCaiMey4sAuDivG1H5RFqqlfesAxlZZ1rlRU,8369
jax/_src/debugger/web_debugger.py,sha256=WJzG5FubQxA9EiuyXzHUb_v2_2GqvovyQiy6EaBomY8,3282
jax/_src/debugging.py,sha256=oKvllSijZ6zbDpjy7d424qWEfoxEsgTIlg4xGlvR6tU,24309
jax/_src/deprecations.py,sha256=pp87x88apDFSAYhBF73A-pXx1nr-dII_Zp_1cKQigjc,2518
jax/_src/dispatch.py,sha256=fmEKVufKGK6TIQXPPNmu5qGFbaJaJatB2bKDxitGXCM,17551
jax/_src/distributed.py,sha256=Jt00nRrt8jlr9f2AgCtyN32_l9On3yciG1LA9lE6vbE,7670
jax/_src/dlpack.py,sha256=1kKc5gNh9AuFmf8PHDonjUULJdyrXWcGC3KdoowHcU8,4948
jax/_src/dtypes.py,sha256=VdsHAV9AB9jQx10Ah54iQlqvZ4rRxIUrP4gwEKm8ibA,28323
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=9Gwd-KeYyMlORpCWQR6a8ruZgnJkom8qT_1RMsjBacw,2010
jax/_src/errors.py,sha256=jJvt8vwEuR8Ie_hdqjBqSwuy5zD74nkIXvbT_ttPEso,23777
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-312.pyc,,
jax/_src/extend/__pycache__/random.cpython-312.pyc,,
jax/_src/extend/random.py,sha256=bAtdkmxevVfY9JfcwrWX6Ah8RXMoNelXg2I644ePHCM,1276
jax/_src/flatten_util.py,sha256=-9ozNDsKw2QRDf5J4UjIlhTPOv3pGloUMKffS9aGteY,3554
jax/_src/gfile_cache.py,sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY,1923
jax/_src/hardware_utils.py,sha256=s0UuPPWP6eB7oqQ8o02dL3RqzQTYTsMKcNCukGoO5zI,1714
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/__pycache__/__init__.cpython-312.pyc,,
jax/_src/image/__pycache__/scale.cpython-312.pyc,,
jax/_src/image/scale.py,sha256=sWBhsUcujSz-pDPJe-gqiK-byJk_Dd2gISoT7in8dXg,13758
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-312.pyc,,
jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-312.pyc,,
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=VBDNP08YLWvuUdt1HBVx8LUcBi17jXScwuTQejB7wMk,15299
jax/_src/internal_test_util/lax_test_util.py,sha256=LBych5KKJ1fYfxHTk9YyyiOuVGljYYyngz1Ckp__BNY,12148
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-312.pyc,,
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-312.pyc,,
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/internal_test_util/test_harnesses.py,sha256=ZJBbtnWCQphK67bXX5dCus2XZMURqyAIxR-JoGJ6FtM,124694
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/__pycache__/__init__.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/ad.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/batching.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/mlir.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/pxla.cpython-312.pyc,,
jax/_src/interpreters/__pycache__/xla.cpython-312.pyc,,
jax/_src/interpreters/ad.py,sha256=im6DEd6FVUASLwkdWWZaPvWvi4t8Ddc_GkY62oFSPJ4,33185
jax/_src/interpreters/batching.py,sha256=NJDTJxEy2T1BVJVXzyyyjMWcuua7lmTUou5Og2V6EjI,50430
jax/_src/interpreters/mlir.py,sha256=393taKa_auaQPX6EByf6VKIyLzPwKOd3ZtX-a3T3RTs,116126
jax/_src/interpreters/partial_eval.py,sha256=z2LteupcWHadhoJefK-iEIlv5h1-xuBrCV7tiA1HzXM,125154
jax/_src/interpreters/pxla.py,sha256=6ATZklhFPPaexN7KeSLFygzfApA5GfICQSJ9AlUq-hs,130511
jax/_src/interpreters/xla.py,sha256=Q6VGpfpqcmWU_FiEF5MCH0wq964jU668BS9p_L7xlcY,12067
jax/_src/jaxpr_util.py,sha256=4mB7jfzh0IhgqiqnStUnL4_msTyOljxgzG56pw1-x3s,6428
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lax/__pycache__/ann.cpython-312.pyc,,
jax/_src/lax/__pycache__/convolution.cpython-312.pyc,,
jax/_src/lax/__pycache__/eigh.cpython-312.pyc,,
jax/_src/lax/__pycache__/fft.cpython-312.pyc,,
jax/_src/lax/__pycache__/lax.cpython-312.pyc,,
jax/_src/lax/__pycache__/linalg.cpython-312.pyc,,
jax/_src/lax/__pycache__/other.cpython-312.pyc,,
jax/_src/lax/__pycache__/parallel.cpython-312.pyc,,
jax/_src/lax/__pycache__/qdwh.cpython-312.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-312.pyc,,
jax/_src/lax/__pycache__/special.cpython-312.pyc,,
jax/_src/lax/__pycache__/stack.cpython-312.pyc,,
jax/_src/lax/__pycache__/svd.cpython-312.pyc,,
jax/_src/lax/__pycache__/utils.cpython-312.pyc,,
jax/_src/lax/__pycache__/windowed_reductions.cpython-312.pyc,,
jax/_src/lax/ann.py,sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M,17623
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/common.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/for_loop.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/loops.cpython-312.pyc,,
jax/_src/lax/control_flow/__pycache__/solves.cpython-312.pyc,,
jax/_src/lax/control_flow/common.py,sha256=jE7vo0resS9tHdg9F9nleNB9Gwcv8X3VUzpMHNo0OxA,10976
jax/_src/lax/control_flow/conditionals.py,sha256=vAUnZQ8vOBNUpu9Fd91Tj2HudAXTpEzuRGhW1t-_iYo,43008
jax/_src/lax/control_flow/for_loop.py,sha256=NA8wvsH48E8k7ey-pcQa0pMi9CbwVLfZbRaCcireSD0,37135
jax/_src/lax/control_flow/loops.py,sha256=oeTy9BzDHuD-oVIJc9oz5lC0Yd9ZAWNqFaCBUvY578w,105071
jax/_src/lax/control_flow/solves.py,sha256=MtOEjg9pa0NTCfVkPaKlfEobIViNhOvKfvyv0CsR1g4,18874
jax/_src/lax/convolution.py,sha256=SNNtSYshLLvMAMwsrTfnDPGO-deZeqR-Y_Nu3rebJeM,44715
jax/_src/lax/eigh.py,sha256=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=izHxbi-S686onnDHQC3WRuu9PbKhDhYkZUtZPx6LxFI,10151
jax/_src/lax/lax.py,sha256=ZcsjQyksb4qygcEOLyuN9ohQHBEOLPVNqr8ZCjgLr30,208766
jax/_src/lax/linalg.py,sha256=pq380CYVFKEUJNLQxJTQYnPXP3qm_dkdRn841ucM0kQ,91611
jax/_src/lax/other.py,sha256=8o-YBA54msT4HNAsEmDOi9Wctha5ad1rd4fjSkyt5xM,10547
jax/_src/lax/parallel.py,sha256=hnyAh7LO7YZ921L_KTR415CO32RG3EkmutdwtbQx3UE,71618
jax/_src/lax/qdwh.py,sha256=f3QJwPMDDO_qC-VFaKrzSXUJ0njCSwSEjICsA5FckxU,8028
jax/_src/lax/slicing.py,sha256=glCJwZAhftQCVnXybrH3KhxGogIqFVpVwW0glM771mg,111664
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=vSEwqDksgYifZlIxtqjDodJVul5b_C819162EZZ5zcE,11705
jax/_src/lax/utils.py,sha256=KHbNG7zWt4w8vrSxSJv5iKubesvLdZlX9HVTXLyr5Iw,4968
jax/_src/lax/windowed_reductions.py,sha256=StHNyfk40hMA9-SxVsI2_yC0w671r5tvQcQbua3A5Q8,40180
jax/_src/lax_reference.py,sha256=hZ9Pm3Dx5iJa5yWm0oSvkrAbyuaEartugDi8uIuW454,17330
jax/_src/layout.py,sha256=ga5-18Pz5gAV2ktKpJJzEPlfD_LRDMvuGtfZSloqNi4,1585
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/lib/__init__.py,sha256=HSyqGdFIAIFHVBk45EIzsqK-M5HHqza5-5ubmXqmipY,5476
jax/_src/lib/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/__pycache__/triton.cpython-312.pyc,,
jax/_src/lib/mlir/__init__.py,sha256=vv85BXVE6lq8QZTUZtOIFPrwcoIWROkeiAqVa5a8SB0,885
jax/_src/lib/mlir/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/mlir/dialects/__init__.py,sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs,1195
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
jax/_src/lib/triton.py,sha256=HbqoIneu52En_DHWH6Mph9rGtj324CRCXoT0SZA3u44,1336
jax/_src/linear_util.py,sha256=9s_-v7r432MAQc6r-juktBEyIu2-tOYh0j9orLJRLs4,13946
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=Pi0W59D3mp9NNidk_2YF6nkoV0CdizcB8c5L-sbzH9M,86263
jax/_src/mesh.py,sha256=I2YfKPwdLK9NRunSqs6ImaAJyXS3JGvYtOFunO_bzpk,11314
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/__pycache__/__init__.cpython-312.pyc,,
jax/_src/nn/__pycache__/functions.cpython-312.pyc,,
jax/_src/nn/__pycache__/initializers.cpython-312.pyc,,
jax/_src/nn/functions.py,sha256=EslJHfwFEjv4vnYy06NclC8n0PVYApxkOFJODJvy4NA,18542
jax/_src/nn/initializers.py,sha256=WExqHajxKfYghowufGfYaoPH_ZLRRpDrKy1xDGV-77Q,25541
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-312.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-312.pyc,,
jax/_src/numpy/__pycache__/index_tricks.cpython-312.pyc,,
jax/_src/numpy/__pycache__/lax_numpy.cpython-312.pyc,,
jax/_src/numpy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/numpy/__pycache__/polynomial.cpython-312.pyc,,
jax/_src/numpy/__pycache__/reductions.cpython-312.pyc,,
jax/_src/numpy/__pycache__/setops.cpython-312.pyc,,
jax/_src/numpy/__pycache__/ufunc_api.cpython-312.pyc,,
jax/_src/numpy/__pycache__/ufuncs.cpython-312.pyc,,
jax/_src/numpy/__pycache__/util.cpython-312.pyc,,
jax/_src/numpy/__pycache__/vectorize.cpython-312.pyc,,
jax/_src/numpy/array_methods.py,sha256=ISRBbYRa3HcXLjMD5egv5Ie7khsn5224QRBw-E57Gok,32582
jax/_src/numpy/fft.py,sha256=rVEbkoRIJ_WiGPnrsl4bztstWiZ50acmrFPvl9pde58,11373
jax/_src/numpy/index_tricks.py,sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g,9888
jax/_src/numpy/lax_numpy.py,sha256=MgE79d8d07Tp2-yhdw653iuZ3E9EOn6OvlOVDKddjaQ,219444
jax/_src/numpy/linalg.py,sha256=rtiU5yILGJD_xcQw0Kk_xTo-XEW99Wnt0gET4UQGCpg,31450
jax/_src/numpy/polynomial.py,sha256=WoFDVkjxgQFDkaLOVV1vlBCBfCqNI3fZPors8Q7pIdI,12155
jax/_src/numpy/reductions.py,sha256=AxVFuvVdYOCiy2dtmUZZKWvpQJKZnu-eNaQ8Oa8gzXk,41571
jax/_src/numpy/setops.py,sha256=pdYBHaTrUHOhQtNIvH2VXy3kk2Vk39VmpM2BdUWwl0w,15407
jax/_src/numpy/ufunc_api.py,sha256=RbGKNxN0MhSzkQPQOYbEeJS7PaaEwMtyDrS4MqbQPzg,15264
jax/_src/numpy/ufuncs.py,sha256=S8ehVJdJ9bGvnq2RcB0GIq-pU8ADK5EwsnCQv2waaKQ,28442
jax/_src/numpy/util.py,sha256=dsr5X1X8dSG3wy2tmzv-ZXY7_iHnSB8QT2yFnWUBEgg,18997
jax/_src/numpy/vectorize.py,sha256=H-OiIbDQvKmjpOJlO9L7cpQOYn8cOJQod_7C6OHOrB8,13021
jax/_src/op_shardings.py,sha256=mb_s-ThVgw-rgM_ZZVH1P6Lpgl1RtMauk__Qn2g9XqM,3895
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/__pycache__/__init__.cpython-312.pyc,,
jax/_src/ops/__pycache__/scatter.cpython-312.pyc,,
jax/_src/ops/__pycache__/special.cpython-312.pyc,,
jax/_src/ops/scatter.py,sha256=kZ-h0mJzy57grnDe9PFZdv4TN6u6_OxRauIRtCVZ214,17593
jax/_src/ops/special.py,sha256=nEM6g8Ha6-LY9Ai7gZaMAH3LkT6Xj362V17yKVoP8QA,3884
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/__pycache__/pallas_call.cpython-312.pyc,,
jax/_src/pallas/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/__pycache__/utils.cpython-312.pyc,,
jax/_src/pallas/core.py,sha256=fJYY46uKMqa__bwaXGOFh0J2VpvT8tVAArDvlQr-m4E,12353
jax/_src/pallas/mosaic/__init__.py,sha256=Cp0KrWFmKWKANvW99kLDl3jNvJmerYbBLJllFCzK_aU,2489
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/core.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/kernel_regeneration_util.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/pipeline.cpython-312.pyc,,
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-312.pyc,,
jax/_src/pallas/mosaic/core.py,sha256=4r0gUkeyD4Gy4H5cdaKIkoeI4sOefhUibW6-2Ynn-Zs,7677
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=MPUBNSd8gBTxjihQqEJAR4hkXJ_I1PTJPpImoN7E58M,72711
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=kudmiRRezV7AaXo4TyWfurs6pUdtWxCXYaumjSC0gf0,4136
jax/_src/pallas/mosaic/pipeline.py,sha256=YuprzcBN3xUgbm1-TF_5fSavKe0GMy5_M1lOwwyxUU4,39945
jax/_src/pallas/mosaic/primitives.py,sha256=J6qUdR4DQvQB-9VfCBqxZVoxm5SKIcXL8R5OOtZVPDU,18126
jax/_src/pallas/pallas_call.py,sha256=GUCgWADA9K5a-Av5DxL0rqllmcoXzX8XmYj35wfExRc,24605
jax/_src/pallas/primitives.py,sha256=yXWqNmRYQ7neNvSQpUfcflW0s-S2klyr4c8J65twV-I,15805
jax/_src/pallas/triton/__init__.py,sha256=ZUafgPFMjksaB6HLJpMxwO_dOpEm8j9pWItrGpgeR_g,1142
jax/_src/pallas/triton/__pycache__/__init__.cpython-312.pyc,,
jax/_src/pallas/triton/__pycache__/lowering.cpython-312.pyc,,
jax/_src/pallas/triton/lowering.py,sha256=Pj808VrHJeQfyPjdiM3R7rWAYlEARgPuKcHCUAOocSY,84563
jax/_src/pallas/utils.py,sha256=lBVpAnj6NNE6wWocwiNtfrqYEqdNwFAvAkkf4lGG_wQ,5864
jax/_src/partition_spec.py,sha256=apayEnjD-HJ0NykjgoUquNhQ9W7PwsEKZOPqGCx6GcA,1688
jax/_src/path.py,sha256=Zd6Q4jJEK8m4MOVUsiT2ALoUDHvoZjP37SZFYJzacVI,1121
jax/_src/pickle_util.py,sha256=UUNYUdIIcN9G8NW5EwRiVuxJo7zP8-p4v66rwq7jt9w,2302
jax/_src/pjit.py,sha256=LMvlzNGTmOjf7KFKHl-ownUmWoNEKG0XSaKGU1RRIsU,102697
jax/_src/pretty_printer.py,sha256=gy89qxIgK7ZG7P9cglSmzrtFFjPS1uXWdcUbUjH1iFo,13077
jax/_src/prng.py,sha256=Qf2hldvrsqWbAv40tv-26m130IujVX0gk5F6tKIS92c,48062
jax/_src/profiler.py,sha256=lsRQTKMbCS5uX4ka-_R6Zi10QAKfJurZ2jca5_qMU64,14507
jax/_src/public_test_util.py,sha256=Z4p1XzuocBDSXwI2HBFkssqPrIGpO3R3VETa5gyi2bY,10451
jax/_src/random.py,sha256=Y-WjC9C_hV_gVyfSk0ZC5MOWbRVMxz3LV-BInbWlMO8,99932
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/__pycache__/fft.cpython-312.pyc,,
jax/_src/scipy/__pycache__/integrate.cpython-312.pyc,,
jax/_src/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/scipy/__pycache__/ndimage.cpython-312.pyc,,
jax/_src/scipy/__pycache__/signal.cpython-312.pyc,,
jax/_src/scipy/__pycache__/special.cpython-312.pyc,,
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
jax/_src/scipy/cluster/vq.py,sha256=lmudnV7wOY8eN0CUT2a7GxA3mhwRGmjPa7cVF2PNB98,1779
jax/_src/scipy/fft.py,sha256=mcKjqvp63yDGDDZTxJjzbkcUAVu9j6piXYf98zV8ffo,6248
jax/_src/scipy/integrate.py,sha256=w3y6b9iwVp7ozcA1GGErSMhBa-DR4QsWPY_YYWJcA2I,1565
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/interpolate/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/linalg.py,sha256=nsFWrobIuzBfLSFWOVUjd9lHt_-3YRo8gA0MYa6xaiM,41644
jax/_src/scipy/ndimage.py,sha256=viYzxaVZVP3pXG1yBKMSG5wssJUKYDerBxUcXfCT2ds,5362
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/line_search.cpython-312.pyc,,
jax/_src/scipy/optimize/__pycache__/minimize.cpython-312.pyc,,
jax/_src/scipy/optimize/_lbfgs.py,sha256=Io1xjWLc90YFTCPiGMz-fP9_FMWlpNuymTIVIKJXjZQ,7790
jax/_src/scipy/optimize/bfgs.py,sha256=fJ5GdaoI79pGHpFlX0USGF-1ooGa_-bKjMVvAaqqu1U,5582
jax/_src/scipy/optimize/line_search.py,sha256=JF16fsbL1kGJnVRRXumssJzXmsV7cQGeDFInmxYnlG0,13229
jax/_src/scipy/optimize/minimize.py,sha256=miFHZRaf8KpiDd4XsEUAKHUGrGodKYCOiKAITye-BIY,4979
jax/_src/scipy/signal.py,sha256=IuhZdas7Uv-mk9u_WR3bW4aJQn_jYakhr0bwGtvEwBE,27552
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
jax/_src/scipy/sparse/linalg.py,sha256=VCjQWXeUaf_9xqTCTIiGTVkMjaa9IvEwKgLrhinavEw,27326
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
jax/_src/scipy/spatial/transform.py,sha256=VGTcZRDye7xJBLFPiIJM1DJdJTDiIPGytz7HFR0o4QU,15167
jax/_src/scipy/special.py,sha256=-Hho_ccefglXuNDKh9OTNfVmY0hl-RuDWFrKQZEoKoE,63694
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/_core.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/beta.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/binom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/expon.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/geom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/kde.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/norm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/t.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
jax/_src/scipy/stats/_core.py,sha256=FLIRKSf8SY-lKXwscEFVn12ALW3iQtu4QDWH6HhqQMc,5725
jax/_src/scipy/stats/bernoulli.py,sha256=9PntVJ6XZC9btsq51pzvgi1Ndpg7IGgAZzIglEOSyL8,2270
jax/_src/scipy/stats/beta.py,sha256=YxIfH_-b5vXkqmOiC5n57J3kZMmkkf_aYEawfwTtHLw,2968
jax/_src/scipy/stats/betabinom.py,sha256=vusiYROP2Vk2PNZs0E1BCiwEpFaIlRC1hZ02v8-Ga-E,2005
jax/_src/scipy/stats/binom.py,sha256=T4wO99oUze6-nuv5Vw6_crdy1jury62cA8k29RYN9CM,1671
jax/_src/scipy/stats/cauchy.py,sha256=ejSOHHOc0MJjsRFTJstpP4l5pTYWFWsRJdczaYlgbHw,3142
jax/_src/scipy/stats/chi2.py,sha256=Q-vgl-gttIpxzlU59fSFIR-08fZkl4cT7P1MPK7cKzw,2952
jax/_src/scipy/stats/dirichlet.py,sha256=v_KIk7RegcQYeNOD5memO7kLmj043NzmggMPVf73WSw,2148
jax/_src/scipy/stats/expon.py,sha256=yVjLkyb0z9tQXtltE0Qk6WHjeQlW7IW0G78ypijU7vM,1326
jax/_src/scipy/stats/gamma.py,sha256=gOi_AqGcD-d1nys898_UFEJF8dEaKzN2F6KzcPHYxjs,2567
jax/_src/scipy/stats/gennorm.py,sha256=ESjS4pr1gpY-dKLfkBLzoBiuVmqx3xn7dZj5hHqvKCQ,1311
jax/_src/scipy/stats/geom.py,sha256=PC6utDsp2J-AAm9ZEt5HOe-C94Eqe6DiNT_HAjtxexE,1394
jax/_src/scipy/stats/kde.py,sha256=r-Ua7AKOF755cE9OPFvtodfXWZ0jo5U3DjpUoaKks3g,9876
jax/_src/scipy/stats/laplace.py,sha256=uaazCUEB7qdWY48bxV2sP0msvWb9dpAxjkNnhqQWzEU,1803
jax/_src/scipy/stats/logistic.py,sha256=3HIoYI-iN2UVGLc_UHRpaNw9a32VfPsoUObjCJWOjS0,2434
jax/_src/scipy/stats/multinomial.py,sha256=Vgndccd5iNj1sV-2q32oHOzOeGo7LNLBq52ICRwq0qQ,1687
jax/_src/scipy/stats/multivariate_normal.py,sha256=T86VsQGq7585Bte2nK9ATHCWKkkUuLk4aiuDqmw-ssU,2231
jax/_src/scipy/stats/nbinom.py,sha256=GTXNvC69AUsqgnnT2ySJcdPe0Eq1bjNdrTF5BPP8rEA,1710
jax/_src/scipy/stats/norm.py,sha256=CXkWEgZmVayncIpL2Vi5rLNFVxqlz215qedtEQm0oPI,2899
jax/_src/scipy/stats/pareto.py,sha256=ahSAdtFCCf10Idfjsb0nvecNzgVvcCm8X5JxJnRAEOk,1509
jax/_src/scipy/stats/poisson.py,sha256=Xikr-qVqCQvD8BM12XQN5QquFgI00gsriu7oCfQnuUM,1700
jax/_src/scipy/stats/t.py,sha256=vx7uvNW6xzMTCOHl6423T8ho6E4oMyRNdqIx-ODAUiE,1834
jax/_src/scipy/stats/truncnorm.py,sha256=UiA72yshSWEQFQJkfxgUO_jHQt4P7A7ZT6z56OxzKRs,3993
jax/_src/scipy/stats/uniform.py,sha256=ijGxgJabQi95MyiSex0aaBOJnBTUAnJ4KF-lhJcBl3w,2122
jax/_src/scipy/stats/vonmises.py,sha256=iCC8-yFGHGH6ky4XD_2ISRt-cjRIh2acHKYBwIPFzCQ,1284
jax/_src/scipy/stats/wrapcauchy.py,sha256=-k6Wt4lROA5sK9Juyc78a8YO3imF0HRx0iYLW6UVHzw,1422
jax/_src/shard_alike.py,sha256=sdSYS0jagWfqS8pkdLEqQg6tJBr2QoU92dEgG0Hv4nM,3978
jax/_src/sharding.py,sha256=fB6sGov0DDVy1eCAJ3MrwypVzQ7Rho85-0QNXv6kvO0,5070
jax/_src/sharding_impls.py,sha256=p1_96b1D-frA-DAdHOYrLUWk4pA-NV5wYDG-8YsZkpA,51271
jax/_src/sharding_specs.py,sha256=9s9WfSdGriLgmreTp_2xlqQtkf1TmSxCWRmtT-q--1U,9207
jax/_src/source_info_util.py,sha256=AjLwjmZXBhYFM13j4kMBWKK0Ch1A9Q07aKwyt3Wcefc,9912
jax/_src/stages.py,sha256=2HsY-TR5r0N-pgTZJn0RmWU7ajBSbArNbi2YRx4R3Ko,26020
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/__pycache__/__init__.cpython-312.pyc,,
jax/_src/state/__pycache__/discharge.cpython-312.pyc,,
jax/_src/state/__pycache__/indexing.cpython-312.pyc,,
jax/_src/state/__pycache__/primitives.cpython-312.pyc,,
jax/_src/state/__pycache__/types.cpython-312.pyc,,
jax/_src/state/__pycache__/utils.cpython-312.pyc,,
jax/_src/state/discharge.py,sha256=sDGAujFXXED-saBzNp3_vNUVqTYaXmsPL9_-ntperzQ,36780
jax/_src/state/indexing.py,sha256=KOrsVTDDLUVFz90-iv8tYycvgP1t3w_zB2FpFaGkMxM,7537
jax/_src/state/primitives.py,sha256=v6urrjQnuqi8RQUpVHgLkIhnwaYR4VgW28MAFFwqZtE,23468
jax/_src/state/types.py,sha256=1x6VDqFra46u-rVO3qO5AElSSxaY9bcTqNjJyXqvGsg,6693
jax/_src/state/utils.py,sha256=AwdUqrfZto_M6x2L5L4KCoBARMsjhYwfb3jvjmawjsU,2411
jax/_src/test_util.py,sha256=BsWtmXI9ffTbg3qMaWOZNBr1DXuyL3OLi_Bqncjz3KU,48817
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/__pycache__/__init__.cpython-312.pyc,,
jax/_src/third_party/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/third_party/numpy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/third_party/numpy/linalg.py,sha256=HDpnzipYvKzz7-xxu_2Ft56Ty8czVtpxhmNB2mhIiVU,6231
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/betaln.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-312.pyc,,
jax/_src/third_party/scipy/betaln.py,sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=A_SKMrHHpf6AnbS1YgGxg3RPbNQo9FWgOGLLkslwHQo,5881
jax/_src/third_party/scipy/linalg.py,sha256=PyQc2ABPlYIjbQF8pH8729L7DIHNAiN6JyjyjYOjZao,2698
jax/_src/third_party/scipy/signal_helper.py,sha256=VqLSblEWCItAViXc6lIXRbF-FT83l9305aHIZP9B788,3081
jax/_src/tpu_custom_call.py,sha256=AO7tcmmxid2jUDYufOXd2On-I93hKp2tgNFoSRTTbpk,12888
jax/_src/traceback_util.py,sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA,9060
jax/_src/tree.py,sha256=qf_oRe7McgufUBFD7LtV9KG8XL_fJB-kn7SZDUaSka4,3492
jax/_src/tree_util.py,sha256=tlkU1fjhPxKJn5GRknSlWxESX0x3aleafGIgziLuex8,39131
jax/_src/typing.py,sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI,2892
jax/_src/util.py,sha256=lTqRA4TJTJ9zyg-FYD5FgJuY0dMoeI1DMj-k7yv-Rw8,19077
jax/_src/xla_bridge.py,sha256=bXxb0dpq_oSJE08RjmkAOMeKEEhdo-hanml-pvBPGbE,36594
jax/ad_checkpoint.py,sha256=N3-hWts5BdssIjnoNm1NnGzACdpQmtjol7fd_priwKo,802
jax/api_util.py,sha256=JnZFSuqh5WBglJwerrWxcIDslBaXYZQrCv6j0jPfu7M,769
jax/cloud_tpu_init.py,sha256=nNdPdZZffdRGb38rEx1YOLXNpymSNt05ZVWSm8lhZ8c,633
jax/collect_profile.py,sha256=exKxJOhMrkMnXcseOR2VwV4m3ZNw00DsMmXAR-9Qr1w,4721
jax/config.py,sha256=xx0PZUvoNtaxkxXQImoymJlY6I7J1oSbnkgCigp1nMg,1040
jax/core.py,sha256=POA3yaxLJnbb_Oj699lSD6epH0S20VO3tSXFuqQZkf4,7340
jax/custom_batching.py,sha256=zQmkJIxOxxgAqEgby1q-vQWHYKzyqp1__n8Hqs04OV4,657
jax/custom_derivatives.py,sha256=FWBXGy0zp70zI46ErbGucn23R6axD_0BuvZoeZJaW0U,1341
jax/custom_transpose.py,sha256=BmKD3o61GFyTE68GoNX_3Z8NBJl5HItzBCi8RehA9ks,644
jax/debug.py,sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o,1083
jax/distributed.py,sha256=_-zv-zCv4BirRPOM8Uy6zXOCp-V4ej4zNDS2YTiCRSw,638
jax/dlpack.py,sha256=Xq3mQbR6R_ak_CD2itRttfJx11cVaXtdl_xLQL3clfY,653
jax/dtypes.py,sha256=Mp2QLfj2hwLdfJHz8Kpzg2FiX85hJMyBLo3rapL3iMo,1146
jax/errors.py,sha256=KvLylW0zzk5D6q1ODpe6hWjLXtFYV6WmkjVvxPaCZdY,1238
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/__pycache__/__init__.cpython-312.pyc,,
jax/example_libraries/__pycache__/optimizers.cpython-312.pyc,,
jax/example_libraries/__pycache__/stax.cpython-312.pyc,,
jax/example_libraries/optimizers.py,sha256=224Ba6KPYLs3zlF3g7dlubbQkIkvoko-WGqDx4svBzE,20559
jax/example_libraries/stax.py,sha256=cMt8-39Q-o-tWwquR00RAlbbiWaitHLK4P03Xj0Wa3M,13849
jax/experimental/__init__.py,sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4,877
jax/experimental/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/__pycache__/attrs.cpython-312.pyc,,
jax/experimental/__pycache__/checkify.cpython-312.pyc,,
jax/experimental/__pycache__/custom_partitioning.cpython-312.pyc,,
jax/experimental/__pycache__/host_callback.cpython-312.pyc,,
jax/experimental/__pycache__/jet.cpython-312.pyc,,
jax/experimental/__pycache__/layout.cpython-312.pyc,,
jax/experimental/__pycache__/maps.cpython-312.pyc,,
jax/experimental/__pycache__/mesh_utils.cpython-312.pyc,,
jax/experimental/__pycache__/multihost_utils.cpython-312.pyc,,
jax/experimental/__pycache__/ode.cpython-312.pyc,,
jax/experimental/__pycache__/pjit.cpython-312.pyc,,
jax/experimental/__pycache__/profiler.cpython-312.pyc,,
jax/experimental/__pycache__/rnn.cpython-312.pyc,,
jax/experimental/__pycache__/serialize_executable.cpython-312.pyc,,
jax/experimental/__pycache__/shard_alike.cpython-312.pyc,,
jax/experimental/__pycache__/shard_map.cpython-312.pyc,,
jax/experimental/__pycache__/topologies.cpython-312.pyc,,
jax/experimental/__pycache__/x64_context.cpython-312.pyc,,
jax/experimental/array_api/__init__.py,sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA,5364
jax/experimental/array_api/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_array_methods.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_constants.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_creation_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_data_type_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_dtypes.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_elementwise_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_fft_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_indexing_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_linear_algebra_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_manipulation_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_searching_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_set_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_sorting_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_statistical_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_utility_functions.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/_version.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/fft.cpython-312.pyc,,
jax/experimental/array_api/__pycache__/linalg.cpython-312.pyc,,
jax/experimental/array_api/_array_methods.py,sha256=6rP6HnjVF1CBdSpwf0cEkj1wLL4rbUTKkiuIBaWueb4,1759
jax/experimental/array_api/_constants.py,sha256=Q8eoyHE5GDoeBUgbur2mF3_RzUW6OO1PhzneUzmnH_c,669
jax/experimental/array_api/_creation_functions.py,sha256=RdV4316PQaXg5sf4wVrWWw_JQ0rf6akXgYj26XC0Kdk,2496
jax/experimental/array_api/_data_type_functions.py,sha256=v3ZQsTh_9qLqdA6zKMGpfbYhRn_OVCueGM3YwqNXtCs,5391
jax/experimental/array_api/_dtypes.py,sha256=4itd8mk4z4SpohYVHfELDpGQF87fu2ScT61nBD3tjG4,968
jax/experimental/array_api/_elementwise_functions.py,sha256=hPsPNdyFwEg8UGjz8xJnLqDlH_vrLT2_44PIf9HWFqw,14835
jax/experimental/array_api/_fft_functions.py,sha256=okg_ZQy62JTj2_CVfOkvrwn6XyCBWwX7ScBdLDnVWNg,3096
jax/experimental/array_api/_indexing_functions.py,sha256=gemN8vpsGfjy1GDilIQbtes-NHIvE9A0x18mqxZR5mk,675
jax/experimental/array_api/_linear_algebra_functions.py,sha256=mdCDSzLE7XMpn73zOe1uq1yyQ9JKrBo72iygziG-ZQA,4775
jax/experimental/array_api/_manipulation_functions.py,sha256=47LbnAVENulqjB4axQlYJlOpNtjD5voEc2VNRCQMAf0,2821
jax/experimental/array_api/_searching_functions.py,sha256=VhWWD3rH1lQi4EfoGzKqC74bnNmCB4pmByk0WqTG3ek,1483
jax/experimental/array_api/_set_functions.py,sha256=Kv-4PcMtsszLMEgG_qK84kt-2qJGai0KSa2ArPr22W4,1375
jax/experimental/array_api/_sorting_functions.py,sha256=XMYMwfgPC8AMC3NIlhuBGe3hGx22WqTmSn65CIegKzA,1118
jax/experimental/array_api/_statistical_functions.py,sha256=0xcB70zRpzfhTqOZyKEkfvFmXYSB8gBQPInuX_ai5T4,2003
jax/experimental/array_api/_utility_functions.py,sha256=Nx0zBTWs_uXSL65mo3rqnoQHAbFatpfkQCsv4UsFCuQ,975
jax/experimental/array_api/_version.py,sha256=fVbhvRAhSrwJXVDwqRQEt5vZWd-PGzRQkxJ1zmfiMFA,616
jax/experimental/array_api/fft.py,sha256=1N17z4OpG086sZne714AEJrfmZ3OnagMyMRYcLQ48aA,906
jax/experimental/array_api/linalg.py,sha256=OEDELaMerquTrn2FSXDImzsrawzBUdvWXG-da3MkixM,1153
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/serialization.cpython-312.pyc,,
jax/experimental/array_serialization/__pycache__/serialization_test.cpython-312.pyc,,
jax/experimental/array_serialization/serialization.py,sha256=GrZsxgsgsvGqxafNUxgWaszMaWttdN0bybmng-dkQI8,21772
jax/experimental/array_serialization/serialization_test.py,sha256=uhoAlXU_T81ILGqDbL8Uv3LvAnOriZjbaaIbWPBXfns,13018
jax/experimental/attrs.py,sha256=HxHFGFOI7FJ__1VtoRiWNa4vQ9FbES7ZlNfkGu6kw-I,9126
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-312.pyc,,
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/custom_partitioning.py,sha256=kw69QWCI96tl9TxrhLNWgHZnW1m0yHY1-LRs7ZMyCC8,22294
jax/experimental/export/__init__.py,sha256=dk1YZG7FSmrB47imONOWDKSEsX26XyyjI5WiHVULU8k,1268
jax/experimental/export/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/export/__pycache__/_export.cpython-312.pyc,,
jax/experimental/export/__pycache__/_serialization.cpython-312.pyc,,
jax/experimental/export/__pycache__/_shape_poly.cpython-312.pyc,,
jax/experimental/export/__pycache__/_shape_poly_decision.cpython-312.pyc,,
jax/experimental/export/__pycache__/serialization_generated.cpython-312.pyc,,
jax/experimental/export/_export.py,sha256=MBw7TS2dUelsfk-VyiDw1MPaB9Zl_hTdArNrVRhjt5g,58835
jax/experimental/export/_serialization.py,sha256=f3Bu0wxMA0DvJ8Fw1pMyd7W_xi4k5P_Npv9k3HXddR4,17212
jax/experimental/export/_shape_poly.py,sha256=Uui3Jv_b_Vjxm2_kf9dqiOZrh019YTznSoeRNTaPUvE,85388
jax/experimental/export/_shape_poly_decision.py,sha256=x9AADPHzkyQSjJEtxbIX0rOsWBqT2T2_TJ1oXsgTmN0,20711
jax/experimental/export/serialization_generated.py,sha256=wc5JKltgI-SJx6g361gxpfY4OK2DvsaVARhrm5neJzU,26195
jax/experimental/host_callback.py,sha256=aYr_TL1MUYrN2xCF5Ymi80fZbzekdmm42CmLrXulUTQ,79254
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/jax2tf/__pycache__/call_tf.cpython-312.pyc,,
jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-312.pyc,,
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-312.pyc,,
jax/experimental/jax2tf/call_tf.py,sha256=1HJWg-KnRYp_8lIHjJWq6tr-Tt9F02hkADUnB7FaauQ,27319
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-312.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-312.pyc,,
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is,2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=1FBpLLwCpm9v4udaL0tEN3umiqF4a1lXSr8k483R9Eg,1616
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=LftkJ-7__c0JpPxUWdx9uv3QiTGbOKRXzLhc1ixkev8,11520
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=UdSLpLnb2zABl5MWLZ7JRfGinscpWMSwXHb6a4rtFxA,7096
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=f4BcJt5KvsNvQrZsKhV4_J6QrxHkppRgu792dy9E2S0,8051
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-312.pyc,,
jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-312.pyc,,
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8,4952
jax/experimental/jax2tf/impl_no_xla.py,sha256=SIvqeN30ld0PqoGduuUC3JwMxvCPvOsqUZnWxF_M9W8,54476
jax/experimental/jax2tf/jax2tf.py,sha256=jNIKOn6igz1_hRHy8qHApYE-9sqCaVujfIiA0ck8wnE,142986
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/converters.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-312.pyc,,
jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-312.pyc,,
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=9FGx-2zxn89IzuQ6OYhaqA0zII0OQ6L6jktQvHzSqY0,5412
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=q8NGxAIyYJqLJX-9DBffoiPKIaoKJTOptOB1D-4Fpfs,62767
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=cM5bEK9TiF_V7FDMA0Ix8NUZhuNAgdb8K6qBYFrb9fc,10127
jax/experimental/jax2tf/tests/converters.py,sha256=eqmQzNLSxZ3fhRIjJ5VpuRUH84VfnIHqju_na1OFFb8,3528
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=mnuni8_Kwueia-gIq_sbqi1l1OHeHscO5rXWNgzT9iI,7421
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=MgFWg6lGzcPaSPGSuWToGUpn3dnE1sBmWTh_w2Thmps,58725
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=m--H4kW4Ud7Jt6lfafDR89bbjRvYWNk5hfI61N4rlRU,67535
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=zabPZ6x7eN4ZQ8HIL4TQtdHf-4OH8_r3mc2lky8I82I,6718
jax/experimental/jax2tf/tests/model_harness.py,sha256=1lzDFrtZV9XmDkkIxoQO0VtNPlmzptr1RUuf15mBjQ4,13951
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=RuQJk4pCGOsum-y5vqA5CQproY-4ycwOJjdAQFrfhpA,14048
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=YiGeI2270Z2NM3alt_e88SeptRd7O5Jyt500GgqzTxE,134345
jax/experimental/jax2tf/tests/sharding_test.py,sha256=TZwLkVD7rxB4n4nib_Q_rscea6P9XnaHYbUFvX0SB5Q,31411
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=8yzPSXRoBV01dZZwkZGnrnr7nCYUgB6AsCv0zHSv_3U,20042
jax/experimental/jet.py,sha256=PXX10GibNrZeoRHlKLU7x9HNISS6xvRv18XiJx-FDlk,26903
jax/experimental/key_reuse/__init__.py,sha256=Jk5b7YRN3xcFPZwDZyhscoYLzhO_2IPC3Q9bXQ_F-ZM,1570
jax/experimental/key_reuse/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/key_reuse/__pycache__/_common.cpython-312.pyc,,
jax/experimental/key_reuse/__pycache__/_core.cpython-312.pyc,,
jax/experimental/key_reuse/_common.py,sha256=2iZLVvqsqsgHqka_nf6MITuflOKi1ogiG_Nlr3jjedk,2370
jax/experimental/key_reuse/_core.py,sha256=SZcULFXK8Gjl5-mhMWsOq27QcfdY7MaOVVOnxK1JyeU,14424
jax/experimental/layout.py,sha256=yzQFzFbz1XVt6pwOHEDEFJndisvA7u06OrUJkYJyKAU,668
jax/experimental/maps.py,sha256=yjycDuNpT_aNQJK_XUBqhDDfY7qHUKeBpfi7w_zwSAs,964
jax/experimental/mesh_utils.py,sha256=Xx1eQGithYc4vP6-S-rzycMw_pa1v0jHGERxZ2iktcg,15726
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/mosaic/__pycache__/dialects.cpython-312.pyc,,
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/multihost_utils.py,sha256=xbm-MycVR98sd95Tcstq-UYewQ8JRpvSwxagmrhLokQ,18355
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pallas/__init__.py,sha256=w3BOpY-_6QW-ryPwGiJkaud7QvkJawlW_oXb7ztCUi4,2300
jax/experimental/pallas/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/gpu.cpython-312.pyc,,
jax/experimental/pallas/__pycache__/tpu.cpython-312.pyc,,
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
jax/experimental/pallas/ops/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/__pycache__/attention.cpython-312.pyc,,
jax/experimental/pallas/ops/__pycache__/layer_norm.cpython-312.pyc,,
jax/experimental/pallas/ops/__pycache__/rms_norm.cpython-312.pyc,,
jax/experimental/pallas/ops/__pycache__/softmax.cpython-312.pyc,,
jax/experimental/pallas/ops/attention.py,sha256=Czxf_V3n6qMn_tUgh3jCIod21HqCQ33F7SXMS4cHBug,18891
jax/experimental/pallas/ops/layer_norm.py,sha256=H43nnnYXGjkc5j8EHguacLtQC3PBm258gaoabhXLjH0,10967
jax/experimental/pallas/ops/rms_norm.py,sha256=yE5gUqbjOQQfUzSJEryTKrHi-F1EN9NXPh4A4JU-OP4,9860
jax/experimental/pallas/ops/softmax.py,sha256=1DN8CrgJU2KZhFKf-VrXp5pbH64Yk-ocxkZmYUOWHDE,2647
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/example_kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=7QywIkmmDXtqcCCBYK2YXvvMAbrI15AM9LKDX1jcTUQ,5536
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=IRzFMThYUib3Wg9rm1CGoF1nasMRtvQukWsJcVs9DJk,49621
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8,2389
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_kernel.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/__pycache__/splash_attention_mask_info.cpython-312.pyc,,
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=nTpEBkxQunCzZWXkBFwSbrZG8cv-Yr0YhF19clp4yx8,73048
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=HddoyCimPGN66heae28qj7JnUOo5d2l-cK3bSlslhro,14542
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=lLtXMkHOllIL2OVH0HewOo-ySqAzTsqOkHlDy71k2LA,33390
jax/experimental/pallas/tpu.py,sha256=9lS5OvA8WdGM3xT9Pej9sfsTLg7RR_DjvWAKIoOBUpM,2133
jax/experimental/pjit.py,sha256=Yq-bII0QH18i7VYcphiiQmvx3xRcr8Ktsr1ijPhVepg,808
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=HL4GGWdBbp5T-4KH6Jwq_FCqtbslppjzf8eCbl85cGU,18684
jax/experimental/serialize_executable.py,sha256=-ytQl3LPU31nJnxHC-eZEJcHx724XiCm9-ThYj2qaUk,3041
jax/experimental/shard_alike.py,sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M,646
jax/experimental/shard_map.py,sha256=kTZOpN2m16p-bAlb08RPrtd--mAl7llmrnDI519nEdk,81275
jax/experimental/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
jax/experimental/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/_base.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/_lowerings.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/ad.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/api.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/bcoo.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/bcsr.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/coo.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/csr.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/linalg.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/random.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/test_util.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/transform.cpython-312.pyc,,
jax/experimental/sparse/__pycache__/util.cpython-312.pyc,,
jax/experimental/sparse/_base.py,sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g,3239
jax/experimental/sparse/_lowerings.py,sha256=NW3WFfrvsf5O9sxB4vN6fMj6b7F0NxtQEuj3L1oO-NA,7661
jax/experimental/sparse/ad.py,sha256=EzkW6zoTzy4t0qakVROQCoHy_QOMfmoq3dQqWsBWJYE,7578
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=pWVUYzg9AVseWedJhavgT4k2DDN_N3mWhY7m2iQv9_Q,128304
jax/experimental/sparse/bcsr.py,sha256=OUfQcM1pEzJLdlh_TMOz6tzcsdiYLmQzXFdRV9bwoG8,38224
jax/experimental/sparse/coo.py,sha256=MKH3YpJvSroCUItkDlUwHZxkJnyG1AXPlLsO47_ooGY,24122
jax/experimental/sparse/csr.py,sha256=5XdYEe3aA2W00_TK8pjZ30J0lnWD0dfwIk3YO0_epSk,24414
jax/experimental/sparse/linalg.py,sha256=fh3KocDf6MVEx_n1gVt3l15mAMfEJrZefFFtptoUAoM,23291
jax/experimental/sparse/random.py,sha256=HZeJZE9eqEr0d0gAwTQQZJKZMTFitnXSFm0zoIIsXEs,3888
jax/experimental/sparse/test_util.py,sha256=41Hvh5w84LiGxrZ1TPZWTQs1yiTSTCrhv5LP0E-89Kc,9597
jax/experimental/sparse/transform.py,sha256=wjtmkcd8-UN_TAT9cejWu8CctzsvtsMAw1bIazEGPPo,36965
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/experimental/topologies.py,sha256=DxNh3ErC2h9eUrG8UzN007aE-r_EiZtU7R2NeyWkbus,2140
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/extend/__init__.py,sha256=67qhpZPLPofDNC0vKaWo6f1j3K2kF6nZj2e9X8UrWUY,1225
jax/extend/__pycache__/__init__.cpython-312.pyc,,
jax/extend/__pycache__/core.cpython-312.pyc,,
jax/extend/__pycache__/linear_util.cpython-312.pyc,,
jax/extend/__pycache__/random.cpython-312.pyc,,
jax/extend/__pycache__/source_info_util.cpython-312.pyc,,
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/__pycache__/__init__.cpython-312.pyc,,
jax/extend/mlir/__pycache__/ir.cpython-312.pyc,,
jax/extend/mlir/__pycache__/passmanager.cpython-312.pyc,,
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/__pycache__/__init__.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/arith.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/builtin.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/chlo.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/func.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/math.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/memref.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/mhlo.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/scf.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-312.pyc,,
jax/extend/mlir/dialects/__pycache__/vector.cpython-312.pyc,,
jax/extend/mlir/dialects/arith.py,sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E,643
jax/extend/mlir/dialects/builtin.py,sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ,645
jax/extend/mlir/dialects/chlo.py,sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A,642
jax/extend/mlir/dialects/func.py,sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk,642
jax/extend/mlir/dialects/math.py,sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ,642
jax/extend/mlir/dialects/memref.py,sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w,644
jax/extend/mlir/dialects/mhlo.py,sha256=KkubU5XifH1vf8E9AY_aDM5nbDDafsnRfraerqfP4kA,844
jax/extend/mlir/dialects/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,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=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU,1083
jax/extend/source_info_util.py,sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI,1215
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/image/__pycache__/__init__.cpython-312.pyc,,
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/__pycache__/__init__.cpython-312.pyc,,
jax/interpreters/__pycache__/ad.cpython-312.pyc,,
jax/interpreters/__pycache__/batching.cpython-312.pyc,,
jax/interpreters/__pycache__/mlir.cpython-312.pyc,,
jax/interpreters/__pycache__/partial_eval.cpython-312.pyc,,
jax/interpreters/__pycache__/pxla.cpython-312.pyc,,
jax/interpreters/__pycache__/xla.cpython-312.pyc,,
jax/interpreters/ad.py,sha256=DqAotpm8H3HVbmaYQIAamDVy8J-NGtA1xpU7Zi98vJw,3454
jax/interpreters/batching.py,sha256=jjy5WCy9pS2mUzdd0fjKViSEPO6L0e5zL5gdkgFR-eY,2582
jax/interpreters/mlir.py,sha256=dQAg30IEPU5_3NypHtc9IoTySAM9x_y5aSDivt-99zI,2800
jax/interpreters/partial_eval.py,sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E,4537
jax/interpreters/pxla.py,sha256=BBH-kEticPc4e06G9kkNL_60BKlhkySRUNl3GPvZHoY,1850
jax/interpreters/xla.py,sha256=w0snrW9QYScupJNeFkjKM6s0kvLMMavMkBeWDRYKb-U,3454
jax/lax/__init__.py,sha256=7SEp4p8rDneBEzYdO1slRtUw_hiLgMyqbtqxmkihrrw,10915
jax/lax/__pycache__/__init__.cpython-312.pyc,,
jax/lax/__pycache__/linalg.cpython-312.pyc,,
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA,767
jax/lib/__pycache__/__init__.cpython-312.pyc,,
jax/lib/__pycache__/xla_bridge.cpython-312.pyc,,
jax/lib/xla_bridge.py,sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw,840
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/nn/__init__.py,sha256=o23x0ZOvhizFah-idwif160uoE8XivJvphR2-YuXXZs,1866
jax/nn/__pycache__/__init__.cpython-312.pyc,,
jax/nn/__pycache__/initializers.cpython-312.pyc,,
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=kKblElZ9IRxOhSyQeA7Iy7fJ5_WFnhGhZmkmGwSpG6E,11346
jax/numpy/__init__.pyi,sha256=Hg1cg6rnUqcxx-drbKGZc1X6euf6vtopWYD5gCuwJac,37916
jax/numpy/__pycache__/__init__.cpython-312.pyc,,
jax/numpy/__pycache__/fft.cpython-312.pyc,,
jax/numpy/__pycache__/linalg.cpython-312.pyc,,
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=olFy0Mtm-dBsv_HhWgPki_J6jBACg0lzdFSO5M37Xc0,1451
jax/ops/__init__.py,sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ,870
jax/ops/__pycache__/__init__.cpython-312.pyc,,
jax/profiler.py,sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=Zzr1TVOuXB7ufX95T161fgz5eNpjOTsfwHU8YkLp9p4,7183
jax/scipy/__init__.py,sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c,1474
jax/scipy/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/__pycache__/fft.cpython-312.pyc,,
jax/scipy/__pycache__/integrate.cpython-312.pyc,,
jax/scipy/__pycache__/linalg.cpython-312.pyc,,
jax/scipy/__pycache__/ndimage.cpython-312.pyc,,
jax/scipy/__pycache__/signal.cpython-312.pyc,,
jax/scipy/__pycache__/special.cpython-312.pyc,,
jax/scipy/cluster/__init__.py,sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI,768
jax/scipy/cluster/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/cluster/__pycache__/vq.cpython-312.pyc,,
jax/scipy/cluster/vq.py,sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM,776
jax/scipy/fft.py,sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE,809
jax/scipy/integrate.py,sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ,777
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/interpolate/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/linalg.py,sha256=TLex5rKvFTCp_pvnmiHFuKWWytX22_Skh6RW1d_wSSc,1328
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/optimize/__init__.py,sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg,822
jax/scipy/optimize/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/sparse/__init__.py,sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4,757
jax/scipy/sparse/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/sparse/__pycache__/linalg.cpython-312.pyc,,
jax/scipy/sparse/linalg.py,sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/spatial/__pycache__/transform.cpython-312.pyc,,
jax/scipy/spatial/transform.py,sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro,802
jax/scipy/special.py,sha256=jdGeirmuJrppj9EFta_eeXa_YEPK8Ma5OGNZMNVlY6E,1568
jax/scipy/stats/__init__.py,sha256=7D_LWgxpwfFnL46KaPYEc4IVTmDj7K5ynv6fRuP_VUU,1995
jax/scipy/stats/__pycache__/__init__.cpython-312.pyc,,
jax/scipy/stats/__pycache__/bernoulli.cpython-312.pyc,,
jax/scipy/stats/__pycache__/beta.cpython-312.pyc,,
jax/scipy/stats/__pycache__/betabinom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/binom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/cauchy.cpython-312.pyc,,
jax/scipy/stats/__pycache__/chi2.cpython-312.pyc,,
jax/scipy/stats/__pycache__/dirichlet.cpython-312.pyc,,
jax/scipy/stats/__pycache__/expon.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gamma.cpython-312.pyc,,
jax/scipy/stats/__pycache__/gennorm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/geom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/laplace.cpython-312.pyc,,
jax/scipy/stats/__pycache__/logistic.cpython-312.pyc,,
jax/scipy/stats/__pycache__/multinomial.cpython-312.pyc,,
jax/scipy/stats/__pycache__/multivariate_normal.cpython-312.pyc,,
jax/scipy/stats/__pycache__/nbinom.cpython-312.pyc,,
jax/scipy/stats/__pycache__/norm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/pareto.cpython-312.pyc,,
jax/scipy/stats/__pycache__/poisson.cpython-312.pyc,,
jax/scipy/stats/__pycache__/t.cpython-312.pyc,,
jax/scipy/stats/__pycache__/truncnorm.cpython-312.pyc,,
jax/scipy/stats/__pycache__/uniform.cpython-312.pyc,,
jax/scipy/stats/__pycache__/vonmises.cpython-312.pyc,,
jax/scipy/stats/__pycache__/wrapcauchy.cpython-312.pyc,,
jax/scipy/stats/bernoulli.py,sha256=V_oR9XYdYZBKZ3Nh5Y5nH_6HicaJtsqoaRFg4GYeujs,819
jax/scipy/stats/beta.py,sha256=rEx9J6IrNfs2Ln0Q1XwKl6TkbNqD5nIu8pdsQe5I-Po,851
jax/scipy/stats/betabinom.py,sha256=-B4rZ-30Nvr4Lxg9D2DXcqxKdBI3pEItbQ-8ncNJWs0,792
jax/scipy/stats/binom.py,sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs,659
jax/scipy/stats/cauchy.py,sha256=mZmRyAN4umg3GfohvaArByDHuXK54FUHSTkN9ESk2xs,881
jax/scipy/stats/chi2.py,sha256=tO4FT3DBLI-jJTfiH1_f5SqEvnRadBHNQUj1vVqm-Xc,851
jax/scipy/stats/dirichlet.py,sha256=KOGl-X9h2kKS0YlfWzsbMT5xEg3ONDgn-MWvQMBmD_A,792
jax/scipy/stats/expon.py,sha256=WS-OP1ng1jUlodys956ykXfnvs18u7hfGzQv7PnKLDQ,788
jax/scipy/stats/gamma.py,sha256=x98RFaDH4iSYxLwgmqKHsIVtEQBLSGGVZvjogfF4Ssc,852
jax/scipy/stats/gennorm.py,sha256=_NdhXS5lWu6cDZgyizZfDlm1rFVDgXGAEOYIk3Cp7FU,804
jax/scipy/stats/geom.py,sha256=uyOJ8a88yGVTe5YTjg_JO0sLZzBi9OdHMAEOYxtpaaQ,787
jax/scipy/stats/laplace.py,sha256=Oi_fMUikhKflSBJCu1R7UjGkgJLij45M-_9Q5VG0yuw,804
jax/scipy/stats/logistic.py,sha256=c-_nDR9e1a8yk7Ngv7Og5G1vfqjhFLzdTMxDE5UwzUc,845
jax/scipy/stats/multinomial.py,sha256=mysALVSGzMZahxx69AeoGbS-p1Vod2RkGdHZfKJAtRw,794
jax/scipy/stats/multivariate_normal.py,sha256=OL3t4XR7qmqBuw7Ojl3Chvsbwf_UsqtkiTL-9gHQ8JU,802
jax/scipy/stats/nbinom.py,sha256=eMTqIxq9njSnXtEInWHGJBfINEo6N4yVrwji4jsyc58,643
jax/scipy/stats/norm.py,sha256=boHN9rGUO9hx51JuRTxMep-on2HmsNxIbM3YoN-c-HM,879
jax/scipy/stats/pareto.py,sha256=gzEhI-vI4gllIHFFSe1VaTvGbf6GYzPG4siMkxOKRsg,789
jax/scipy/stats/poisson.py,sha256=hMSNeBSToe3D9W6-isGDSZDB-Qyk8GBUtdT8J2PHCMU,804
jax/scipy/stats/t.py,sha256=RO4Zd0Lm2creTHIsFR2tVerxe5TdzRWMOlSSyaH3s1w,784
jax/scipy/stats/truncnorm.py,sha256=PnB_oNVAW0trLzh8ua_T9V0vPnrlBmeBpA1FkPDPoXE,855
jax/scipy/stats/uniform.py,sha256=CfbYJ8U5fEmNa15Rgeqq-X3nEVY1P3GtVDKWj8a5gy4,818
jax/scipy/stats/vonmises.py,sha256=nQvzYinMztVFMPZm0fd14ZmHxxoi1CZC_fvRgmBwAMo,791
jax/scipy/stats/wrapcauchy.py,sha256=aWOW5KZnRAFa2rgHxwTybH1c-gbzYE0KhGVdy6BNDbY,793
jax/sharding.py,sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94,1184
jax/stages.py,sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg,1272
jax/test_util.py,sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU,835
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/__pycache__/__init__.cpython-312.pyc,,
jax/tools/__pycache__/build_utils.cpython-312.pyc,,
jax/tools/__pycache__/colab_tpu.cpython-312.pyc,,
jax/tools/__pycache__/jax_to_ir.cpython-312.pyc,,
jax/tools/build_utils.py,sha256=31JhjUYidLlxyogfXJ3KvjiKSpKg9nb_XF8qLFu8Rrc,3410
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax/tree.py,sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA,956
jax/tree_util.py,sha256=z9oGFs_ZJOhnsc9_ljiQ_rbkkwVTooz6XOxUunAAq80,2971
jax/typing.py,sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk,3331
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=3Y6YZGqyfmkIEbsuWj_DtIxI1JFRp53q_G7DPRb0VBA,6132
