jax/__init__.py,sha256=0QUDGKlyOA6eKMcydFKPy7mi2Vepg8hyIKqyCzITwag,9441
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=XVCIPglqBadvnmZz0rWy13_wxOVoR1rf0qVcUOx1dqA,4716
jax/config.py,sha256=eARsQrI4Ti7e6y56r_dAj6BJfd5qS4BhpI1ccBdLf5w,696
jax/core.py,sha256=I-pmIH_D8788fqp4vcXbY1e3ZEoI81FE7Q_qAtjvIO4,6502
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/linear_util.py,sha256=vH-b8qwOfaImoYX2kFiJlVfoLFixU9Q9W1_0b-FlR-Y,2453
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/prng.py,sha256=xjrTOsXwDdOk2p83GGN8ZV99gqnZffYgzYdUlYjUTXw,2395
jax/profiler.py,sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY,1119
jax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/random.py,sha256=ETEafqTeO0MnK-bHCzhwW1ql1fMEoHWafMj9Q8cnzzs,8405
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=DVluJFPiUUmksGXpXdOfbItu1b--c2GcOjjf09ZMcXE,4139
jax/typing.py,sha256=tpCqNUt1m5JUbkLUQeF50O0DOonvrCahZuqgiiaXTxs,2996
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=mPQReTd5ghmcNOf0g6VxBocntgArD9ZQX-30wJTU21E,5008
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/abstract_arrays.py,sha256=ZpKoD3prFnp0yAECVMMAA99F4v9MJxnPOyC7Exkr_gY,3203
jax/_src/ad_checkpoint.py,sha256=QGT-3WVa6B6-2MWCIqaTl4_DAnp8iAbzJZnAbUkh-V4,34700
jax/_src/ad_util.py,sha256=BBNIshxskhvZydeapb6UCNhnkmWBnbJmPF88Y5T4jMQ,3966
jax/_src/api.py,sha256=mrwa7lCWkNVQur0Ox-avOVU1omAiLHFc4itzYjs8bGk,131674
jax/_src/api_util.py,sha256=6B1lRAVwlioU7jjeWgs7XoNj-L6b3qzxbMZHYeRRvIQ,26503
jax/_src/array.py,sha256=2jvpGH5TxmMfEmIhCWjyjlKPZ19Q3-0rJIZqA24mp6Q,31349
jax/_src/basearray.py,sha256=dhWhN_7PfXHomkAo3NdOTQ42bNB_96y9t5Q-XuqJ8Sc,2665
jax/_src/basearray.pyi,sha256=_b1-7At-gMEIAE2CDacw_f5NrY4HbM-KMkUo-w0f8kA,10608
jax/_src/cache_key.py,sha256=IDuC_omg9P9hBgNapen7vdm17LAV_51qMi36Wr4Lu2o,14632
jax/_src/callback.py,sha256=BkCdeSdi1ku67pkCp7sCIWTZ3KifUzdOXKwQCLFXKrg,17841
jax/_src/checkify.py,sha256=DAL1hLoLvI2YH0q2zKQGuS0iQKdcpPPrO9zhkKb5eA4,52863
jax/_src/cloud_tpu_init.py,sha256=mSt_RTDt9Qnxa7v6yHLEFHAFaIGBmytgYQdrXiN6iAY,2698
jax/_src/compilation_cache.py,sha256=OxUQHFA_2zcWarAx3oKchPHpUxE5OFKc2YPP31CzSLY,5080
jax/_src/compilation_cache_interface.py,sha256=xRXZ8RjI3kLnCDAxkBh8jvr_cN3e9Sk9E9CR1xkbrpo,831
jax/_src/compiler.py,sha256=DXm8QIk5M_b-dKXsZDm5ReUmfUGA5RBcliHKGDFU76k,15859
jax/_src/config.py,sha256=fIaor4lGKoFNVj9E6STJnoYNBFK73HbqvcrIHTwZ6hE,52541
jax/_src/core.py,sha256=dk9GKYERhomlNFJbYInry2XbEqtWHIu-yiX-IpCCdR8,113577
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=N690CQU4mr1AzJMh1aFjkS0Qjh3AFj6_rzaZPwn9yCI,10475
jax/_src/custom_derivatives.py,sha256=W5ICX8xx6qzTARg4HLdys3BCguYj8Td1SgyRu-mfDLo,58835
jax/_src/custom_transpose.py,sha256=ET_Nq_yDITag-JYLbR2eFzs-xVMs5LEN1J8suy5GglI,8287
jax/_src/debugging.py,sha256=e1Tbn2p2vpeBoE78nT-85Y0CVTXKjP-dv5mHdAYvh88,24353
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=s5udoYsXMrIHb3kfix6GKMoSoB5XhRVFvJf10G45lVo,20943
jax/_src/distributed.py,sha256=TsIOiBqijDIZpfwuRy_5eHUdUGMQahaeaK3sqaUBqQo,7930
jax/_src/dlpack.py,sha256=qpZ45B2c_HZvf_NwOi20iQ1c7WMgAi2GQXJE462kbMk,4702
jax/_src/dtypes.py,sha256=vyzyowvXGA4jGF1RyQ6ac1XPrfcnbxOtet0y_gzPvGM,26632
jax/_src/effects.py,sha256=fPkC1_SzsPleP2CWiVeOYsPCdElT4SkfUASwdtvhulU,4671
jax/_src/environment_info.py,sha256=F5p2xq60p9BT36TStxLxLZoRPojHmEFuDur4zj0zl00,1970
jax/_src/errors.py,sha256=jJvt8vwEuR8Ie_hdqjBqSwuy5zD74nkIXvbT_ttPEso,23777
jax/_src/flatten_util.py,sha256=jzKbWVjG04I3Xwltjw7rOxg80FjuZLhZoSD7QftE18Q,3557
jax/_src/gfile_cache.py,sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY,1923
jax/_src/jaxpr_util.py,sha256=XziwIzp02nBzpsWS4tGnBMHR3syP5d0QEzLhcqOdjkA,6528
jax/_src/lax_reference.py,sha256=jAUWKbRz4u2JkMgkUgFHwykAeKBt_Y1ZNWmJrSpfjf4,17312
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/linear_util.py,sha256=cjdGvOJ0cyE8wITxEkB2CP5vg2DiLuF4TsOWqmf25_Y,13769
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=Bt5RhneUgAs5MCZ3w7N3iW_oDD3OW03_5Fb2LOnelBY,86643
jax/_src/mesh.py,sha256=xVEkDZB_UT9_YK-3SIfLSssHZAFrlLDXqmhm4nBqQ8Y,10991
jax/_src/monitoring.py,sha256=tc8WoS3cZLumoCCg5XnJs_R2NAAqolvwC7T1NRACE20,3882
jax/_src/op_shardings.py,sha256=QYjptRAT8mGKgCbLQB2hmgq9G4yotIUBr7C3k3WvuUY,3877
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=JsTOrMqyw5uP-ANPdFzT1EGhhUtRjarqneGlwx9pl_s,90281
jax/_src/pretty_printer.py,sha256=6cHfrLAvpyz1O26zfEJWJaUV8U9EqNwGiAoPFPQ0epM,13088
jax/_src/prng.py,sha256=rugwtkVEHbj0kF26jgyDduX0odtdaCNU2tUq48EAkGs,51009
jax/_src/profiler.py,sha256=8s6pWtBsFhJT3E_UlqMxgQud518UmJufiyOIXi4Ap1s,13918
jax/_src/public_test_util.py,sha256=9cWntWtU3Io5-JKOo7p9dZ1nts4myeutDjoRl-8-k5w,9917
jax/_src/random.py,sha256=y1S3CmlfwN84Nj-HFZwTMPwLBv6gOOjY1Ra-ITU2IA8,92804
jax/_src/sharding.py,sha256=T6r_pLVJ8mApjB0br4nNSWj7x45nWJc3FauyXOHgcd4,4921
jax/_src/sharding_impls.py,sha256=inqm8W_9VYONBYfw-RQeklZiksb_TGeBwGaKQW8lYGY,51032
jax/_src/sharding_specs.py,sha256=UVydUNzQnQbKBx5sUTMdjr5Sk81DYWwt7swmsUS6o7Y,13532
jax/_src/source_info_util.py,sha256=U_Uq9Ax2u0HGlNRo2t21Vr1aT8Ei2UI-kiiDkfANmhc,8973
jax/_src/stages.py,sha256=dlx5TEyB3O0zzu5ke9TOnBtPZV4QGXiB240Je42jMYA,24468
jax/_src/test_util.py,sha256=HZeHZ3rOXY2AcD0itdAulQnniSXgBDGshMgfdZBUWN4,46412
jax/_src/tpu_custom_call.py,sha256=sBXS2oHk0JyswTu9Pd0zS57gsaU-yLyiJR1hwP4AZ9M,14484
jax/_src/traceback_util.py,sha256=z-lRcmKrAUXPBe9uw5sO8Ibbw5qulzqCJ0a5yrIdgKM,9042
jax/_src/tree_util.py,sha256=4nJlKrkEJe9-c71iceIam7H8_SHq0JIzpyvH3-6gYzg,35994
jax/_src/typing.py,sha256=zyjBa0jDgfuweBiUDdutk3x-7HIbMtNchI2ZOm97-3g,2740
jax/_src/util.py,sha256=YEmjXPwsizFI7DKbTDL1VZdKkHpG7x6WXlQGARaqNP4,17313
jax/_src/xla_bridge.py,sha256=xB7qWe1yycs_3Ya8xBaXrYPf1bAfVffyKwa9QSeK01Q,32427
jax/_src/clusters/__init__.py,sha256=NPJVuAPVR1oyikfhx5dIMfhTybhoyEkhx2pzWyowNQI,1047
jax/_src/clusters/cloud_tpu_cluster.py,sha256=i4_JhJOvkzR4GTejlvxmAquWe0U4dLchthfJPZCNFcc,2456
jax/_src/clusters/cluster.py,sha256=HqtlD38hX_Cs-eBM1gYYPFYVuXni-PtWJejI8JgEGp4,4781
jax/_src/clusters/ompi_cluster.py,sha256=T9UIKSuWvJWHj5otL5apd17ivnjzn1aweTTaUJgXRBw,2263
jax/_src/clusters/slurm_cluster.py,sha256=MZ4YyNkVLGXLLT7QITOWimK_ZH5nobsBHb5zxTs76eI,2266
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/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/scale.py,sha256=jROfJ5tpA6ikcBhR6zbwlLeatvUpt_TKvKm45tujUQ0,13747
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/lax_test_util.py,sha256=MKeoUyHd2ecj5dYutoRo_Jy_kQPjDCkqah6hwP3t9XU,12054
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=OGvNxvIUMK5TA0UGmOPRGNsjZLBb_IU4gZCQbmkOx5o,33498
jax/_src/interpreters/batching.py,sha256=84GFg9rQxZAB95bBMbFrOq6rbEhBw8eP29B2Lu9bwwI,50700
jax/_src/interpreters/mlir.py,sha256=Y3s25YecKCv6bd_6nr-GMHDXkWJEkoeZ5IahepH-b8s,109025
jax/_src/interpreters/partial_eval.py,sha256=TO0_WUrXczYO5xt7a2h6amFivt217D2K1k16FaZFwrA,118315
jax/_src/interpreters/pxla.py,sha256=xVatH9l3SpP5e9KCeMznOsvdMrl_mHky2FZSNAo1tn8,127738
jax/_src/interpreters/xla.py,sha256=U0JkJytgmkx0W6APAc5bXx6XWuvdzt1OM2CvyjKkwYg,12099
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=dXLjGxxG-jUOmvcuGHe5ETbA21g8vxIA0tJLtrH_EaU,17633
jax/_src/lax/convolution.py,sha256=422HulAiRS4vvdMRWLOzFaTLWIJ6nuZkct7AzWJklPA,44732
jax/_src/lax/eigh.py,sha256=tZCLdbrArpe8HskzEPhyEV_qSewLEbPJqc3ebdzCSqI,19853
jax/_src/lax/fft.py,sha256=CAQSoUNKnIKMFBCUG7CWt7jAwugqYA62yH-1KXa73xk,10360
jax/_src/lax/lax.py,sha256=6dSTKqFAQaMOF01SfjVZaDPi4jKOXvUqikj0XkmNKUQ,207174
jax/_src/lax/linalg.py,sha256=hX7G340WQo2bTuvzr4_Tu_iEmy1F1k0R6oT7SqV3znI,90965
jax/_src/lax/other.py,sha256=xiqAD3ZNXkIs0IzslMdhCoEOi_9xvsRzRRkUg0TAr7g,10554
jax/_src/lax/parallel.py,sha256=RWfksWwXrQ2zLr3AUXPYN8slEr6Xqo3GPkvnPotomnU,73078
jax/_src/lax/qdwh.py,sha256=dtvODeExB-lk69iL3qr_55Xlku5nNVQjFqYz-1EDa4c,8023
jax/_src/lax/slicing.py,sha256=uYunf05gjvSQ_PM097tznJQ7sLV3OZIVZGX6oUTp4j8,111760
jax/_src/lax/special.py,sha256=M9T4Nz31xdwZGYhLMWBRLHHCBRMTR4cWcTPE7CSdOzA,26954
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=7qpha_UrB-6aJjwYY2JFf-U8GqTqa05flWipbWSrSBc,9833
jax/_src/lax/utils.py,sha256=nbLjW7RfPRdBW3AXOMk2h7fza5Yx5wDynDzhnavA0lA,5006
jax/_src/lax/windowed_reductions.py,sha256=MLcgS4lOBCoPZ5pGJjWx5W_4xK_e9vx4NK5ScgOVdy8,40184
jax/_src/lax/control_flow/__init__.py,sha256=DqqlTNYq_SVE-LIAKK8I13gAmKi4OdnsA7FMihqhz0c,2029
jax/_src/lax/control_flow/common.py,sha256=qxOTlGoIJfo0qpgCDsXWsNwL_8loTFlU03y8UW168t8,10272
jax/_src/lax/control_flow/conditionals.py,sha256=4YTy4WPirZMiuETD7Q5gT11FPsRE2LdCXCPd_XVvp0o,38169
jax/_src/lax/control_flow/for_loop.py,sha256=cXMp6k6k0QSdFyXMJ30mkeMzJZKCdqm10YXM3mKBlxE,37033
jax/_src/lax/control_flow/loops.py,sha256=Jlxg5fPSDr8xSArIeBY-__3xdW7canmDjhtAuAPyeug,102256
jax/_src/lax/control_flow/solves.py,sha256=D3PEw6MsfWehqEV8bvW2Ol3zyFO-funjbMzaJR3ArJ4,18898
jax/_src/lib/__init__.py,sha256=3SJxGB5CKgCVvEwqQPkqysF-OxxuhlBIzVoTmXeXS_k,5736
jax/_src/lib/mlir/__init__.py,sha256=BiNbgNL5EDNE2U2PR1awdC8Hg-_G9FGVcAVmFTgyKoE,678
jax/_src/lib/mlir/dialects/__init__.py,sha256=5_3i2XisZ6FpQXmxTUP9zTrTo9McUch44_DLM_EdUyk,1358
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=y3G9XAAObBkE3Dg3-Hkc_hwTX413hyxff8VVDvRLNow,18408
jax/_src/nn/initializers.py,sha256=4mSxtH-Tn5SeyIfRGSYy4i_UgI1G2QyLbsIOwcF3ekE,25598
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_methods.py,sha256=Bv4gSr6lSRZJATixtPbTd6ATumgTig5J7_HKfEqsTBM,31852
jax/_src/numpy/fft.py,sha256=QW2yjWa_UM8YdDgdC3VA9Weic2ziYmmkA8dKvf_IqYU,11347
jax/_src/numpy/index_tricks.py,sha256=MtTMoLMQl4g6G2IM3S2sFMYLcXNnVUIgXPsth-giBMw,9896
jax/_src/numpy/lax_numpy.py,sha256=XLbHRLXG8aZHZRMLkiLPNQML84P5nFD9m5u587Ldfx0,204674
jax/_src/numpy/linalg.py,sha256=r9K2hxT5kzK5SZcmXgC-BC0PxtyAOTKEtshXaWx_Wn8,26570
jax/_src/numpy/polynomial.py,sha256=lQIcmiTjlC3XGVcaFbglX5zjsunI93PeUP3wxT6bfnc,12126
jax/_src/numpy/reductions.py,sha256=7Y2Fn7t1mubVBRtgYuUlS-tQPbEKD9GCcwdmsgkiouI,40935
jax/_src/numpy/setops.py,sha256=tPFFHXWt7fxGlNcq_SjWqPrZ8TvpSXrDNULIoxe_j-k,13736
jax/_src/numpy/ufunc_api.py,sha256=qkJ50sEk4pf4KiOpBnmFppaJGxcpRz9jDhKyhD3ADMw,15093
jax/_src/numpy/ufuncs.py,sha256=ON9Wie_N3DODgAZ4clmqTtVseYWoOL6_ow5FMyY9HEs,27256
jax/_src/numpy/util.py,sha256=_Nasz9KXRSAGeD7kkAq265Ot2Ac5QXwb_b612YhSNf4,17913
jax/_src/numpy/vectorize.py,sha256=SDNE1cW6Mn0h9_Q-PkDFKqhO4OcTEceoCkLH0u4OxLY,12120
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/scatter.py,sha256=8o91_nuqUV9Mpfpj8GzzpeD0fQfqCdSm6oicrPf1MKg,17496
jax/_src/ops/special.py,sha256=owD6DDN7gaBOK6JviavWyaiow9UFd0_ywrIha22ovGU,4372
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=xPxeWZwg3K61H8zEfOrWQeLrX9V1LDs1FCbI1CUOisA,9052
jax/_src/pallas/indexing.py,sha256=6Hl3I-3yzfZdmxruG6g5mw4pveRGeL3zd9mdFIMYaRA,5771
jax/_src/pallas/pallas_call.py,sha256=hooIHEC9T6aKPQF7_6dy3NJhDodJZKNdrQxdOIOagUQ,17569
jax/_src/pallas/primitives.py,sha256=FCyIu74afFtRC3XKj2CeZ_P76OD9dK_NiU2vrV_OQw0,17106
jax/_src/pallas/utils.py,sha256=lC5OPHzZRgHfeV_FQvepg66kXQeKf8FzgkDMX3Ordk4,3140
jax/_src/pallas/mosaic/__init__.py,sha256=khRLuAeXRvKYwez0MzJfRFs5Eqlu5KEk6QW_8a7TOW0,1818
jax/_src/pallas/mosaic/core.py,sha256=yWuKDen45jy5rqKsBFwHndILU7AnEzTqN2MZ7_vLQq8,6486
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=wPx5QR2dK89X9zQBMdJrur9xhWvazl5XsLcqrqsryJ8,57846
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=IJm5vIKkGVILESLceAuOqNB0zXlU7K3aSj2lDvHZXGE,3369
jax/_src/pallas/mosaic/primitives.py,sha256=NYPFaL6O9Gukw1Kd24m369lWMybIAlqIf8Aymnif3GI,11260
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/lowering.py,sha256=ZpyfSdeASB9kdFlRg0YESyFIphnccnoWCK-geMIKQqk,54590
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=c9OoGCT0J9sNnDE_LjGxJU0-zEpphnGMeHMbTCF02IA,6279
jax/_src/scipy/integrate.py,sha256=kHJe7Bmki7VIg5iauSwQtreNyoDt45eIqhOoK9hy4rY,1553
jax/_src/scipy/linalg.py,sha256=i4RGluKqJdlC-steRkEg3NpSpxf8kEuxQtsjImdpROU,42254
jax/_src/scipy/ndimage.py,sha256=uydh28hKMyyxjHQC_J5nEHDC-lYRJ8wNMHU_fBDHxDY,5354
jax/_src/scipy/signal.py,sha256=Y21Mkdo-KP5IqavSWVWDOy46YrEIN2tDM-UzxmY2Ggg,27560
jax/_src/scipy/special.py,sha256=frMNh3YzBYBugTiST8CX-8zv9JDR8z9VZwwAn17jMmo,58135
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/vq.py,sha256=OxDujGx3RvogMoqXKWWfpxsKigzBwxG2AYPCAsqLh4g,1771
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=QaFeCeGTSFtccHubFlWvX2_1HO3TDGuOm-HUqPWnpqc,7827
jax/_src/scipy/optimize/bfgs.py,sha256=MskxYeLfZVfJLEemLHxckhrQmE0tRdd4D5e8Ksw8Kkg,5613
jax/_src/scipy/optimize/line_search.py,sha256=HaKP7Xz1_Ui7efeZMFPyAExloco4kznEANO_bkyPyiE,13410
jax/_src/scipy/optimize/minimize.py,sha256=GAnbB0g-meC0HoPBhDPtXP5T1dp5iE6vranXkCpjGJo,4999
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/linalg.py,sha256=Ev_-7aIZRjnyisSZkBydXAfzZULpg148s-znErsFhpQ,27330
jax/_src/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/scipy/spatial/transform.py,sha256=7D5c3xula9Gp4fam-rFZ8wZW-o2BYyUCU0MDyiB1sW0,15149
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/_core.py,sha256=tXG4oKfZdt_oWRwYeferkxVL89vjke1sejqBJMj12Bg,4969
jax/_src/scipy/stats/bernoulli.py,sha256=ad0-qS___0-0PXi_JSNcnIc8-2kspvmCECwPSCW1lBE,2250
jax/_src/scipy/stats/beta.py,sha256=vSpdgyaZ4gkHKB9uxFFQr5Fx9fC5Ei_X1etBLj7UHPs,2940
jax/_src/scipy/stats/betabinom.py,sha256=kBh8yc_9QaCBmkHcgs25HfG7vpk9MkXYklw22ffbl6I,1993
jax/_src/scipy/stats/binom.py,sha256=tFSUOmAWytQVitfp4zz3cOAMivDNCjVHD3vB0i9IXEM,1634
jax/_src/scipy/stats/cauchy.py,sha256=lfA_Jhs_mfKqRV9dfE7jyj0lnnX8QU29gOoRwghsBKA,3106
jax/_src/scipy/stats/chi2.py,sha256=R5VIaCOsI4cCi1AYDlUxe_NOCZY8b7_WXib9bEd9aDw,2924
jax/_src/scipy/stats/dirichlet.py,sha256=0cp5kgUBlEvAO-B4vWRsa1MFwpHg2aOztpAWbKxLYO0,2136
jax/_src/scipy/stats/expon.py,sha256=zOROY78Y70-DQRA7hrEOcbarXLHhfb_9BU3VPISKmpc,1314
jax/_src/scipy/stats/gamma.py,sha256=EdsviA1SEfCLmsnZv-aBHLV8dZv25sf3W9BLMEYR5Tg,2539
jax/_src/scipy/stats/gennorm.py,sha256=Y70sqdWpQnqPiL577XFHZxTIW4ZC-hq30Rg1MRahxg4,1295
jax/_src/scipy/stats/geom.py,sha256=fDxS432YyqwPuvWNvHwR6CuMckopnFGMUOsNOwrryAk,1382
jax/_src/scipy/stats/kde.py,sha256=m8SFp-zXy5_rZi4Ku90rOh4CTbeTEKfDBMUSsxWDvCA,9840
jax/_src/scipy/stats/laplace.py,sha256=0UxvVoID2Fj9K-UBQI28pGKsaGv7OKLqAXBVU1VuMvM,1787
jax/_src/scipy/stats/logistic.py,sha256=aSRXac2fq0gks2MXUNJFhWwPywQiKu8a8Up8e-29cY8,2406
jax/_src/scipy/stats/multinomial.py,sha256=I7kD1jq3Kkp-EA_0VRRwegazpDKQ-jCn_Yi26_N620g,1675
jax/_src/scipy/stats/multivariate_normal.py,sha256=-FrmiStPBfnU_2zgxyOg5qILmIHzo9YvaBzOoJ3CYGU,2219
jax/_src/scipy/stats/nbinom.py,sha256=DviJZ0ATjbVeONzaBrXOians399z-uAZvRxKQj_nTZU,1698
jax/_src/scipy/stats/norm.py,sha256=7zlhjqHNIjWp0ysIkZ2avQAGm2skdSsLih8sOBq3I20,2863
jax/_src/scipy/stats/pareto.py,sha256=Yt_WwIdmC6mvf8xGwh-SpznyRHgyglYcg64HikBUBjE,1497
jax/_src/scipy/stats/poisson.py,sha256=56Y1k4sOfWz7h361KZHH-eUAx-G9ndPKQX5pMNObedg,1684
jax/_src/scipy/stats/t.py,sha256=nX8aCRT6Muy3GNg36piZ-wmaZwWjbM6N--YN7S1CKZc,1822
jax/_src/scipy/stats/truncnorm.py,sha256=zJoVLIKz1y81wcmuIkfYnTtaC5oQ_4kh7NIIONtgu3I,3965
jax/_src/scipy/stats/uniform.py,sha256=jTU0WPQwX6ohnaeLwyvdkEN5a3Zh40gRJm-k6ATCWBU,1325
jax/_src/scipy/stats/vonmises.py,sha256=NUMe-B35glSUasmZ9Qny8rCB7cgklCTVKJuu59b_suU,1272
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/discharge.py,sha256=cgy--wkPAESVob36QDEwhRomHWRB1wQtGdEn54A-RQ0,33487
jax/_src/state/primitives.py,sha256=biv6QHn8GcyL7IuajPJi6YGFFnpriWyoSAA42TqCyHI,21661
jax/_src/state/types.py,sha256=wzA6nlca63Z9R1eWdFbl6M8IozfOyaSNah07FNBrs_4,5729
jax/_src/state/utils.py,sha256=fn8NOU_WBT04Rd21eWmzecHiPKtBTaX5OqTqgeri5V8,2340
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=TZc4moT_05TmTxRqL9fzP8DyzYBYq5iBGZIgyvtoFO0,6211
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=PMusO102slpONLd0mydFZ1UAHRQsSD3PIkUbgBnFoNg,5869
jax/_src/third_party/scipy/linalg.py,sha256=8epwj40m-vHg56ehkaZGWKQHnarJhQ3UoAdQEqItk-E,2690
jax/_src/third_party/scipy/signal_helper.py,sha256=3lDx1SLvQeDiQgnsCohP1_mPsNyvlT_JXeFBa9Ehpaw,3070
jax/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/optimizers.py,sha256=d7pdstUbzeu5JMI7iCg-BLZaj__OjDdGDMFPl1lLO_Q,20536
jax/example_libraries/stax.py,sha256=cMt8-39Q-o-tWwquR00RAlbbiWaitHLK4P03Xj0Wa3M,13849
jax/experimental/__init__.py,sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4,877
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/custom_partitioning.py,sha256=g3_8sVVyxxkHUr2iOGghXq4QWUs_yw0xQQ0B2B30JrM,21392
jax/experimental/host_callback.py,sha256=dPL_CubY6fZ0nKA1nVD6iYasbDVHCJTycT3BQeUmEmM,79426
jax/experimental/jet.py,sha256=4Lk7tOMSThQDy5zoH3QJ72Qo-s8B6jvEfUUgw2eQle0,26899
jax/experimental/maps.py,sha256=yjycDuNpT_aNQJK_XUBqhDDfY7qHUKeBpfi7w_zwSAs,964
jax/experimental/mesh_utils.py,sha256=H238SsjwlkGQLpzytmB6H5boYMpB5SP8FbNw8IavmdU,15727
jax/experimental/multihost_utils.py,sha256=S3GN1u4YTL0IU8mOfyYwtZeqSS74kqDR5Db2iywoD3M,16085
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pjit.py,sha256=yy8-zVSHV9SJno7hEXmoyWmbdYqG6fF68EVoXs3CX9E,1440
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=o4l_RTRlf3Q2M9OiJl3BFfPgwICOIcgA029qDAQmJgs,19371
jax/experimental/serialize_executable.py,sha256=8wx9DBxOd2SxDYvDUSJQ6WmRc8Bsls9xUIcCOYCU2ik,3014
jax/experimental/shard_map.py,sha256=SIYOTJeltlEHxVZyG_zMlPUJSW59loeV8_DGTEzJaNs,76315
jax/experimental/topologies.py,sha256=IN2LBhwTIYzFZnph7cRnJUHCH1zXREZjLha3g8uVFe0,1864
jax/experimental/x64_context.py,sha256=QXJ2sdwE_hStQ-L5QPbrNVRXws3k_FuzZnWa4iTR_VM,1736
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=cTMJLhwQl-z0AsQzA6PNBj-1ThKdNYej5FxLMgKRNT4,20946
jax/experimental/array_serialization/serialization_test.py,sha256=zECsEosyj2YcggxZYcZTJ_74weqZ-u48X1nkfeNgJpU,12965
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/compilation_cache.py,sha256=FuNWsnrwmZgC00X2-K27s-D1bKIR7SC7gkMGLQ1WxmM,731
jax/experimental/export/__init__.py,sha256=2cJ-OgrtfVES902JD29-I0UJH22lXrSoUepUwWoEYrU,682
jax/experimental/export/export.py,sha256=LL1jf-JDh0rNjSZpLNcwm4uzhhIEJtMUU-UpNMnsPMQ,49024
jax/experimental/export/shape_poly.py,sha256=HYdIrotpIG0yDwmxcHDoAUKGd4Cahqz0I5JYj45h8Es,60454
jax/experimental/jax2tf/__init__.py,sha256=7tXoLqrwpmFJsqrRnkqCg0EqbiIFJ9J0YOnlRGSXbHI,1085
jax/experimental/jax2tf/call_tf.py,sha256=F15-KEoXxAbprK54X2rRlw47_RW3TpxNGsz9iXaMY70,27278
jax/experimental/jax2tf/impl_no_xla.py,sha256=JsALjz03twI9Gg1LT8z5jb4vB5OxuCCBXSXr7uDDgJ0,54003
jax/experimental/jax2tf/jax2tf.py,sha256=xjQ0k_z_BCi4JsGU3HVRoO9yFQSa0eADQn95CgoNiwY,138368
jax/experimental/jax2tf/jax_export.py,sha256=nYLyGdT2HXf6ZyJ01BU1DybnvP-Jp0rnmHfaQFVT8hY,820
jax/experimental/jax2tf/shape_poly.py,sha256=AmkM9n71JwtSaKqWYBb29_q0GCDDWIGieqPmZA_w7SY,972
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=V9_KOPwPOR6cBCAg-piiF3WGlIJ4AaYl2iNIQmnEYLM,11486
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=54tvM9aneRhhhVNyZULCAuX6sjglaZY4xQkBSRYVwW4,7095
jax/experimental/jax2tf/examples/saved_model_main.py,sha256=f4BcJt5KvsNvQrZsKhV4_J6QrxHkppRgu792dy9E2S0,8051
jax/experimental/jax2tf/examples/saved_model_main_test.py,sha256=w6QzDTG-t8f3tdu2a7qfYLoXXWw5bcU7ANJi1nuUIRc,2453
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_test.py,sha256=Huzu1oET8zPRbbRedDtHmBP8puQGPtjVjj-v4lSVxQ0,32341
jax/experimental/jax2tf/tests/back_compat_test_util.py,sha256=mF1eUdwexzQgaL-NApA30lBahwzl-vx6reC6C4okknc,14955
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=aUC3suxV7kXM6o4UBa9jV2tqIg0NOfv4vNPvpfJ3-Uc,5346
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=9bApl6qIy7l9Ib_gNpItWCxTlwmmFwwAOw6iXBfchGE,58962
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=LlIDjrL29S_LCJYF42h9SBU2AmLPH0JNMkh82PfWsv4,10142
jax/experimental/jax2tf/tests/converters.py,sha256=eqmQzNLSxZ3fhRIjJ5VpuRUH84VfnIHqju_na1OFFb8,3528
jax/experimental/jax2tf/tests/cross_compilation_check.py,sha256=l1lxZV6YvxJGy19qqGBQ9HSo5JNNROOIGl4g9EU2OxA,7387
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=agmXkz5QEMt_rBdp2Z8Xzw92e30nmhN8L8L-bUJ8FGI,58717
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=qHPRmtYA51p1buGPXJ6Nd__KOQ0XoAlH9RmZbEdODOY,65788
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=PE7AMvKWKK2T37SyzpNUev148u-17v-LYCANeDGGhZY,6697
jax/experimental/jax2tf/tests/model_harness.py,sha256=4wbCdtEReqIbLozI12nI7ZQy2MhkTK4nJsYl--sXngE,13927
jax/experimental/jax2tf/tests/models_test_main.py,sha256=fX4d_7NWKNdx4hTLriKIFu0udhIGU66t3_Cqduueqpg,8978
jax/experimental/jax2tf/tests/multi_platform_export_test.py,sha256=hBcnQyNhh-piOfp8nQuWK1ULB6F5AOQx4c5eTLCZOK8,4949
jax/experimental/jax2tf/tests/primitive_harness.py,sha256=PQtJZgblHHOSCfYLKEOgl1z_a2zbMhvuiFgjbqOxhVA,124874
jax/experimental/jax2tf/tests/primitives_test.py,sha256=PU_XgJWCKsXmFxH64oCDhpoIcK4_OUVjFIW2mwIrzRQ,14037
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=JHiNoijn-kVZquR03I3VyIDfEdYfB0uKCM21QYTc2HU,149188
jax/experimental/jax2tf/tests/sharding_test.py,sha256=ABlvN86HgS1W0zYVlfqa8Dv7DKabJ9-5_Oneh0uu-_g,30511
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=TJpwrDw-iF4lUDQFWZyxTNz4-Eyp81I8YCU5or-SNcI,20102
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/dialects.py,sha256=fSjIbjy1Ygx6EhF1XRj9EEwX4JEo8K_Ky84_KY6EbW0,958
jax/experimental/pallas/__init__.py,sha256=R6j17GnEqAa_bLaJdpsiQXKCCaY9JxQSCTlmrz3Jz74,2174
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/tpu.py,sha256=T0XyUQn6KPQdI_qYy5r1KfY-s9wtB8VbEGDaW-NHBGA,1563
jax/experimental/pallas/ops/__init__.py,sha256=OOnVMiFg3iFxhCoKLDFL8FzN1Z7ifDdzlxXUbNqem-k,781
jax/experimental/pallas/ops/attention.py,sha256=-ychF4e2tClqJ27t3iGyLOtSeosZxbZs4Ob-XNzvAws,18706
jax/experimental/pallas/ops/layer_norm.py,sha256=T-zec_Kec0e_sLFg0h4sa8TUoUOvxR45CRcNJmBON7E,10949
jax/experimental/pallas/ops/rms_norm.py,sha256=0NCrrGab5ZG0-xlzUbU2ePW3Y9h7pghC6TztM8nmL7A,9842
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/flash_attention.py,sha256=i8cebgDzW_0nExEGxq38DDfcHQFziAO8qUGyb-IwLVc,39735
jax/experimental/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
jax/experimental/sparse/_base.py,sha256=GV1aSqPsxkysLQQUQ1TU2V2sHZejnKD0SK4HVxuqWus,3188
jax/experimental/sparse/_lowerings.py,sha256=4BKVcIwJ1gJqO3gi7_X7HVTmY10HoZ-UnWzqCNj-bBk,7229
jax/experimental/sparse/ad.py,sha256=Rwg1Lz1SwE3CorrypbPOMdOrIyUViPsns3akjtbv9JA,7572
jax/experimental/sparse/api.py,sha256=jOBzLmtMafrEssVTtn6m2wSIdsRP79W7HBIz1yEgo8A,6609
jax/experimental/sparse/bcoo.py,sha256=yBULdSLBx_T_ZXf_uYdKQ0O0VLtqkc4MJ22gCfr1yAw,128153
jax/experimental/sparse/bcsr.py,sha256=_xVe9Lyp0tXaigyD-yRdS2g-dEBv7j0cbTV7OBNG2DY,38192
jax/experimental/sparse/coo.py,sha256=RCDodUA9RJm2fZ33eGWfU5YAgn8PLCmcJ9gSSAdE1K0,24134
jax/experimental/sparse/csr.py,sha256=j7ib2vljbZl4gBJ8ijpLUO-cixg_4A4mK0c1emGcYgo,24422
jax/experimental/sparse/linalg.py,sha256=99utVyAwT2cDeqX3_kxkI8ZVko4pQMcdWaT6TqY3c1g,23285
jax/experimental/sparse/random.py,sha256=HZeJZE9eqEr0d0gAwTQQZJKZMTFitnXSFm0zoIIsXEs,3888
jax/experimental/sparse/test_util.py,sha256=ZuRUmf-6NNjRw0ODd14vGDuW80xwYBxNGpLUfPvYOeo,9585
jax/experimental/sparse/transform.py,sha256=dm3jtQx-NjPXzpcxhT2QqDEtfoalkzLnCyO7xGjZg5I,36908
jax/experimental/sparse/util.py,sha256=7JtVLQkhjvJ6LXg1TpSrFT_tcAzpyL36Ee6GqFnPhXU,4384
jax/extend/__init__.py,sha256=I2_4ko3nKcm7dF5ZDDJ5iowO6JG4gGa_xU-N8hs9BZk,1183
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/random.py,sha256=qqI9s7a9JNBWmBHGNbHDak2swUig1Xx5lS3Djpnqc9A,1193
jax/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=u8bSc_VvMVs4DB8MMb9aO2sAMqqeJUzivFnE8JmMfuU,2748
jax/interpreters/batching.py,sha256=LaCZQ2bg8xB3NdlsxkdI9DmPH0D5KJGrzSl6OYr6E2I,2688
jax/interpreters/mlir.py,sha256=VKlF0B2TAFwaAOFrpWMJYwtaG6xYEaer1OeiXeIQG1E,2679
jax/interpreters/partial_eval.py,sha256=KzL42jZ5EZsj6NZOp8PAigI7m9wGyKjqC9Zdv0Fshu8,4213
jax/interpreters/pxla.py,sha256=juyzYAMMosmHElRzhwuBDXxewUqedLV-ajhhEZfltJs,1896
jax/interpreters/xla.py,sha256=3pIxS72NZCAk9QlEPDp4-ZvthMsilkG5pOvL_POV3zs,4720
jax/lax/__init__.py,sha256=iYTZW1D93sruwVOHiuQ2bFBlbdi38kA9J2uhJC7zKUM,12742
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=n6Ugts8VenX6PmSqtbE3cxWrVeZCVYevWggJFZ2sgAc,769
jax/lib/xla_bridge.py,sha256=rO_3LqPwRtrZFVAh5ZtjGP8Xc2mb2GuREFGBJbm60SU,842
jax/nn/__init__.py,sha256=xEdi9Drex40slz4dJRQmndY9qm7lf0ceu67qNmkKdQo,1428
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=AnDFV7KzKv8nUKpjLuV9ne3CYPS04dFXaaMEV7deurM,12112
jax/numpy/__init__.pyi,sha256=zetKUuaSDGvX0AvorxEYta3H-j6F2qKBXKjB0ZKBb1w,34587
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=1-lGB4PtTA9xr5NZR4SfwFaYM0xGTEG09pZERJEzCqY,1203
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=pd7kF8mgpRnaCxrY_DI2koCGleKYwl4tBSMIs4NMlQg,2012
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/special.py,sha256=emQ2I1-KiU68KK2hu8DUMpY9dbmqaNSyGgwg-Tsl_zs,1490
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=lBmF2s1V6vqb7BiI4j5JDg5teVPHzq13FOt7b_WamcE,1930
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=HKppa0nu4HOg6RFGyV735zhgqX0xsMwYfM0mr_2kQ28,790
jax/scipy/stats/vonmises.py,sha256=nQvzYinMztVFMPZm0fd14ZmHxxoi1CZC_fvRgmBwAMo,791
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/build_utils.py,sha256=g-mMHsDe9GJiqBnruZkXRjm4upkjLIIkcMudplD-e4U,2842
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax-0.4.18.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.18.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.18.dist-info/METADATA,sha256=y4NMV5rdWajT40Gu0eluhNMKBwBVf24zoB9-4A9WdpY,23442
jax-0.4.18.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
jax-0.4.18.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.18.dist-info/RECORD,,
