jax/__init__.py,sha256=gG8CIh3OtIrjTuRqRDKAYNPg3O8M2wscqvPypiftu7k,10649
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/core.py,sha256=luyU1tln3uLnCSOqhZSWVzbJQnOPHcuv2BMqK1rk_3A,7788
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/export.py,sha256=3GnVQzmZviW7c2Rmih_VKsH7F0Pf-wv7RuyalU7FKvo,1365
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=_89L1qOWMoDEgKQuz3ixo_ynMiAAckvkcmpvxKwkZJk,9858
jax/sharding.py,sha256=rdbVjKk_jrWIMSCY_5iN9DmKzrB2ao8lXynYwoREOVI,1743
jax/stages.py,sha256=vjr2-Kn8mia1PS9zorkDVv-C_0E1C3LiDm63K_RLnTY,1314
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=QdjqSbU3XdN-mMOy9N3YCqxQZEne9V2IZf-KOit9jd8,5877
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=fjKLz3ClAqzBI8jMg26cexz5MZgy0fguqkMkPuQZcJw,2804
jax/_src/ad_checkpoint.py,sha256=5EDT8FlZMSM5coulbFzHWyVH0HPVMTffIbKGsN8xNPc,39528
jax/_src/ad_util.py,sha256=ZrB6ynsS1MKqNEwVFTX6pTZafZmJnEkiYZu4LFOUR6Q,3800
jax/_src/api.py,sha256=QgCPxkedFDtqkIlPYZCbmxm9AylkPl_ZmqnUpjnYwco,129731
jax/_src/api_util.py,sha256=2t7Vjk5T9SPwcd9OE9Kag6wA9TusALRlYKIFKprOtY4,27429
jax/_src/array.py,sha256=CFcDWPcswA6a5AGyUB2bfcpiWZp33irhA9QpIlD22tY,45456
jax/_src/basearray.py,sha256=Yz8UYwtmfZYk8BR6-jlJbDI0oaLl06B2p0VU52TA8_U,4915
jax/_src/basearray.pyi,sha256=dh4gHJWEgi4a4AKB_wdNtY_s_39vS5bIK7JLh2lkJe4,10704
jax/_src/blocked_sampler.py,sha256=_Yow0h3WsUkChkB-ySxQ2EMEYnmcIs2lsD5EtpUmUQw,5686
jax/_src/cache_key.py,sha256=0NJqK01RmW0YW-ecBknb0ED7WjvfLb0bjjfrGbDBKlU,10610
jax/_src/callback.py,sha256=IjYHWh_MsQ153Hq_MunfIu4P7RKezmvU2ztmX2gyito,18539
jax/_src/checkify.py,sha256=7dHGcDUOEUjxYVCBJY8vNS3or_BqBQPwMtX5p4HB-Xo,53008
jax/_src/cloud_tpu_init.py,sha256=fATLhPXLPr9HjEuav3XpZiSXZ9APjEqPpDIxJhU4ChE,3461
jax/_src/compilation_cache.py,sha256=sglnaOTtSl8UTFLmp5UJBQUx_609OttiaG7SMHTKH90,11083
jax/_src/compilation_cache_interface.py,sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c,863
jax/_src/compiler.py,sha256=3KFLuI42ck9VZ2W3MSu6mbzVKBDGleSMm3FdUnIu-5I,26691
jax/_src/compute_on.py,sha256=Uj3md0M7BRXm0pKOm_trk0384111N59LjOFZbJe2hqc,1847
jax/_src/config.py,sha256=aYdqATh_AOFdRsLy8FRIceWG9yMfS7radLeQOgTBTQ8,65915
jax/_src/core.py,sha256=RR-2NYFL83D4-KJMf-jAT7bh3AtQVmW0MxoabFuLimM,119511
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=XEYH68cFrOI0Y0ES-8AKYvvUCBtziGMClDnlN3yWLzI,10483
jax/_src/custom_derivatives.py,sha256=A4fokg_T4GPR-yZTtH53roe670lJ5E_apLXPlsk-yiI,60680
jax/_src/custom_partitioning.py,sha256=XQJzp4xRFAxa8i8-pIU8J3ZNTsAHg1AD3j985q-lfrI,22446
jax/_src/custom_transpose.py,sha256=QD6CdGixm163Vy0yUWtXFQ-jJ67Jq8oQFtdbKAoGAgQ,8283
jax/_src/debugging.py,sha256=6pkvknI4hYcsLmNNp17dvM2muvS3bhFn-32zDas3mLI,24929
jax/_src/deprecations.py,sha256=uD6aF_Hlo4ebKqmg7Ee4F-nsZr2FxHMyjVVP7jqPu_Q,4085
jax/_src/dispatch.py,sha256=ykI4JV3cO7se46EYKGN0-48f9AUsPyqwmhhRR5rYcGY,22085
jax/_src/distributed.py,sha256=wTpmmqI-80EZsWzfsJ2tHMLvNkEEIwg6zVCYhSIHR9c,10751
jax/_src/dlpack.py,sha256=VBrJCpLyaftAqahoe_r0BiVlq2ZGB6mx2Q8YQIuirBA,11160
jax/_src/dtypes.py,sha256=maTRwU4gVS3s915wLeuSfuiUdbceUseqmfDmbfRPbyk,30647
jax/_src/earray.py,sha256=tvmElCRaDb5MDECCUu13HtJZ-naN2P2lEKmLQA-YfxY,4307
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/hardware_utils.py,sha256=O7FhdHfejzwVA0uYiKFcQmfthnU93PblEqE27_eoVUY,1970
jax/_src/jaxpr_util.py,sha256=Jv3fcRlWFNUt-WQwHYfmLeXOSKAYka9RKEwrkYl-BkQ,6439
jax/_src/lax_reference.py,sha256=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM,18163
jax/_src/layout.py,sha256=_ZPXV7XN_mTVOpWKgBcLOrF-RbCMhyHLTj378RWuJx8,6309
jax/_src/lazy_loader.py,sha256=XUjGdZvkPKRzuo0755Cmdqy88gVJbJtB4PC2zmtquDY,1333
jax/_src/linear_util.py,sha256=BMIX-Z88yAgEbW5Hmnb7UuZwzEaf8ZVW0guHJ_jmYSQ,13829
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/lru_cache.py,sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s,6661
jax/_src/mesh.py,sha256=nyoI-XbjF2knAX7yrAzt662LYRe0hJZwO7kENB9lL1s,10977
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/op_shardings.py,sha256=ZYx-ASvqGPXifwEBC1utdbAJA-3ZCgWM037y6l4Wp4I,3865
jax/_src/partition_spec.py,sha256=Rlw94T11ETt8EGpQgagi4QJ3sBCuQ4uvg7FLTMl8SzY,1689
jax/_src/path.py,sha256=2HKo4ATWMA-PEz0lkwmCD9oDe4io7wk1juyVUTSsVvs,1172
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=i6sLqOX2vOLJyJtn60E-uZ0n3qBz8hgm7AfCjjKLOCw,114304
jax/_src/pretty_printer.py,sha256=dgxhPpIoPCAuw0e0eKYTPQd9qunJoFlgWyXQSQreTzs,15503
jax/_src/prng.py,sha256=WdIiCjhX9386wFIfKpvfclBvCWJE8y5Uh4L1QTG9S8Y,44587
jax/_src/profiler.py,sha256=YiWJ6-onyV8S5SSOUZgr66qIEzxw_bJnsFyqFQef4vM,16260
jax/_src/public_test_util.py,sha256=8HSKZbFsZXbSsb5d_PBZDcMyEbTh9-DK-CtULw8MdfQ,10573
jax/_src/random.py,sha256=GkXVCIEtRGpyuKz_F_x-HMxZ7PwZ_V9gRI5byWDg9oc,99109
jax/_src/shard_alike.py,sha256=sc7ldeOevCvlBX-XRkO00a8J0EctrBYoWitT9ocbE_g,3827
jax/_src/sharding.py,sha256=Fs00riXcq3J9jDqSV6l8MXdf76dHobw5q5q-FeApHHA,8823
jax/_src/sharding_impls.py,sha256=B03kZS3LDTxHWGMc9u5JUhomHRhyUCPl-pQIkj6MAVU,61821
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE,10140
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=MR8twgBHC7X5xfcG9WnqIaPyBqxUE79v7pem4WhPkWw,28988
jax/_src/test_util.py,sha256=LCXbhuhTJNmHDjiy-xSjgu8az5wzXOXMOo3yvxIUVjY,70903
jax/_src/tpu_custom_call.py,sha256=Y5bSwQk0X90eMTQj5oaOKX89b_iX397Gk9-iehngFbk,22096
jax/_src/traceback_util.py,sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo,9103
jax/_src/tree.py,sha256=pQY3bwJ4puT4TZqPcAmmeAdoeHsoG4EkMwFPN-0iwqQ,9310
jax/_src/tree_util.py,sha256=gzqK914_-D5rFSd1vF9YJk5AuRA7Cg1IMefSuzbd9zI,46749
jax/_src/typing.py,sha256=BlDW38--aUNFQ8UWoi1saofTgRcaNVLgNjG49K8Mlv8,3287
jax/_src/util.py,sha256=3MEkYS8nptzE6oEaPw8hxwxaT3vMHkVW3PyRw5S5Slo,19977
jax/_src/xla_bridge.py,sha256=O8AxmQn_39ntatRciq_b0FwNsxhETPmlbKjL7T-3FwY,44573
jax/_src/clusters/__init__.py,sha256=2qbeIGwB9G7LBjlmYqPV0X22f_f0mCfYKIWMpc91j8s,1137
jax/_src/clusters/cloud_tpu_cluster.py,sha256=9rAgAICnN2NGAh7NgArRN5Kp2Fh1xLN9OCCyFPcKE8I,8220
jax/_src/clusters/cluster.py,sha256=3cxnC1kNeMWfWt0JiMKRlyRq1YrHD199ifPAg4tSdZU,6061
jax/_src/clusters/mpi4py_cluster.py,sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg,2608
jax/_src/clusters/ompi_cluster.py,sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ,2317
jax/_src/clusters/slurm_cluster.py,sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0,2321
jax/_src/cudnn/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/cudnn/fused_attention_stablehlo.py,sha256=_F6V_EJAw9QgCjPXA1E5G3XGRSwtD_CdwOiND3toRJc,38804
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/export/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/export/_export.py,sha256=ZxSFPEytR_yGoYEA76zSdfX9y3R4Bc0lm99cEvvUPdk,61987
jax/_src/export/serialization.py,sha256=GDeQuoYirJqQNDMOBZK7w3aI1MTbA4jbQAjxHGPMpYY,16973
jax/_src/export/serialization_generated.py,sha256=MkKvLYnFffFcwcNfhkodReB33jl1A9_vSPgVxwqQrFQ,26039
jax/_src/export/shape_poly.py,sha256=k3h_sLA9PIq0v4OJ73wYF49YJvAR07egD6Mj41JEYkY,84990
jax/_src/export/shape_poly_decision.py,sha256=RKP2LFk6SD3JaAfroAldWp5IywBb1gMUSSdLSSjSR8A,20690
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/ffi.py,sha256=tP2Q7B2YkvHz0J-lYUfIhYct0aTq8SEnRjs6eXiObE8,10637
jax/_src/extend/random.py,sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw,1258
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=_Hezs_YE3kCj6_cTn6XvMuU26sPiedsiJR8-VVhKEDA,13740
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=Ebr7V03FZtqonF3zIHXImgHljXqzChDV6ONbDktEtRQ,15402
jax/_src/internal_test_util/lax_test_util.py,sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo,12140
jax/_src/internal_test_util/test_harnesses.py,sha256=fV1YNlyP1RVQJwFpkOGASXJDK1voY057NJdlqTuZqic,124386
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=pDD1g-mgLSXS3O3VhAWfoWc6aouWgUA4BWqvsn8WQ6s,32786
jax/_src/interpreters/batching.py,sha256=widzwMTKol8XgoDr1kOHrUj8lFzw2eTVCt4uwPUQ13g,50729
jax/_src/interpreters/mlir.py,sha256=OZs4y0YL8SxELs_Yi5c9OzGczLe9OFEoJcwQDyts0V8,123372
jax/_src/interpreters/partial_eval.py,sha256=jVyAJCfv9gyZ-UODjPBouCVhbTXmmQaq7jkZbC73pP8,127214
jax/_src/interpreters/pxla.py,sha256=S-eO6sNQnPvaaVscX7ls0_n6sPK0JP6sC5xpA4Nva4c,126463
jax/_src/interpreters/xla.py,sha256=Ok2_SEAS1wNYa_ofTPekBZdhbiPRp2-pRmWlHvvPtyM,12187
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=cHNYlNllhOAEiVC5jBsVd-cYb_9TbJNSy9Xgvp7Ppk0,17287
jax/_src/lax/convolution.py,sha256=5dD_ZJjYZh10PXnrO_4LAyC8FjBtD1YKSBA7OlGXV_U,44602
jax/_src/lax/eigh.py,sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ,22235
jax/_src/lax/fft.py,sha256=xrCTzjke5flYvm4anB8URXLiTwHRJDete2s0Eh6qOdQ,7138
jax/_src/lax/lax.py,sha256=sYLIxhyFN4bakmxpTG_eqP3JSQq3928iZhXBPOnTMR0,217154
jax/_src/lax/linalg.py,sha256=vIGbx3InBsoQm8gjW7Z3JdJK9Y18sERw_8aau7YfT70,92901
jax/_src/lax/other.py,sha256=h5x5sZ9Fi3MBT4OR1ioSN2nAia8oQ-S5-rU_OQabkKM,10469
jax/_src/lax/parallel.py,sha256=JtPgS0Lv9xqw4ebVTeNs_282jsW7PW2UQvc6fcDte8c,65774
jax/_src/lax/qdwh.py,sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo,9391
jax/_src/lax/slicing.py,sha256=ELkOvBdNV4Ku3vOneUbS1pPmigmPd6AN5qRDW3oR1g8,111920
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=iVY3iFwt0ug1ow_SkFIJd0lNYdn9eXrCKhNbZ0Dh4Bc,8794
jax/_src/lax/utils.py,sha256=-qUXEzn9AMITKqkVdlKcGEG1wmW6oLpW1x-TiVsmzm4,4521
jax/_src/lax/windowed_reductions.py,sha256=8TLZ-iEuipUTS2kvrtqr9GV2igkwy49m0Y3nCIiKSYU,42345
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/common.py,sha256=Wpr7T_Mm9I3WHLCO_aH-LxjyVWK6h0DtuO_BpMzLlMc,11041
jax/_src/lax/control_flow/conditionals.py,sha256=nzAbpi-iBGVFEHWquPm0f0poDzDwPzFoHmfDga9PkQY,42725
jax/_src/lax/control_flow/for_loop.py,sha256=c963GtflTnjdjFO4XpWOPGGkgNNyrStcmA20H-sC2BU,36042
jax/_src/lax/control_flow/loops.py,sha256=Fvl4zom3d7sCnZAKOMTdi1VcLl6iJImFHJWmZgU7hvY,110409
jax/_src/lax/control_flow/solves.py,sha256=JCyYNtgi_hoQ8Qy1tdyr-UbZ4Y5JhZzwUXwFc6fz4s4,18876
jax/_src/lib/__init__.py,sha256=va0tGglMz4rS52QxQ0ioA7j8R3skIDJySD0tfWX15aU,5392
jax/_src/lib/mosaic_gpu.py,sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4,898
jax/_src/lib/triton.py,sha256=6zzcpqutI1cZW4sG9tLwPFyPDw35xhpPoNH8wkG-VbE,662
jax/_src/lib/mlir/__init__.py,sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM,733
jax/_src/lib/mlir/dialects/__init__.py,sha256=6yV7CMoadrSIsTzrqdGNHnW_36OH2GTPk2Jl4xA0XyU,1669
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=QQANGEm326niUSRn0CkkaMlwouC0N4J0ZRX-HEXgdQE,26833
jax/_src/nn/initializers.py,sha256=mV2u6HrQcvKEXZ_lfH9eL8boZ8sYanQjmjviNQfN-_E,25507
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_methods.py,sha256=y4JYeXyUxa2qr_oWbAk3LzrE3X9rop5mqmXG-WNaPK4,33149
jax/_src/numpy/fft.py,sha256=RW31QI8MeruuvAFASMpnfpzZsXIZmr2tGi9_JBDtRqc,30563
jax/_src/numpy/index_tricks.py,sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g,9888
jax/_src/numpy/lax_numpy.py,sha256=mlYUAlfldowKVmfk6kodxRSIYMrjNmccxk1QW39cfAw,340135
jax/_src/numpy/linalg.py,sha256=JFFJOSO1wsJe7fSGLqFnSLpao-duVuYkCYvG3Q4nFj8,77208
jax/_src/numpy/polynomial.py,sha256=3S7bDP1_up7zVF2AplDkLVo_SMrJvQoF0psDdEE9rzs,27748
jax/_src/numpy/reductions.py,sha256=6a4APGA92FAEis14tcmxHMob665M6DI6uH2Pl_e4b2k,85956
jax/_src/numpy/setops.py,sha256=pzxC0dDtpX6T8JIGOwaEa5vg7D_wS7L-6sHAGLq0ifk,39353
jax/_src/numpy/ufunc_api.py,sha256=q1of6B54ZQwt-yvsl3OgWfgOqmmYrw9brbX6vNvLlzc,15284
jax/_src/numpy/ufuncs.py,sha256=xJRedKM0q7G97wZXekHGdUO6nu3Z5MQTcXuLtkEXRSQ,39440
jax/_src/numpy/util.py,sha256=KlEVXvKoI0kwUSBqys4NujaorH2q-tGF2eDJ51ot13I,18940
jax/_src/numpy/vectorize.py,sha256=CbJlnSuVtH1xSP-_1a-eJNUWaqjbKMzVUBXKe8ctqsA,13882
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=ORkZSW5StWhAnE_ZWHSyvficC304XVYQSV18KTTTR7Q,17484
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=HV-4hOyxyoL_UVqOK6T1Nrf3kpPaJcZx_glCTVZZQL0,27354
jax/_src/pallas/pallas_call.py,sha256=crMlXs6q1zxWtktoqFzRmTTc9YKdcFO25_-aiwBDSBQ,45893
jax/_src/pallas/primitives.py,sha256=OCjge4KqBQw4sKOegn5sT2K66HcOzUstFDyGwpNJAWY,28415
jax/_src/pallas/utils.py,sha256=5a227NBrkjt8q8QNyGEU_QN8EtX6WMOZ8u9SwC9fk7w,6444
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=twfhprNh7VgY01luZ_vT03EXE9VsihQ-n1tYW36z8xk,5531
jax/_src/pallas/mosaic/lowering.py,sha256=Gpgd2o_PWv-NQOVO_ZvAg4PNmzYVg_0GkBoTh_O5Jrc,102791
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=PmDlxcmQ3kqyCd3jxLArjOmTcZkb7ZqzNNA1QQEChtw,7397
jax/_src/pallas/mosaic/pipeline.py,sha256=GPrndA8p_4HtoBkBnhUGMiS6l7KkYQa5NBhfeQh3UQg,41047
jax/_src/pallas/mosaic/primitives.py,sha256=E-RjitscuXUV36Cudqxhp9EpnLq-grZpruH8wA1_9Pg,22920
jax/_src/pallas/mosaic/random.py,sha256=MQ9gPaivB_rEoy1Hm9IkxiLMdRxb_OjAMfmGSEG2bIg,8028
jax/_src/pallas/mosaic/verification.py,sha256=3fUcVwmGTqREWBbevC6T8Dh6SBsJMA2udRe0qgli5cQ,21757
jax/_src/pallas/mosaic_gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/_src/pallas/mosaic_gpu/core.py,sha256=9Ftp5rzGi6w2W2Y0Mm1RCr--SIdb_NZJv9UcgxqBSvs,1604
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=fR7qwJA4eygcire8-avMKmVuy-PbDgScfQF4VnLqWF4,14073
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=Itl_5ztLTh1a-6QR76nBhxTU7MY6c9qY13ubpwJxb90,2121
jax/_src/pallas/triton/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/triton/lowering.py,sha256=VdakiDvPxNLK2A-mNEI3p7isGQ9FZQVhDTOXEbu2oMk,87155
jax/_src/pallas/triton/pallas_call_registration.py,sha256=Gn7WAS5yQflbz59GiFMC_zCPHXLvoGI-wEBc5rUh2AA,3777
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=kUXRvow-11n_mWhdJbTDHErJKE7W1g414ythr6qNKMA,15241
jax/_src/scipy/integrate.py,sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0,2346
jax/_src/scipy/linalg.py,sha256=fyCz9suMWHpcpEE98Zl4jOIdJQT3CdOYPoMifxJ2puo,75007
jax/_src/scipy/ndimage.py,sha256=PouGpJbsYoQvFdmamPUn8XiFoB3ZYqMNYh1jkNyd6AM,7092
jax/_src/scipy/signal.py,sha256=_UG_X-a1kbSAr_yh-GHxT-kNvqsNy1OLhJsjy7BEgv8,47513
jax/_src/scipy/special.py,sha256=QQkhhu6jLuSUKBRlBbSAkfAGIaJ5OVfNV93_lfFUIh0,80217
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=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw,7769
jax/_src/scipy/optimize/bfgs.py,sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE,5609
jax/_src/scipy/optimize/line_search.py,sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc,13235
jax/_src/scipy/optimize/minimize.py,sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs,4979
jax/_src/scipy/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=kVxig00YnjMhNiTkI_fzQNPF9BFn_hbgm9tRcl0L62s,17427
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=dUhvAaJQoGBeTEdmlZdfd8JPI69M-duDxa3ITV6ax34,10620
jax/_src/scipy/stats/bernoulli.py,sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo,4541
jax/_src/scipy/stats/beta.py,sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME,7988
jax/_src/scipy/stats/betabinom.py,sha256=_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=Dei7nhj3j5gB1lBi9q_ZpdMHi1UY7XCr3GTzNuaWF1E,3433
jax/_src/scipy/stats/nbinom.py,sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw,2603
jax/_src/scipy/stats/norm.py,sha256=u43L8WlSayMB-34dSHcdF-MUdIVxcQXTh6AnL-qoIYo,8499
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=VJ8nhLkxKy-5d76UixpVTuOT6svH8w78pt1xWGOov50,37342
jax/_src/state/indexing.py,sha256=N7TVaZZL_ilj3e3ewSqn5KcV0Kg-4kjCpxrsV2Wj5Kw,9183
jax/_src/state/primitives.py,sha256=FCmJUrLClyA49mKKxdnvWZZcZI8Mqk2uMIVod8ean6o,23504
jax/_src/state/types.py,sha256=FITBz17Wi725KfCCJdUvqeB_57tqvTmKgJ6U9zJFbu0,7240
jax/_src/state/utils.py,sha256=lYvZJw5jwTrQxCFqURujcglSyCX8iEKp8bqSmapkleU,2778
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
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=5C9-SwIUrwlpBUgAKYskaSLgTbOtpwESJRPv-6cjj0Y,6307
jax/_src/third_party/scipy/linalg.py,sha256=SzmpCoi4kSnLTflDsRa5QdoBcijKHuwmtN9xac2TWlw,3786
jax/_src/third_party/scipy/signal_helper.py,sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU,3495
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=52Zdn7_FjbHaHUD0-95V3eGqa3l-n_u1r_K0lK61jXc,20563
jax/example_libraries/stax.py,sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8,13862
jax/experimental/__init__.py,sha256=OG7eTYk40QQN4XxZ7E7mFnz7pI2kzo7FH4tU5bteG60,930
jax/experimental/attrs.py,sha256=V5USiVvEd4tfZ7DSNQex16ya9BfozvazNiYFxJlVUT0,9463
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/compute_on.py,sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE,644
jax/experimental/custom_partitioning.py,sha256=fy-UzhOZLDgXbCE0enzQ5Jhnckp_nwkhx6t4poLeY8Q,856
jax/experimental/host_callback.py,sha256=mBi8z2MjREJcFP-ZvjV9WNeXcI500i0bNk4lrCvFAvQ,84986
jax/experimental/jet.py,sha256=J-Y4BjvYRd1xfEX8DVLTJg4rJ0tPKdQa1ymt6wr3-94,26855
jax/experimental/layout.py,sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY,675
jax/experimental/mesh_utils.py,sha256=4tumnl66odeb6xARzFCszDzlX-SCfjUgoOmADpNnQSY,29152
jax/experimental/multihost_utils.py,sha256=9iHzlkWQJoAGZBtAKlAMN2SWcwfX7k2KCK0x-OmBNS0,18592
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=JfyOKcTOYm-78pDjuVhXb4qIJs4LF1f6cOExHQy19bY,18666
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=XaRuekEiKlhvsn7kBULEBmpovY7uchyOM_OwPpQWEkw,89239
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=LNhj8U2ueJhdp9BrJ8C5u6w_Eufqb6H2AuiggOmpS4s,5214
jax/experimental/array_api/_array_methods.py,sha256=_yVJXQNRZwSvZZbTEJmUdUD_JaLTPBd2eTsuXhnLq5w,1185
jax/experimental/array_api/_data_type_functions.py,sha256=qz-WHgv4J4Wj4bWNPIZTf9JLZvKGUAsSYJ-yAdr-T7U,1414
jax/experimental/array_api/_elementwise_functions.py,sha256=H58tYuS4BfnzjhT_c7hfgwD5Q3auvL2B2tr68CA31pI,2081
jax/experimental/array_api/_fft_functions.py,sha256=UsD6PchkbDTnefrp9sYowcOCJ_in3cq9Sofbsja4GDo,1044
jax/experimental/array_api/_manipulation_functions.py,sha256=q5EYezNtB43XwzzyuZQ4wBufh25uSzULaEBmF5eNoZo,885
jax/experimental/array_api/_utility_functions.py,sha256=9MGNfgAs1Q1CxTDvlT50iBFXF1NCGjUmSsxYz3ocGNg,2672
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=UG0-C6e_LE2_o3_qqqnNAB0VxlKMZebALCVLV_QdsrY,1117
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=JIz9yTwPprjX8gDUKXAO9xvngqL7js36fYF7UH1h7A4,26178
jax/experimental/array_serialization/serialization_test.py,sha256=XMrb2dhH7XuYE6deiHqQya4-xLGUw3vnG1HckDTxD3k,22918
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=voUU40tcps7sW-o3oVUbT1yaf_jtbsSRdBDFjUFFQGs,3344
jax/experimental/jax2tf/__init__.py,sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw,944
jax/experimental/jax2tf/call_tf.py,sha256=jxZ12v_ZV9VbMLsrwDR4IABe3r_8Us5B7KNGT1-DkEs,27344
jax/experimental/jax2tf/impl_no_xla.py,sha256=QBCBTTLeuj9uFVbq5nRd_-8txsv2WICDUhbglT_jfBM,54442
jax/experimental/jax2tf/jax2tf.py,sha256=c7XpcILTEBoACNpvPHCIo4XmpcIm-odVKJewVu2MHtE,143741
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=SQEnak_ty2Ggj6jSpLuIygFC5WXI2RdrpZXSY1ooiAw,2852
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=P618Ht7Xrt4p4T5Oz83i9KIGC8Ziiv_-4iQVxiBz4kc,1608
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=O3aRH1uGKFaA8N0lQVdstCHJDFO_hz_qicAA3vrGe8s,11424
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8,7031
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U,8003
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g,2467
jax/experimental/jax2tf/examples/serving/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=n6D9kZ_6fsrNjyfoQDn1vUy7A7DtBPiQjZl70mrF5rs,4900
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=xOP6CFs9sD3yKiI2vkk0W4rPKdB5T5C_4IA1E5-H09g,5360
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=Q_CRaStt395Zzy3UzQemKP_zwet8MGYndj8nQtjD8rM,64722
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg,10108
jax/experimental/jax2tf/tests/converters.py,sha256=6VtplsM_xFdZxfIPzSkDDypkkBZCXPWWwoaBm-IPcM4,3557
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=JaozDkwQO-Aaf1fuooGumLG686s2nnE3OB9fHd6aWwc,7372
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=hCOJAEY-rCb-iw1bAXNoEzCbmjCCQqTYvu7nP9DuChw,60812
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=DV6Irhu8nE0rmtsIpcpzmpyc3ySj-fPvleJ0qAP2xIA,65637
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=h6G_o2ctliPnuSF8a0qB-4Qz3XxXWZx_RXak6mlR454,6656
jax/experimental/jax2tf/tests/model_harness.py,sha256=zhPlq2VnostLmvDfkPdE74A6UPec7wLRS6c_0oFwaLA,13934
jax/experimental/jax2tf/tests/models_test_main.py,sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac,8928
jax/experimental/jax2tf/tests/primitives_test.py,sha256=Eur7lc8lqrJBoDjUgasE2jujMiK3HekY_5dMzIeeGFM,14286
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=aRv0nMM6Z3GNMW0Ubzm_sg0lBEt5Ct5JKukodYu8578,12772
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=vfwA1_XsLXDqgnM10PxBeW8I9TqTV7zQSbyxTZke0Xw,127367
jax/experimental/jax2tf/tests/sharding_test.py,sha256=56aa7sJ_b83WcjwYWlyv5L1iJG4ghTwTusnY2OMHfNk,25578
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg,19962
jax/experimental/key_reuse/__init__.py,sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ,1688
jax/experimental/key_reuse/_core.py,sha256=RydzFNDWPeUXcSdkPnrkW4WXOVtuWODVX2N8OOxzJdY,22816
jax/experimental/mosaic/__init__.py,sha256=jurBAcW7zN1OrDKqkEsEEfWNzMYpvurRg9CFEnIFrJw,835
jax/experimental/mosaic/dialects.py,sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y,762
jax/experimental/mosaic/gpu/__init__.py,sha256=GU_hZXtNNtDfo0NDrT6sItdQiBOVgXMytWb4XZLJKXk,30014
jax/experimental/mosaic/gpu/dsl.py,sha256=YxHxKTqa2cfp4KJJefG64u5AdWbcL9OmNUHTtAPJH8Y,1349
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=2t5dTr9teO6dRftB-DtLZHlP5SCZTHYTCrI2Rzft7FE,27478
jax/experimental/mosaic/gpu/profiler.py,sha256=bLrO_PpxPOod6TBriErGiBHAU-iQztq7anukQtDLwUw,10411
jax/experimental/mosaic/gpu/utils.py,sha256=O8AUecs6_HUrUO7OA66wlk8-M1YPyr2z4zycVQXUsf4,30070
jax/experimental/mosaic/gpu/wgmma.py,sha256=uLysc7XnEBVn_qoIyeJhhzNz9VGgPsV13HkvKEJwTrY,17182
jax/experimental/pallas/__init__.py,sha256=1usEMNjoFP_yHo7q7s35fObl73VOfM_H4fUUDEeWv7Q,2523
jax/experimental/pallas/gpu.py,sha256=sgvpHfp8om5F87TEh7lyxzM_nuMFkfnrsEWmB9ndVU8,745
jax/experimental/pallas/tpu.py,sha256=aP8du3c11Bm0YdwxhwOKjbJeDHfdi-8k0I0nncyAtOQ,3241
jax/experimental/pallas/ops/__init__.py,sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM,764
jax/experimental/pallas/ops/gpu/__init__.py,sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU,581
jax/experimental/pallas/ops/gpu/attention.py,sha256=d3wOrL3tHaAHMeZCcW67DI90bv5kgLGjSspBq4Nma3g,19033
jax/experimental/pallas/ops/gpu/decode_attention.py,sha256=AfSQKqxJbJVRvSETmRFQaoT0yP4CKM0T2IpUHoMSExs,10177
jax/experimental/pallas/ops/gpu/layer_norm.py,sha256=spl0w-aQwpYCvgB-5UaHCaUsCDzeHP7UL1BWbdEtkvw,11070
jax/experimental/pallas/ops/gpu/rms_norm.py,sha256=qaquKh22ka6HMiMZbq2h_vZhzRyZUHjZ4kp92ePohX4,9992
jax/experimental/pallas/ops/gpu/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=hwXVTF4S4Qa0x3lzWIC5Ps7KEuBKm-yNdK-dX1Q3TpE,49695
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=1IHtvUqYVul7oGfAHTE8TRVxJh-TdPBA77kTQp957cg,27319
jax/experimental/pallas/ops/tpu/megablox/ops.py,sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw,2903
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=LbOeQMTlv47YiAYA-eCRb22fY_sUksNMSdCPSkrXw8I,21798
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py,sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss,2556
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py,sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8,2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py,sha256=l6i_Ks-tMOJyMMF6W_PuZRq8BBpQu6nHKHkXxH9mTNQ,73615
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py,sha256=o-9hXrMgy6vq7nq8SiUFPceTGUdvvAsXNHR1bnjPa5A,14554
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py,sha256=smfmy5BKcRUofB-hpca39y63IZQkz01TfhwlzP5jkbA,33409
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=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE,7784
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=bWONVmbz7cYkbr9hev9rZwkrm-knPJmUKf6MzZMwffw,128411
jax/experimental/sparse/bcsr.py,sha256=9kZOvcqex0Q9HbezVBqjGw0KjRyl8bEXQSd97omBtQM,38216
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=CL_JSBTyIa2E1bx8WHBSNdNyrKaRMipQn-gKGt3nFDU,23293
jax/experimental/sparse/nm.py,sha256=_yfZ2HKjZmZUicFG5BOC_sqlKfo79K7nCvw9sU4Ih3M,8563
jax/experimental/sparse/random.py,sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA,3882
jax/experimental/sparse/test_util.py,sha256=bCe34sMR4-ak0sc3vU0CqXlPR9NkUY42FBKmM0sGvxE,9464
jax/experimental/sparse/transform.py,sha256=i6fzxSub8SlogK2eOiaD8PcP9stEKD0WcXv-wGPjkII,37152
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/extend/__init__.py,sha256=rzXCgE1QFpnc0HPRj52vJOxD9_a5LzALYh_bQrYCHIU,1265
jax/extend/backend.py,sha256=zDbInIE5czEJduevvmZjKRB_9Avtnvdkf7fHFYsQPrg,776
jax/extend/ffi.py,sha256=i9_MpO4GObP_bTZKRXf1UQgyjOXyhk8uUqclFyH3a7A,915
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=PcqG5EoHrprDM5lY_Ll2sat6HxHGVm1CbytL8o50TBc,6352
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/sdy.py,sha256=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc,746
jax/extend/mlir/dialects/sparse_tensor.py,sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE,651
jax/extend/mlir/dialects/stablehlo.py,sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE,647
jax/extend/mlir/dialects/vector.py,sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI,644
jax/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=OLDnNDROIGVMCC7-nrsCRBGdzd6lAUXGm7gO_GZYj14,2896
jax/interpreters/partial_eval.py,sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E,4537
jax/interpreters/pxla.py,sha256=B-WvovzyId65My-eOizYtBI5mFvn8GXWIdhqowDRDrk,1824
jax/interpreters/xla.py,sha256=jeBirvMzH6DZQZpY-gxH7EABKsePx_v_7ZF0CJY2hDI,3003
jax/lax/__init__.py,sha256=lZzaemTeNCvit2wSwKkBcsw4L7OMHJhcAVsE58NYKME,10806
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=_lV7R-KvNig4XeNM9ku_fDfPsgdhaLtffFnpaZL4csU,1930
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=Vy0_PEl1cFx6hZ0gEOK5sxWzMBhVTguJXvgWEiRh0oM,11834
jax/numpy/__init__.pyi,sha256=Oh5ET6bjhLCbwAXwsGkvOxv6cITJEcM9nv0ZEzRzaH0,39636
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=1v976Pjx1-03riLwe5hwMurNI7e247hitNB4vIrzXM4,1419
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=kPIlUY6uEFvcFn_AQ6GdvNHTTij9JRag81bza4E2hKE,1644
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=zqgkBIwuwOlCGd-aMwRPdsx9JbykpLYMvfzwEHiMDA8,4194
jax/tools/colab_tpu.py,sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo,890
jax/tools/jax_to_ir.py,sha256=GaLgxfDS4Y37yErqOL5BpoicRsnwMHTeA5ucaUR5qBU,8804
jax/tools/pgo_nsys_converter.py,sha256=udKwSZG-RDhlvA_Z0vGkW92nR1BLXGVBdxTU0xLJhVE,2716
jax-0.4.31.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.31.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.31.dist-info/METADATA,sha256=2gG6U2I7w8EuQ3ZvXHT9aMJAl5u8RPkjaJfgJPhDdc0,22356
jax-0.4.31.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
jax-0.4.31.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.31.dist-info/RECORD,,
