jax/__init__.py,sha256=2jvnype07-kWIE4IhOQRHPyYEiCpzUCQcV2TT6pXhlI,9302
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=B6dnRzeKmU8XTeZLlKAT06VQDwoCLw8c7fJoSsKYOuM,1119
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/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=Zzr1TVOuXB7ufX95T161fgz5eNpjOTsfwHU8YkLp9p4,7183
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_util.py,sha256=JUrEicdopHCDcvhKWwbcPXFzx3BEWAB6i84OxO3Ivi0,2931
jax/typing.py,sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk,3331
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=S0gYh55XODHC8mjS4iAfG60F29kxUQYZNt645OtJS4M,6132
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=b4bjzkOmImOQn1kvGqYJOo1QWRJ9tj0w-1QBHl-Y4JI,2788
jax/_src/ad_checkpoint.py,sha256=ybaGDYAv5oywpcj0fO4rkvxNs-leqHt06NT_XRc5we8,34691
jax/_src/ad_util.py,sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY,3773
jax/_src/api.py,sha256=78J5Vc9rxjtMx_SBwZJEO3eZuX-foywFq0zp5FHdwHc,132535
jax/_src/api_util.py,sha256=rjEz-vlf2gtWDvWTWQsU0cEWr4OvH1QstT1kRPxT_5w,26475
jax/_src/array.py,sha256=fE-azR7dgA52usjSVFcC8pgab_RBHWKv-0eX2LWGNy8,36726
jax/_src/basearray.py,sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A,4296
jax/_src/basearray.pyi,sha256=mf8HC7mdr5lD1N6g0MAZmW8jyqh_oDIrumizAxbPMhg,10655
jax/_src/cache_key.py,sha256=PJd8i7h6ugdOkjFGU7icN5Vtsz8anbtJmCUDsneoho4,9523
jax/_src/callback.py,sha256=y6GhYzA0H6PaZzVn83O-SEfKbdpTtNK4XvSUzAYx3rA,18282
jax/_src/checkify.py,sha256=JN7HbRfmbiG5a1fYHPxdvAcDV3DztGIku_gbfz4wA1E,52865
jax/_src/cloud_tpu_init.py,sha256=IEgfq__4iWV48193XEI9O-GAxrWX4pc6kPbdF7_jDSI,2860
jax/_src/compilation_cache.py,sha256=JzU-uiX6OsQJxgb5m8rrung29IAPavx9mpZ1LFYAmMU,8674
jax/_src/compilation_cache_interface.py,sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c,863
jax/_src/compiler.py,sha256=gliiaeJeajK3l5aN6ZRyz_gm1BzoV-569bcCOQEuo5I,16425
jax/_src/config.py,sha256=68UqhY41kSml-BLnX2nCZTmRS08AAQlBEtZCUtC8RrY,55806
jax/_src/core.py,sha256=hNoOyXKuLO4M_-zXzDETgeboCPjGiKEIkmwgZrx8xAU,117095
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=X29Q8bbi33PT1Iw12uggsM2BQYWOF1m3bCb_cawtWEQ,58852
jax/_src/custom_transpose.py,sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E,8310
jax/_src/debugging.py,sha256=oKvllSijZ6zbDpjy7d424qWEfoxEsgTIlg4xGlvR6tU,24309
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=Q-3NW99t3Xa8rS4F-h8zzvxwQfejoAikcR5FOZg0hMA,17768
jax/_src/distributed.py,sha256=Jt00nRrt8jlr9f2AgCtyN32_l9On3yciG1LA9lE6vbE,7670
jax/_src/dlpack.py,sha256=1kKc5gNh9AuFmf8PHDonjUULJdyrXWcGC3KdoowHcU8,4948
jax/_src/dtypes.py,sha256=QFFOrnhvn7Lt_E9VypecPE-uOrWgq-EjCqRoSlDqmfA,28876
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=1aFtOHf6p62XEmy98E2uq-XW6xHNuQOPJH2kVWJL2nM,1962
jax/_src/errors.py,sha256=jJvt8vwEuR8Ie_hdqjBqSwuy5zD74nkIXvbT_ttPEso,23777
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=ga5-18Pz5gAV2ktKpJJzEPlfD_LRDMvuGtfZSloqNi4,1585
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/linear_util.py,sha256=pk_KxAS9L_CipMrDSca-grxfplG1OIdp4gomh_AVS3E,13974
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=cKijsSr9txYp4VxQ90JNZAhPOCFC5g9L45wcrePpwyc,86301
jax/_src/mesh.py,sha256=I2YfKPwdLK9NRunSqs6ImaAJyXS3JGvYtOFunO_bzpk,11314
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/op_shardings.py,sha256=mb_s-ThVgw-rgM_ZZVH1P6Lpgl1RtMauk__Qn2g9XqM,3895
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=gXhd7LO2zTZEEuXZU2VZkqHZouAy2vpGhIcCBpagKKI,101679
jax/_src/pretty_printer.py,sha256=gy89qxIgK7ZG7P9cglSmzrtFFjPS1uXWdcUbUjH1iFo,13077
jax/_src/prng.py,sha256=krGm9koKBAF-6MivvNO4SdwgMcpWR-oAJdmemZUCCcY,47433
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/shard_alike.py,sha256=sdSYS0jagWfqS8pkdLEqQg6tJBr2QoU92dEgG0Hv4nM,3978
jax/_src/sharding.py,sha256=fB6sGov0DDVy1eCAJ3MrwypVzQ7Rho85-0QNXv6kvO0,5070
jax/_src/sharding_impls.py,sha256=vXJzpanDV25UOTVwQCj_YSmUOoJpyX91PvRRhVOgeRA,51249
jax/_src/sharding_specs.py,sha256=9s9WfSdGriLgmreTp_2xlqQtkf1TmSxCWRmtT-q--1U,9207
jax/_src/source_info_util.py,sha256=j8L8uGsyv8_qdqd6nE1TMrPf0yrvedySAreFtvxRJd8,9348
jax/_src/stages.py,sha256=2HsY-TR5r0N-pgTZJn0RmWU7ajBSbArNbi2YRx4R3Ko,26020
jax/_src/test_util.py,sha256=Wvt9X0i_qKrROzr-LBwCf7wnIRM5ya2dzQHxG3MS__c,48326
jax/_src/tpu_custom_call.py,sha256=3GPHsopx8zspBCLyv5h77X-UqYSFX7x85pHQgmxtW0Q,12931
jax/_src/traceback_util.py,sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA,9060
jax/_src/tree_util.py,sha256=8w6TS-61ujJR9BGFT_Iobi57E9ysJacqD8gr6D7DxCQ,35275
jax/_src/typing.py,sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI,2892
jax/_src/util.py,sha256=k-_R0b5o11PXxCqjjeBRIEOrWxd-1xCt-Z1CodGvpW4,19093
jax/_src/xla_bridge.py,sha256=Ydj-o1iRN9jX4icz07fHnzmL54_ET0hRp0W5XlAz0tY,35654
jax/_src/clusters/__init__.py,sha256=zGB6xLPiYJsM-zWSgDaofQTdP1Ot1vPXYk-JNzYXZ6w,1161
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/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=0rEPhWjcCaiMey4sAuDivG1H5RFqqlfesAxlZZ1rlRU,8369
jax/_src/debugger/web_debugger.py,sha256=WJzG5FubQxA9EiuyXzHUb_v2_2GqvovyQiy6EaBomY8,3282
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=VBDNP08YLWvuUdt1HBVx8LUcBi17jXScwuTQejB7wMk,15299
jax/_src/internal_test_util/lax_test_util.py,sha256=LBych5KKJ1fYfxHTk9YyyiOuVGljYYyngz1Ckp__BNY,12148
jax/_src/internal_test_util/test_harnesses.py,sha256=NYaAAGVLkjAsAlanZg0zoDBxvijIgbOr5ZJ16bctzCw,124463
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=xDUb8pLFQ26kNXDG2m3NPFJaPtWV6uPiytpe-YG3y6M,33184
jax/_src/interpreters/batching.py,sha256=H7MnzegdiNwC7y6wA1gR2ZnVI7CExEEHWB4Poaw-2vA,50439
jax/_src/interpreters/mlir.py,sha256=oLHmAhdREbUfwUY_CcnEZrhi7r2_Z6dxHIf5jit3qQw,116393
jax/_src/interpreters/partial_eval.py,sha256=FrDNGiiPsKwwPFQTbBdzl9QD0rm_mDukt37xP6tDv14,123717
jax/_src/interpreters/pxla.py,sha256=_puViRWrgIxeDIdTbIkPa3g98thecqJYjDLCm5TyFpE,130566
jax/_src/interpreters/xla.py,sha256=Q6VGpfpqcmWU_FiEF5MCH0wq964jU668BS9p_L7xlcY,12067
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=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=izHxbi-S686onnDHQC3WRuu9PbKhDhYkZUtZPx6LxFI,10151
jax/_src/lax/lax.py,sha256=1A9-bxzwEhSE4ADoNzWbi0bkyDVdEvepHxFNbfj0jLM,208280
jax/_src/lax/linalg.py,sha256=375XnjyoI3eK7a7RUX1k9pOkq_NDt4aKtnwtHeouRJg,90481
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=MnZtzS8EvfStX-qOmOfLDvG49iIU7ZFRec5ycyN0_HQ,111667
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=ns1oFXqOUH0h6T3VXjuPT30mRCOodvJFJhGr1YgXaxs,9838
jax/_src/lax/utils.py,sha256=nbLjW7RfPRdBW3AXOMk2h7fza5Yx5wDynDzhnavA0lA,5006
jax/_src/lax/windowed_reductions.py,sha256=GrCL0bs6njK9irDC1x75CT5TV2F2IbZ2puEbf_MHV9Q,40056
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/common.py,sha256=pFYEahOuMuxangIgcPk_SxAIATpZ9lBr-KikgkkOJH0,10984
jax/_src/lax/control_flow/conditionals.py,sha256=e3QzJ9y17wmiM8xnVJLg-fl-FbGacUn2p_7bmQPNt9o,43107
jax/_src/lax/control_flow/for_loop.py,sha256=NA8wvsH48E8k7ey-pcQa0pMi9CbwVLfZbRaCcireSD0,37135
jax/_src/lax/control_flow/loops.py,sha256=XiUEfMVodlAKxSBTIS1B2qgWRWYRBpPiJKIvl63I8nw,105003
jax/_src/lax/control_flow/solves.py,sha256=MtOEjg9pa0NTCfVkPaKlfEobIViNhOvKfvyv0CsR1g4,18874
jax/_src/lib/__init__.py,sha256=HSyqGdFIAIFHVBk45EIzsqK-M5HHqza5-5ubmXqmipY,5476
jax/_src/lib/triton.py,sha256=PLazT6GRms3NHRzcYAZq9nmJBKs2VkycCmlN91LyAmY,930
jax/_src/lib/mlir/__init__.py,sha256=vv85BXVE6lq8QZTUZtOIFPrwcoIWROkeiAqVa5a8SB0,885
jax/_src/lib/mlir/dialects/__init__.py,sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs,1195
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
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/array_methods.py,sha256=ISRBbYRa3HcXLjMD5egv5Ie7khsn5224QRBw-E57Gok,32582
jax/_src/numpy/fft.py,sha256=TYlqYNNq7R577HfrrfS3pTKKoMXEgwBqg2ble-lgMXw,11279
jax/_src/numpy/index_tricks.py,sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g,9888
jax/_src/numpy/lax_numpy.py,sha256=-wjg2R8E6WbycQq_AcnwPqwFky3uyXx097hKdMsd53Y,217811
jax/_src/numpy/linalg.py,sha256=eIic1p2dXdrkp5yu5llTswSx-SttXxLc8nveeO-FoiM,30592
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=KxIOmUx3kuIp8RpOU3VUcghjZ8HtAxDKzMyY641nX7w,15256
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/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=Vqv56IkC_DNIC3Qhr_366KxURar6ynOyxlXmcsYTOws,17437
jax/_src/ops/special.py,sha256=nEM6g8Ha6-LY9Ai7gZaMAH3LkT6Xj362V17yKVoP8QA,3884
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=-YqaR9QL_Xr4PV7hSxcAP0kTOlp4YUb78oZ3tDwWxIk,11747
jax/_src/pallas/pallas_call.py,sha256=Z5F24eBlusQk-rH-SqAtIj6jXC9Bpb6eVs1EnqqysOc,23327
jax/_src/pallas/primitives.py,sha256=yXWqNmRYQ7neNvSQpUfcflW0s-S2klyr4c8J65twV-I,15805
jax/_src/pallas/utils.py,sha256=MzAZ8ZYNBGlPPSXkUTlUL1MhRI_dz0Vse4YqIpwd56M,3115
jax/_src/pallas/mosaic/__init__.py,sha256=csumg1HwCYrch7RDIUKm_JA7ILvhazphGrLFfBR2AJY,2384
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=g529ud-jZsd6u9sFOkngqnQy7JyoEGVI5zAYe8RC_kA,70986
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=kudmiRRezV7AaXo4TyWfurs6pUdtWxCXYaumjSC0gf0,4136
jax/_src/pallas/mosaic/pipeline.py,sha256=6fc4Kot_EUkbg0aGJt23_mqohIrlu2jdJPCkUQb0GXQ,36151
jax/_src/pallas/mosaic/primitives.py,sha256=Yzua8YInMkVU8X16pv0ZpaMv7zQtzE43xRa1fEmhEu0,15282
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/lowering.py,sha256=g5esFQhkxFqFb07q_IpqnjQtWan3brfD_Q9nZGk8E4I,53742
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=mcKjqvp63yDGDDZTxJjzbkcUAVu9j6piXYf98zV8ffo,6248
jax/_src/scipy/integrate.py,sha256=w3y6b9iwVp7ozcA1GGErSMhBa-DR4QsWPY_YYWJcA2I,1565
jax/_src/scipy/linalg.py,sha256=nsFWrobIuzBfLSFWOVUjd9lHt_-3YRo8gA0MYa6xaiM,41644
jax/_src/scipy/ndimage.py,sha256=viYzxaVZVP3pXG1yBKMSG5wssJUKYDerBxUcXfCT2ds,5362
jax/_src/scipy/signal.py,sha256=IuhZdas7Uv-mk9u_WR3bW4aJQn_jYakhr0bwGtvEwBE,27552
jax/_src/scipy/special.py,sha256=-Hho_ccefglXuNDKh9OTNfVmY0hl-RuDWFrKQZEoKoE,63694
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=lmudnV7wOY8eN0CUT2a7GxA3mhwRGmjPa7cVF2PNB98,1779
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=VGTcZRDye7xJBLFPiIJM1DJdJTDiIPGytz7HFR0o4QU,15167
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
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/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
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/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=HDpnzipYvKzz7-xxu_2Ft56Ty8czVtpxhmNB2mhIiVU,6231
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=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/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
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/attrs.py,sha256=2EvOuHNtUSLA8CHTbr2LcuJZHAgqSrxbBENcRHcwHQk,2094
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/custom_partitioning.py,sha256=7MsUb7WClXwq8rE_0Ovz4lD26Fem2tPCpu_QSrdtR_E,22294
jax/experimental/host_callback.py,sha256=8KbUgU2zVdur3jh-ikMQft1hFwNM2SlMcuQsD-V2vpU,79261
jax/experimental/jet.py,sha256=PXX10GibNrZeoRHlKLU7x9HNISS6xvRv18XiJx-FDlk,26903
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/multihost_utils.py,sha256=1UNkZqSZcu5OiS71W-kojDYro7mp_4YZJ8j_mwKj47s,18355
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
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=qhCHDF_SEOVfnLW9VOQ5YqdBSFq64UF3cEY7TJQ7Xtk,81268
jax/experimental/topologies.py,sha256=Kv9jgBHyIaJXM6hkqOWD5O5OLryzp4O2rq-itGWJPX4,1888
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/array_api/__init__.py,sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA,5364
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=AJET6fKqPfyTzGkoFZ1A-x4KJaHpd0KpK_5aKd4vl8U,2816
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/serialization.py,sha256=J8_ZZeTmQEW7G9nSVvYDEP3dNzpnJDZIorTWNCCVetI,21772
jax/experimental/array_serialization/serialization_test.py,sha256=uhoAlXU_T81ILGqDbL8Uv3LvAnOriZjbaaIbWPBXfns,13018
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=qN7aApuL9xMiR_4sfaGB_c1oQvcBrl1t7oMJcG2bylE,58837
jax/experimental/export/_serialization.py,sha256=f3Bu0wxMA0DvJ8Fw1pMyd7W_xi4k5P_Npv9k3HXddR4,17212
jax/experimental/export/_shape_poly.py,sha256=RU-fIUGQgz9WrGu3z-jbb8XKy8JDHlc_4MOKBg8x56c,78944
jax/experimental/export/_shape_poly_decision.py,sha256=fVV62rDd37AXe6QFT7FD8y0cpeKkEdFIm1ZSs0ylzzw,16496
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=1HJWg-KnRYp_8lIHjJWq6tr-Tt9F02hkADUnB7FaauQ,27319
jax/experimental/jax2tf/impl_no_xla.py,sha256=SIvqeN30ld0PqoGduuUC3JwMxvCPvOsqUZnWxF_M9W8,54476
jax/experimental/jax2tf/jax2tf.py,sha256=LwjGnadXN2PLn6ze9R3xi0F_i6mgrrQNjrbvWZ18TZ4,142776
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=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/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=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=XigjeRlxoZ4dII60c9W37VO8hp3O3GT1fsMQavjH4SM,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/key_reuse/__init__.py,sha256=t4C0gG8ht16IHAJZkE36_NyNIeO25ojiZan81y88MSo,1552
jax/experimental/key_reuse/_common.py,sha256=Au_9Ho0KMpd0ERZIb5R0RsveeJRcVGuGoyMfzranyoI,2686
jax/experimental/key_reuse/_core.py,sha256=S5i0koJ3xrz75u44fWYM8GyM4zRypOvtiO2GL9qP2vY,1737
jax/experimental/key_reuse/_forwarding.py,sha256=6myGDjEQsqxGnEhxEoT8tImcFPcXNqT0fNy3J6gnSQA,14901
jax/experimental/key_reuse/_simple.py,sha256=ksQEZgNC-STKBoSuYHvsImplSLrEjClPmzrfEyHZXOQ,13202
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/pallas/__init__.py,sha256=89QGSQaGLnMguyyMdhiDEY9PkoySHOrCnk6aR8uF1Z0,2214
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/tpu.py,sha256=qN_HbiuRP9pyT2BoOYO-HXednLvm6A9VcLcDhsTMRI8,2050
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
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/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/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/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=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=7uqEUQ3V-Jbla7xtUS175HZYzgDvwsbh4U38XMoG-3g,36927
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/extend/__init__.py,sha256=67qhpZPLPofDNC0vKaWo6f1j3K2kF6nZj2e9X8UrWUY,1225
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
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/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=TRT9OoYUY2UWB7mVOaoorZ364OJl-qmBqnCIPtybeNM,3457
jax/interpreters/batching.py,sha256=jjy5WCy9pS2mUzdd0fjKViSEPO6L0e5zL5gdkgFR-eY,2582
jax/interpreters/mlir.py,sha256=seU2TW3QctFatISTBuq6Z4ID8CwzBJFFZ_FlmcW9H8c,2801
jax/interpreters/partial_eval.py,sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E,4537
jax/interpreters/pxla.py,sha256=BBH-kEticPc4e06G9kkNL_60BKlhkySRUNl3GPvZHoY,1850
jax/interpreters/xla.py,sha256=LBXHtaMrOU_hgTwNCSSaIRZCUFbDlOXZYuq9QzmStvM,1012
jax/lax/__init__.py,sha256=7SEp4p8rDneBEzYdO1slRtUw_hiLgMyqbtqxmkihrrw,10915
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=o23x0ZOvhizFah-idwif160uoE8XivJvphR2-YuXXZs,1866
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=kKblElZ9IRxOhSyQeA7Iy7fJ5_WFnhGhZmkmGwSpG6E,11346
jax/numpy/__init__.pyi,sha256=ioZTc4m_LGz7AnUfrLAel_QDKMvckeL4xcVzCD7oT7M,36992
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/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=TLex5rKvFTCp_pvnmiHFuKWWytX22_Skh6RW1d_wSSc,1328
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/special.py,sha256=jdGeirmuJrppj9EFta_eeXa_YEPK8Ma5OGNZMNVlY6E,1568
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=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax-0.4.24.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.24.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.24.dist-info/METADATA,sha256=TRcXY2DTdGW2_27ehhoOKtznN0mdI6sQhzJ2WAM1ucY,24288
jax-0.4.24.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
jax-0.4.24.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.24.dist-info/RECORD,,
