jax/__init__.py,sha256=90dQC_B7Jae0ZzldsV22R40p2KMcr5HyiWlIXH3-KEk,10170
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=KhdPBKHMkvCdp6DEsnprRxZZp95BHa9sSclsqMFeXEI,7322
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=zIyWVOrlGmSSVF5-50dOxgZLfgPeZU7VNeQx6i6wFpc,1272
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/profiler.py,sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=DLfiGlzcPAOi-DCoo8wIqn39c7ZyEyXmBNgMykTs7ZU,8990
jax/sharding.py,sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94,1184
jax/stages.py,sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg,1272
jax/test_util.py,sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU,835
jax/tree.py,sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA,956
jax/tree_util.py,sha256=CQbhHUdGuidEAZlZKIyCD83F7jebmkzPVit0lclT-4c,3075
jax/typing.py,sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk,3331
jax/util.py,sha256=hcswb5c2PcbRy4oRxlPXUb1NZKy3lhwUaUgIFU5rxHs,1135
jax/version.py,sha256=5gxY9R9sZuRjUyZx5QoOyP_1x9vzptqS8egINoMjoYo,5877
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=r5HI4QdiNeLvvysAx4HoMR7MqrEzEbypFIzc8549OKQ,2680
jax/_src/ad_checkpoint.py,sha256=FFRmP7llpRq_KBKMGhr26gsw_gUFJ9iuEvQ41T7RCSM,36532
jax/_src/ad_util.py,sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY,3773
jax/_src/api.py,sha256=BknD3oSLRsHtSi4ymA-D992cEh-nLe4lOOVP724kJrE,129750
jax/_src/api_util.py,sha256=iU48VZ6sCPb5leHvr1o9D60czAi-W9M2ACGKEo--6IE,27586
jax/_src/array.py,sha256=2s9e5ZKzL-6aT1U1FGBueL7UxVGGHaI3ZvBAIUNTDUU,37568
jax/_src/basearray.py,sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A,4296
jax/_src/basearray.pyi,sha256=ey7Uh7kE9uLt-qLtlzWhPNuAQiwH5wkUn-P-pY4OvxI,10647
jax/_src/cache_key.py,sha256=mQ2C9MivVsJiA6AFEKNnSNilu9RcBcZJyLX5fRbl7tM,10551
jax/_src/callback.py,sha256=TkU5U4-Xq9q-TdZ2RqhzwmvGtAEjVB9BcyPKSiekpH8,18168
jax/_src/checkify.py,sha256=RPSiw3LPI-zjkeBOaA_kxqaZMm27kHhzqdSOSgrK54g,53134
jax/_src/cloud_tpu_init.py,sha256=dQ55WGdv84B6omg3TwHUHHcJ1SPz7eucYwbejqcoIfA,2561
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=ZRKzHjt4TeSPy5WUWAvXBWNzfm1i3jPg2SAFS20ZaHI,20674
jax/_src/config.py,sha256=NbNmOXkvvL3VseXEmUWat8UkIonJG0NercCPtSGfLrc,63165
jax/_src/core.py,sha256=Ao2LGPz41iks7fMGnH-pYbWgTlgVXsKyHmjI18NZ2lY,121084
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=5WYKAypjpDmodPsR6L_6uuVjwBx07KnE5p7avfsp-f8,60720
jax/_src/custom_transpose.py,sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E,8310
jax/_src/debugging.py,sha256=5jxy9Gy5xdrvKQopvc-sXf8UnLHpXHzl2yX4_PRALv0,24665
jax/_src/deprecations.py,sha256=c2fclD9ZpSKNQHNTndSiV0nR4dsmZS4qG3Tfnaa4JKE,3064
jax/_src/dispatch.py,sha256=7aR_-Yh6eXi1hIkXBWJPwBde5tTGWIFa9Qlv5fIuu-w,18945
jax/_src/distributed.py,sha256=8Bm845HsOZvIE3B6xzszVaxsfSLUUYnkE9AB-279wDE,8865
jax/_src/dlpack.py,sha256=_4UkTqOhxarjFEuRnlN4SJoafNdtNj9onfvJr3RlLMM,11191
jax/_src/dtypes.py,sha256=thuDu0dmdtL0y3dmkgMsvQpSuUVZ8c8VYc8q_CBq--4,29260
jax/_src/earray.py,sha256=bWTYQTuxZzDmitZBIGi03bhgp5UMAIRmhpLe7A_E1Xk,4057
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=2yIUoXYA4XLYehScnEoerQIu6geOQskNqQV-_6jerG0,24766
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/jaxpr_util.py,sha256=4mB7jfzh0IhgqiqnStUnL4_msTyOljxgzG56pw1-x3s,6428
jax/_src/lax_reference.py,sha256=hZ9Pm3Dx5iJa5yWm0oSvkrAbyuaEartugDi8uIuW454,17330
jax/_src/layout.py,sha256=gY35O7f8LuO0IqZundMAPZOR_WkFge4E1cAhn7pEGXw,3377
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/linear_util.py,sha256=9s_-v7r432MAQc6r-juktBEyIu2-tOYh0j9orLJRLs4,13946
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=VISiVmV9LYjNsvuV8XgTMkjxreQl4osK3LZSd1p3QS0,86634
jax/_src/mesh.py,sha256=I2YfKPwdLK9NRunSqs6ImaAJyXS3JGvYtOFunO_bzpk,11314
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/op_shardings.py,sha256=4FR-A5ldSwoP5oLU0SW_kjmOPr3rc6usnZ56B11wl98,3897
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=1eLjnbOfqgj3UUx2ZjgD3t24a18p5X-9Mnq2QxbN6gI,108079
jax/_src/pretty_printer.py,sha256=Lyy44ODraJilE7vr2INV1Fayq3nfWcy3xzNHQFFoSGc,15505
jax/_src/prng.py,sha256=RhIAsSS_EaSikiVtur4dNX2hU8eF5fvb83f51rJjIjo,49534
jax/_src/profiler.py,sha256=6rSB2VvWXcQ_2RNbmwfHI198reKxGhSGjjgGPQmtj3E,14529
jax/_src/public_test_util.py,sha256=Z4p1XzuocBDSXwI2HBFkssqPrIGpO3R3VETa5gyi2bY,10451
jax/_src/random.py,sha256=L3M4FGZeqh7w1w3BdFsr4lQ9osQrkW4NZ8YPiuxxJ-g,100240
jax/_src/shard_alike.py,sha256=sc7ldeOevCvlBX-XRkO00a8J0EctrBYoWitT9ocbE_g,3827
jax/_src/sharding.py,sha256=fB6sGov0DDVy1eCAJ3MrwypVzQ7Rho85-0QNXv6kvO0,5070
jax/_src/sharding_impls.py,sha256=aNYFhyXiosE3LfWm0tR3BI_e55OuDSLy_M9smyahO-c,50719
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=AjLwjmZXBhYFM13j4kMBWKK0Ch1A9Q07aKwyt3Wcefc,9912
jax/_src/sourcemap.py,sha256=Ho5H15VaHkIcPqVd3TXAz3_0VkZmH4yxXpxsdHdwj4Y,6613
jax/_src/stages.py,sha256=zW3A4y4ZD8nsYqGszJ0j7f6NO5DRZxcbCr0FIC4XsEc,26637
jax/_src/test_util.py,sha256=JECOec7Nh96EM26sRfD894wfeSDKl5amWB2ZpqdIak0,65924
jax/_src/tpu_custom_call.py,sha256=goKRf8oMtTuHF1emoHa-7e1dtcOgl53FqNvxGln_CAw,18227
jax/_src/traceback_util.py,sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA,9060
jax/_src/tree.py,sha256=qf_oRe7McgufUBFD7LtV9KG8XL_fJB-kn7SZDUaSka4,3492
jax/_src/tree_util.py,sha256=ZIUmVUuuceLHHh592tGoyDJbIk53Qsu0VeQN6erkYKA,40504
jax/_src/typing.py,sha256=sE1S2htGBtwgb86jXkBvBXRjpUiWZTDBkeg8d6KG3gw,3212
jax/_src/util.py,sha256=SSCcwyUxy6UHcrWd5kvp1IC6xmKpIJJ4Fsw1sN--zO4,19369
jax/_src/xla_bridge.py,sha256=F-chIEDKtu9LpM2byY-aU_eH1v57YIFo-J-DKEh3kMM,44268
jax/_src/clusters/__init__.py,sha256=ZIzPN7-Awrf3mDafphxQXBXWwItcp_8axkztAA1sBYc,1095
jax/_src/clusters/cloud_tpu_cluster.py,sha256=ILjkSzl0OI8cZyUzwCTAsXDG2LcguWfr1o4B4ZzKyuM,8148
jax/_src/clusters/cluster.py,sha256=4zenRWGqAo0CHbhEq_X0lNZDvDm5WsWwNFzHYDmvtPQ,4901
jax/_src/clusters/ompi_cluster.py,sha256=EUK4RbIetmu9LL_jzxr1cT3kRtTfgiRI__VcOLGKqmA,2294
jax/_src/clusters/slurm_cluster.py,sha256=1AmqbARp7rNWe2v7UI_AD100v3qdg2o0IV66zuCf7M0,2297
jax/_src/cudnn/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=tMep5Cykf62pQx-2LWi8Jg6RNjjLebX3teCji1yJoeo,37959
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
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=vQP8ysVp4tDbmTJvS4SZ2VrRfPb69lWSXFUH32FNJ4I,8368
jax/_src/debugger/web_debugger.py,sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w,3433
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/random.py,sha256=bAtdkmxevVfY9JfcwrWX6Ah8RXMoNelXg2I644ePHCM,1276
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
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/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=1a6Om8APtLp7tM0KgMJkjtj3JTFD_5l_--l4WmYqWYE,15448
jax/_src/internal_test_util/lax_test_util.py,sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo,12140
jax/_src/internal_test_util/test_harnesses.py,sha256=tRmHAbI2PoYz03LA3POZ09cuGWHF1F_Dqq7Yvi4yfSI,124650
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/ad.py,sha256=XGIiZyYNXujjGE4M4T6feZ8yStPKiXaSSBiunyn5sJs,32701
jax/_src/interpreters/batching.py,sha256=NJDTJxEy2T1BVJVXzyyyjMWcuua7lmTUou5Og2V6EjI,50430
jax/_src/interpreters/mlir.py,sha256=h5rHhedkK34VsEhjcyR2L-QoPygDQIU_z2UBwZDKcO0,118017
jax/_src/interpreters/partial_eval.py,sha256=aadSJP5e2qK6TNSoHAPMhg0dk0LLp0OANXQizaDiBig,125764
jax/_src/interpreters/pxla.py,sha256=BDAPT6wvrvt0QXFxNmXqRpfMJdKJCWTYneUZCISFbbg,132423
jax/_src/interpreters/xla.py,sha256=AFLJHc6vzGVwcnD8Bm7ywuLeYraYJg8gEDXANMd_R2U,12203
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M,17623
jax/_src/lax/convolution.py,sha256=SNNtSYshLLvMAMwsrTfnDPGO-deZeqR-Y_Nu3rebJeM,44715
jax/_src/lax/eigh.py,sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ,22235
jax/_src/lax/fft.py,sha256=xrCTzjke5flYvm4anB8URXLiTwHRJDete2s0Eh6qOdQ,7138
jax/_src/lax/lax.py,sha256=ibVE68CeI5usTlvoWCe_fWynm9ib7HBQpCi4imtbSHQ,211036
jax/_src/lax/linalg.py,sha256=he86j7DSr9WZDGF6oLDda4wRGr5yPSwQNzJcrIzWqQQ,89497
jax/_src/lax/other.py,sha256=wTmrVnNdLQajm04LyGa4ioIiJseZ9dT9bWbQmCSmO-k,10539
jax/_src/lax/parallel.py,sha256=PQeCK2JjoolViIcUpIotceeYzOebzCJ5IEeWZFCHLJ8,75671
jax/_src/lax/qdwh.py,sha256=_kincbp02AQqml2giwIjIxa5vgXQwDb1smf4iJiaRZw,9352
jax/_src/lax/slicing.py,sha256=2vweurIzJiYKWYWTsY_Mq-pqe3eO8j1wXeihkQKnKm0,111676
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=-vjd9rJsQ7Tua5Nq6rUZpndQJ_WsMy0xr0zposQpYOc,8795
jax/_src/lax/utils.py,sha256=KHbNG7zWt4w8vrSxSJv5iKubesvLdZlX9HVTXLyr5Iw,4968
jax/_src/lax/windowed_reductions.py,sha256=fsWcwcqdImfUtNeRI0rbsIE5g-bb1mt9Kq7nqLB6DNA,42314
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/common.py,sha256=jE7vo0resS9tHdg9F9nleNB9Gwcv8X3VUzpMHNo0OxA,10976
jax/_src/lax/control_flow/conditionals.py,sha256=kxQjGAd88z3KuA2i0BEqvqh0916J1ZeGdtPRRikP0eY,42960
jax/_src/lax/control_flow/for_loop.py,sha256=Xj5lAkcgI4tVM7UEJa8osAvK2SRn-f9qWO0RnS-owkk,37136
jax/_src/lax/control_flow/loops.py,sha256=asW5qi9LSzYPncZblORrZmS7OyC6qtBBT5I6u2lkAP4,110663
jax/_src/lax/control_flow/solves.py,sha256=JCyYNtgi_hoQ8Qy1tdyr-UbZ4Y5JhZzwUXwFc6fz4s4,18876
jax/_src/lib/__init__.py,sha256=7tpUzhxrqZNGVaXIHhePyfZteb8aG1TEFP3YqerL3lI,5509
jax/_src/lib/mosaic_gpu.py,sha256=5ysDEKIKWeBT2vsuME8Xy8-21ZzeluvpSwB3Ip93Foo,849
jax/_src/lib/triton.py,sha256=HbqoIneu52En_DHWH6Mph9rGtj324CRCXoT0SZA3u44,1336
jax/_src/lib/mlir/__init__.py,sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM,733
jax/_src/lib/mlir/dialects/__init__.py,sha256=d1ytY1GozntPVDvB9Hs-8VG9MWqki_EFtBXkR1ind3k,1463
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=MjUYab13E_JsizxW3K3dLTrhoQ7nPw7hCAy0G8rqri4,20815
jax/_src/nn/initializers.py,sha256=526DihqSYNUWM4i5EET9rvZl0btVP27vdZs8Tpl-Qu0,25570
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_methods.py,sha256=xpuooSQKi8x5gCuzx0mCK7OHW1CkmALhIxJ4T3moaMg,32851
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=IzaXUVx-s3ikcJVZx23qoy4uS-jNhwEyztjoMnixJjk,258470
jax/_src/numpy/linalg.py,sha256=-rfnB3MqZyHUjqp6-6ukOZhw4weNV08Uj19zFqxDGa0,72683
jax/_src/numpy/polynomial.py,sha256=2uEH06Zqya_x15M2ROYEaV68fwmwCmXs-gKZe6mnXzw,12162
jax/_src/numpy/reductions.py,sha256=qPdKHsbhihVpQPtD4qjpEI4sWxoS31zFlNlyYiSY9ao,43233
jax/_src/numpy/setops.py,sha256=pdYBHaTrUHOhQtNIvH2VXy3kk2Vk39VmpM2BdUWwl0w,15407
jax/_src/numpy/ufunc_api.py,sha256=gLv6xWRuPPgjiCDd5FqFle6CrEYirCpGTM5hWBrHKOs,15256
jax/_src/numpy/ufuncs.py,sha256=E6Js5-nSgxuxUe5LGTVm09LZnrIYOHVDeFjYW6f0gi8,28777
jax/_src/numpy/util.py,sha256=55wWiEaqu2SHthyoIB6SYdIWCQeUoFmehrWiP9kYwfQ,19129
jax/_src/numpy/vectorize.py,sha256=H-OiIbDQvKmjpOJlO9L7cpQOYn8cOJQod_7C6OHOrB8,13021
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=kZ-h0mJzy57grnDe9PFZdv4TN6u6_OxRauIRtCVZ214,17593
jax/_src/ops/special.py,sha256=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA,4151
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=So66wozwzRaH9kcEQiKslkBKhy0vB0VDS5vfdFJiCmI,12377
jax/_src/pallas/pallas_call.py,sha256=iFIUzwOr9wLUmDgVoBC9xnrpVCIzd1O5FMqrTBa8Vjw,25214
jax/_src/pallas/primitives.py,sha256=KDP-3p2G8nLQmoDiBjTG9JEMS7rnjjttLkGb-q_Oh5Y,16820
jax/_src/pallas/utils.py,sha256=aD-qsJhsPq1IYpEFH7pHWld18Tvpy08atzyrPWPFKOo,6047
jax/_src/pallas/mosaic/__init__.py,sha256=SNH698p-06sP-_vtmw3LqVnyb66p6TolDBZ0gqcIFe0,2470
jax/_src/pallas/mosaic/core.py,sha256=pIsvseu1Eu8kRKBe1ftlMVMWfvOjh9s-SPJyItYse6A,7643
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=N5iIItYmge_yW2WiIvhoINEAgFtJqyCreNXtgI7yK14,1994
jax/_src/pallas/mosaic/lowering.py,sha256=5p_qpN32yS1Qgf19ipEifIJLV5sWJtBfQoUBAELnvZg,79971
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=mikkdvNFY0raTXs9aEzgNUHg7HFHFxx9A5d4JFPXB-k,4480
jax/_src/pallas/mosaic/pipeline.py,sha256=qCTTr6MtNCqbhV_77tvQYLlnw-eRiIM6lvyhfJfwLfg,28856
jax/_src/pallas/mosaic/primitives.py,sha256=kJXN4sNPDygFKya8XEKnEEnLI4AU05-P1bkUHDe8Hz8,19232
jax/_src/pallas/triton/__init__.py,sha256=sgvpHfp8om5F87TEh7lyxzM_nuMFkfnrsEWmB9ndVU8,745
jax/_src/pallas/triton/lowering.py,sha256=YgnLnTaPVNL30LHJ7QRhxb6ENA_cvkhTwwyBIqzdgzw,79878
jax/_src/pallas/triton/pallas_call_registration.py,sha256=m6kkWcyS0MDhca0yHDAcVw5xTbb0f83pcd_ySu5k5wI,5389
jax/_src/pallas/triton/primitives.py,sha256=9ht9HAeiusMIVkUewn3wbdVeXSYJYrjMYUXVQ7r9GUg,3663
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=e0OtkbLCYIYQsJkH7mdoocourvHj7Clgt7CwAI7LZwk,9401
jax/_src/scipy/integrate.py,sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0,2346
jax/_src/scipy/linalg.py,sha256=0fBUFGWcOCwmn-Y7168lqzwdcIePkDl2tPNY0vFjM6s,73276
jax/_src/scipy/ndimage.py,sha256=-7zjlsJJ-Ec5Shd6svevkHZB6M92HZo9_gwn57vgLK0,6586
jax/_src/scipy/signal.py,sha256=h72KyW1p-8WHwStiYQJScmoVHYn_wICZLuEonSVSey4,44924
jax/_src/scipy/special.py,sha256=QHc79nKSBm21Iw-P33A4qyo1yM5Q5oeCleBIWC08HiU,77626
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc,3043
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
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/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
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/transform.py,sha256=TZTud9BHHcEztrE62Kf203FT4FULwOlJ0gXDaODycgU,16904
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=uS-4jeesXR3po51vSVmGQkWfzVCL0kQ_1M1xBnGhnqc,7206
jax/_src/scipy/stats/bernoulli.py,sha256=kKe8UJUzKFnsZCzAwu-gpxEvHwA0rA55tkIl5-IeERY,4543
jax/_src/scipy/stats/beta.py,sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME,7988
jax/_src/scipy/stats/betabinom.py,sha256=_B3d3G87qXGtUn6lhllxGfTNoOH8X9oNqgMVIKwqQh0,3299
jax/_src/scipy/stats/binom.py,sha256=Imb5lRNyIIQKvOr5D7YZNvllEf8Uo7VDhseeY6bvbDY,2562
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=fzZUpOFkFrxqjbx3fzRnQNd9zp_6RhQ-m-my2MCjsds,6957
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=QhiPQDIYpc7j7Yyi2K5oD9PDLm75lYd_FAXvJuePUgQ,3431
jax/_src/scipy/stats/nbinom.py,sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw,2603
jax/_src/scipy/stats/norm.py,sha256=tPv38wvCiXyjeJ4k-nIqtq8D0Y9sGfEhOhStTeBRf_M,8500
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/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/discharge.py,sha256=I0GfrSfhSsbf13F4aZpkAEgybrjjgTlnYwBRAcrvuxk,37163
jax/_src/state/indexing.py,sha256=uzcQKJwqZOEGEbsCRX7AqPZcOapz5T0JKs88augDIes,8495
jax/_src/state/primitives.py,sha256=v6IOxjWutFixvreCKU3PboIeY8NlIsL5tAKjW2QXm1g,23555
jax/_src/state/types.py,sha256=AA-MoZixDqnHWh6upqlJwtfaGlKSuJfwi8tVf3SmBbQ,6685
jax/_src/state/utils.py,sha256=AwdUqrfZto_M6x2L5L4KCoBARMsjhYwfb3jvjmawjsU,2411
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/linalg.py,sha256=VqvNAxMDyyr1vlwXCZWRbrvx4uBiIzSNC5T6LIfHpC8,1512
jax/_src/third_party/scipy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/scipy/betaln.py,sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE,2373
jax/_src/third_party/scipy/interpolate.py,sha256=N1tC_kaNwNy46qkAw5jzSn0EbPya96MaEj8qaFJeevw,6352
jax/_src/third_party/scipy/linalg.py,sha256=VV1zywtvAn3ieZ4SE0WtmyDDlfkD59ATHyD2nZkCLvk,3776
jax/_src/third_party/scipy/signal_helper.py,sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU,3495
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=224Ba6KPYLs3zlF3g7dlubbQkIkvoko-WGqDx4svBzE,20559
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=OG7eTYk40QQN4XxZ7E7mFnz7pI2kzo7FH4tU5bteG60,930
jax/experimental/attrs.py,sha256=m_iOiq5bKo6emXg7CfpdWAvkYq56Gedhmoe4u4osWgE,9155
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/custom_partitioning.py,sha256=ziO_WOvZu7OvDk8Pt_s6jeR1_ys1RQK_FLn0-keUACs,22374
jax/experimental/host_callback.py,sha256=v5DhVUJsqwppa2xSpUZhlfyEvTuv6KxD5tyt8eBVH-g,85234
jax/experimental/jet.py,sha256=FjrIRKB48rgy6rK1apiWXvPT8gGibs7ZC2gkzEz-CeY,26828
jax/experimental/layout.py,sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY,675
jax/experimental/maps.py,sha256=NmRFDPQdGIFLD9YnSOr9Ui7_kPZV3JaV3MXGcVr_2vo,1660
jax/experimental/mesh_utils.py,sha256=n54kJr0Q-Vtwc2w9Yi42bqFYpyz611K-XcqdZS2Bh9M,27505
jax/experimental/multihost_utils.py,sha256=xbm-MycVR98sd95Tcstq-UYewQ8JRpvSwxagmrhLokQ,18355
jax/experimental/ode.py,sha256=5MJfwT04Jgk0_J72aGLuMVPJ9J_B5gEtOMs8g_F6qnI,10651
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=m7lOKgQvkNvEQFl5ooEyXI39OWHPh3-07UEjQH7-ZiI,83306
jax/experimental/topologies.py,sha256=DxNh3ErC2h9eUrG8UzN007aE-r_EiZtU7R2NeyWkbus,2140
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/array_api/__init__.py,sha256=QZb8tEnU0uyjvu87sCyk31gOcP78sf8SFDWi5sOlPyI,5345
jax/experimental/array_api/_array_methods.py,sha256=ShBvQnAkkkTTlb-ObrnJWxtA7wR-Q0_UIPbaFig4Syo,1649
jax/experimental/array_api/_creation_functions.py,sha256=ZzDon7kbUgeIXkRGypVeaI-PQDiGJnEyKpl2uIywUpk,1361
jax/experimental/array_api/_data_type_functions.py,sha256=MDFWU2Gov0Ac_n7tOaPGqVKMjixpLRdzXHiHlWrGU44,2380
jax/experimental/array_api/_elementwise_functions.py,sha256=HnC1Jp0hMI3sk-UXw550awquLAZV1pB9xhHT6YTlQbs,3064
jax/experimental/array_api/_fft_functions.py,sha256=UsD6PchkbDTnefrp9sYowcOCJ_in3cq9Sofbsja4GDo,1044
jax/experimental/array_api/_linear_algebra_functions.py,sha256=qQK3Wn1YsZ1aZwGKHvB0f6gQ5rtHkpprCDCDhxxlnhA,1125
jax/experimental/array_api/_manipulation_functions.py,sha256=hZKe9fP3PQBF3JdodT4i_mtgHfvxrtSNV6MwhpJ2LOU,922
jax/experimental/array_api/_statistical_functions.py,sha256=0-QgXgMkoSjOz_IYInL9Tm1eowDpuQQ4vJr9IWinqt8,1068
jax/experimental/array_api/_utility_functions.py,sha256=sq0nyOK-k7HQeZv8QO_OATXplWLcgvLORKC5pPTLqT8,2697
jax/experimental/array_api/_version.py,sha256=5WxaUsOOnZkdWHKRESGLI-rt7A8tkXP2rJWnI56VKoE,616
jax/experimental/array_api/fft.py,sha256=vjlTeLNMh6pZn6CjXFeP9itrRqNJvOel6i9CA1_fHFk,937
jax/experimental/array_api/linalg.py,sha256=8ElH9va5pwAzayNw2X40ZU7rXxkyEuGyRj_dj1xaow0,1255
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=x4k6rk--1kT_vWgV9b2M94c351NbYMYCuusvP1G7Xco,23514
jax/experimental/array_serialization/serialization_test.py,sha256=lkqlanmuqCkuTdL50ZJiRNgtgCZa3dfeBYOaYNo8pfM,17487
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As,820
jax/experimental/export/__init__.py,sha256=dk1YZG7FSmrB47imONOWDKSEsX26XyyjI5WiHVULU8k,1268
jax/experimental/export/_export.py,sha256=qsvrsspY13HUIns77zyZMwpxT_qqeBpQgq0gTXxyZpE,59155
jax/experimental/export/_serialization.py,sha256=f3Bu0wxMA0DvJ8Fw1pMyd7W_xi4k5P_Npv9k3HXddR4,17212
jax/experimental/export/_shape_poly.py,sha256=gElBg9TnYmy_gpxDGsLAwEd3Wa4ZFqa7Dx9HbL_Eo14,85388
jax/experimental/export/_shape_poly_decision.py,sha256=x9AADPHzkyQSjJEtxbIX0rOsWBqT2T2_TJ1oXsgTmN0,20711
jax/experimental/export/serialization_generated.py,sha256=wc5JKltgI-SJx6g361gxpfY4OK2DvsaVARhrm5neJzU,26195
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=l-pGVA4gIp-5k5hX3wobB3Eqz6pdhlnHzo_EhNH8i_8,27718
jax/experimental/jax2tf/impl_no_xla.py,sha256=SIvqeN30ld0PqoGduuUC3JwMxvCPvOsqUZnWxF_M9W8,54476
jax/experimental/jax2tf/jax2tf.py,sha256=Dr73V34d5N4y41wz64xCu8AUBraB5nprKonmsEuNYpU,143349
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is,2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=P618Ht7Xrt4p4T5Oz83i9KIGC8Ziiv_-4iQVxiBz4kc,1608
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/model_server_request.py,sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8,4952
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=wlSavCUpxsHoENM-Uqheh4NHxNBX2u5TrnRz4AO6JRQ,5404
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=yd8aSO2ND3o4XoZtd9-8tSvKYf0azOmUMJpKRW--kVw,64818
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg,10108
jax/experimental/jax2tf/tests/converters.py,sha256=eqmQzNLSxZ3fhRIjJ5VpuRUH84VfnIHqju_na1OFFb8,3528
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=UVVnIwHOdzvbMyMxCekTLW2nZ8a7731kO4zjT1qdCBA,7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=nHQxK9PvIHYRN1IQ6ehrL11_oJPra-jLiPfoynvqLks,59363
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=BkEZsKkXw7ZMI56GoOE_EY5V2y-RU4vg7a_4uHLfDEA,68326
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=F_tkcePs1M1XDh83wlgFEu8b8iPZMJwkvPu1yWYI3cs,6710
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=wP442L9HwVRIG2hBbUy1N1GNLfkPxCS9wIkGS05o91M,14099
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=HWgHfC58w5olpNyVtErVVZSa0N-nJJe81iXkKo3jwbU,12796
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=YiGeI2270Z2NM3alt_e88SeptRd7O5Jyt500GgqzTxE,134345
jax/experimental/jax2tf/tests/sharding_test.py,sha256=a-OWyXIgrushNtZqrEyoSXCWj6h1X-1JtN9Yvq4k1os,31884
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=7L-xikndbablYLUybeHvRQptIL4vsZ-xd7Yoyh_XvAQ,20034
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=DQ4pjWN7KWp2wmUJ-XGESoJsU4IgGRpin5zAMT6zqu0,22789
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/mosaic/gpu/__init__.py,sha256=Mn45HqgLGGuMaruU7gLG37XlV5tDT2rt6U1JGkfjs4A,27244
jax/experimental/mosaic/gpu/dsl.py,sha256=-MdXdOPk25U_O8FYN_w4JzHX_n1ZRTLvNX0gwpEpeqk,1202
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=d4sHd3AVsF_ukO8MdR8mZ91M69NeOhiljVyuu7mJ2CM,19258
jax/experimental/mosaic/gpu/profiler.py,sha256=lrlJ23kcbQ-6Iw9yGMVCML40UsUThPmP8CUtIdFGFiA,9108
jax/experimental/mosaic/gpu/utils.py,sha256=9Sw92jT4gd4weLrgMGYWHM9_YxvKK5V-e3SjAC53yvs,20927
jax/experimental/mosaic/gpu/wgmma.py,sha256=Wmj5oOI5qaZrPi_jNPx-gG7FxeOBnKC84boF6A8UIjM,13785
jax/experimental/pallas/__init__.py,sha256=gGfMuwK6zh0Dgg5QMYa3OJM0PEgNKG7rm3Drncb9oMA,2094
jax/experimental/pallas/gpu.py,sha256=AsysKem-rADhlQ6YvtpRrQzTBh6nTRuhQzN2oeS2_JU,723
jax/experimental/pallas/tpu.py,sha256=DIqDmiUj-qx4BKuzYNrKYpOztpSDqa7tLVbi4I3BCa4,2195
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
jax/experimental/pallas/ops/attention.py,sha256=hzjxm73o_bPA6IQV6bHkblFwYLO_ov8GXZoot57Zl6k,19059
jax/experimental/pallas/ops/layer_norm.py,sha256=69JJCFFul_osv8y5p5RDyXGObLetVH1AUVUnDwY-hNU,11099
jax/experimental/pallas/ops/rms_norm.py,sha256=qaquKh22ka6HMiMZbq2h_vZhzRyZUHjZ4kp92ePohX4,9992
jax/experimental/pallas/ops/softmax.py,sha256=M8TJA66HR9OhvdQrA0SzrTrZNyU5n-kSkYuogkcWEIg,2702
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=x-piJ5UMQ3XhJh6f91VS_EZu4vi6jDbV83AxZmbH478,5575
jax/experimental/pallas/ops/tpu/example_kernel.py,sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM,802
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=xGgD5gzwlD3jwCDs23uvAnUhgQUgEdE1FbMknANQAh0,49743
jax/experimental/pallas/ops/tpu/megablox/__init__.py,sha256=Qutkom55JBjOTRrTwsmPrjkZPUx3om4iqXxp5IUZ5c8,643
jax/experimental/pallas/ops/tpu/megablox/common.py,sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38,1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py,sha256=FiyYHnafRgeMsbl_HUTb9fxM-sV3cL0M05AVeOqTPFg,27332
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=UMptQqdj5o3jgSM32_1zupFNHgD4ft7MQ1zfPbb5xZI,2944
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py,sha256=bCNpNWR-hJeynDYamiq0BfIsRtvvBoGu2SopmYG48UY,681
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py,sha256=kZC4YfGb1MS8IxKp1J2230Fh6ZyvHjJwPruSzHmjRu4,16551
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8,2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=itT-Nbq0HNYqlTQuTyhLD0vCfCdJyWx7ud9t2qbTi3c,73245
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/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
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=TFSakBd97x3rP0i9KemXKlcIl8vfijmpxdvOggiT6mU,7588
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=HBgYTdLjjWKYYfST6623Y41Z0lJXuolN3-fGCVO2Tjs,128298
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/nm.py,sha256=47qFK7uZv5mUkqsxYRbeVzRXExoTOnPJX4GninBLQWI,8453
jax/experimental/sparse/random.py,sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA,3882
jax/experimental/sparse/test_util.py,sha256=x6vD7pf9qNEqfOBj1a8hf8my7mW_Mu7RR5EK0TmBWF0,9489
jax/experimental/sparse/transform.py,sha256=an67mxJSgq4Q1fRYcy9GMWdsID3xX8SkLbwgo83vA0E,37274
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/extend/__init__.py,sha256=cDhCUK2LM4fNu0JYq8amg4G9dHflTjkrSyHSgmSEyMY,1249
jax/extend/backend.py,sha256=zDbInIE5czEJduevvmZjKRB_9Avtnvdkf7fHFYsQPrg,776
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/random.py,sha256=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU,1083
jax/extend/source_info_util.py,sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI,1215
jax/extend/core/__init__.py,sha256=XqUEruxcH7n0c4POF88VQ9vM-Zd24HjvDQ-J3-Nl1R8,1036
jax/extend/core/primitives.py,sha256=qCXH7_O1MtW1_pnwJq4PlhuGWmbYE8Ze100fCH95O20,6199
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/arith.py,sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E,643
jax/extend/mlir/dialects/builtin.py,sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ,645
jax/extend/mlir/dialects/chlo.py,sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A,642
jax/extend/mlir/dialects/func.py,sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk,642
jax/extend/mlir/dialects/math.py,sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ,642
jax/extend/mlir/dialects/memref.py,sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w,644
jax/extend/mlir/dialects/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/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=BjuN2XLu4xEetW4JuqsC_FC8oiYRjHDqa4-p6kcv7AQ,3302
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=snyYN8ZtNsZnY_ZEC4zKNLGslOkbMD-kVt9LxOlSLCE,10943
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA,767
jax/lib/xla_bridge.py,sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw,840
jax/nn/__init__.py,sha256=ltStD3DDti3TjvcEnBJTub9Nd4A7D2mJDlvy8evIhBA,1948
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=6mjMj_21P-quLn0_4gnSI2T8hx0ztBoXBKfJEr4OjF8,11642
jax/numpy/__init__.pyi,sha256=ux0cetzNwWV2bdZ0W4H7Z2eQTAvl74IrRaqPtq8kb_A,38668
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=aW1R5AYjely8r0YZ5mJsjSun3tbKzZVg6OceUhohNlI,1451
jax/ops/__init__.py,sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ,870
jax/scipy/__init__.py,sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c,1474
jax/scipy/fft.py,sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE,809
jax/scipy/integrate.py,sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ,777
jax/scipy/linalg.py,sha256=yqsNTdJ19X6BxTrZfi8RFJZ_vSjebVkIZnuuKFPCshY,1350
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/special.py,sha256=BoVGqZF8EAkYm5x2GtZHup0VOLNC-DH-aryO4Xv6gRM,1592
jax/scipy/cluster/__init__.py,sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI,768
jax/scipy/cluster/vq.py,sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM,776
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/optimize/__init__.py,sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg,822
jax/scipy/sparse/__init__.py,sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4,757
jax/scipy/sparse/linalg.py,sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/transform.py,sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro,802
jax/scipy/stats/__init__.py,sha256=7D_LWgxpwfFnL46KaPYEc4IVTmDj7K5ynv6fRuP_VUU,1995
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/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/build_utils.py,sha256=31JhjUYidLlxyogfXJ3KvjiKSpKg9nb_XF8qLFu8Rrc,3410
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=lEcqDhTx0b1jzxc4Azx90p7yonudePabO6SkEnS_n9g,8708
jax/tools/pgo_nsys_converter.py,sha256=udKwSZG-RDhlvA_Z0vGkW92nR1BLXGVBdxTU0xLJhVE,2716
jax-0.4.28.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.28.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.28.dist-info/METADATA,sha256=dkFNbYsqley9GhJjUMkt43fDj329OViKhMmtNU-3lng,23708
jax-0.4.28.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
jax-0.4.28.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.28.dist-info/RECORD,,
