jax/__init__.py,sha256=ols_ejymtyaPV_s5b7F-sHTP8r1RRsLy2HbfngHw10o,9489
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=B6dnRzeKmU8XTeZLlKAT06VQDwoCLw8c7fJoSsKYOuM,1119
jax/core.py,sha256=LBu2Vk67OQbED3zjJO31eHuMeaBVf1ZAoRmrYZZ7skg,10096
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=TvyHqtnUARG9OLpFInVWCNTsFmnbyhVIj8LKhJvVEPc,9586
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=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk,3331
jax/util.py,sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA,1091
jax/version.py,sha256=ZgoC7fkCDC1NxITUg8NihEhM5V079yPvAbG8C9u_siU,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=SpfHdz7feXhMTAgHRQO2RuV9E6hTTxqZYwxmQsUhODw,34697
jax/_src/ad_util.py,sha256=BBNIshxskhvZydeapb6UCNhnkmWBnbJmPF88Y5T4jMQ,3966
jax/_src/api.py,sha256=ykOSCy3uhCGiDY4iRkVJTCeDGzitH93TwJ4au2BID_Q,131446
jax/_src/api_util.py,sha256=3Tff0ihu9bZckKTHFdiSYXzVLAExNnfIEc-ReoZUQTQ,26499
jax/_src/array.py,sha256=DA5u_EZjyBMUpzpP_f52Rb3UdDPlvWy8qSX9lIdX5B0,32458
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=x7llz3h0JfbyVb8VwMjgIihm6NvNIp1S4unVPB5wsP8,14627
jax/_src/callback.py,sha256=yMSu6rJq4fd8FUSjhGx0n6wF-x6G4CeQhHsmOVUoSZs,17987
jax/_src/checkify.py,sha256=myIFji_h4X5pmMUOgy4Vhru9YWPK_bcm0zWr76VG0oI,52834
jax/_src/cloud_tpu_init.py,sha256=HpJitsv4AwaP-hvmgwboEYoDmMXuXP9-CDRYHAEdPjE,2647
jax/_src/compilation_cache.py,sha256=OxUQHFA_2zcWarAx3oKchPHpUxE5OFKc2YPP31CzSLY,5080
jax/_src/compilation_cache_interface.py,sha256=xRXZ8RjI3kLnCDAxkBh8jvr_cN3e9Sk9E9CR1xkbrpo,831
jax/_src/compiler.py,sha256=0Rk775qK16TqNSqAS4FWhCxfdofzu-WPTN7e_ROeTIw,16133
jax/_src/config.py,sha256=el5im20XEJ-enyZNiV90ew5oIRMYE_fkNxIgz2Mj_9c,52182
jax/_src/core.py,sha256=fUj1DD_bJ1xnNYiY2wa6Wp6xZ_cmvKBsYRZNRhbrTTc,113519
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=hb8CkCaHNNithL63cy8PJhS3dqY7iZZmqgsbu3lyDD8,58832
jax/_src/custom_transpose.py,sha256=ET_Nq_yDITag-JYLbR2eFzs-xVMs5LEN1J8suy5GglI,8287
jax/_src/debugging.py,sha256=_hbANOHki2OO7IpFg_bQeKGTIRShWvX6itInTXc272o,24342
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=9U3FXqVvIoYT9ySIiA58zipMwux_5PceX7Vzyu5k5Io,20950
jax/_src/distributed.py,sha256=9CQ78m3z0ss2ZTuLcxWpH-1kVvJyELLSxk5KvcrmuYw,7923
jax/_src/dlpack.py,sha256=Ll_SCkS6Uel434h4GtBzgBAiDJJBgpGq7auwFmw0RqM,4849
jax/_src/dtypes.py,sha256=5gbKf58jwPWbzk15cizqkUDJuz_XpfkXUjGoDHO79YE,26880
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=ZNPepQQYw3Ejbd1Kj9QdO6wUoIL763otW8BwAZBABHQ,13793
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=ptaa-unwcq4_iPp14rwbyL7cJNilzwyRMzBE7R5hSRw,85960
jax/_src/mesh.py,sha256=QKFiScGqtK3aL_TBGKWpTiDeST-mA_ElGaL1FQqbRh0,11086
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=1gVqtBIVlmB0CrQkZIuapxRyF-bfwDCtoualIY9Bdt0,91059
jax/_src/pretty_printer.py,sha256=6cHfrLAvpyz1O26zfEJWJaUV8U9EqNwGiAoPFPQ0epM,13088
jax/_src/prng.py,sha256=ob-d2N-fNXtfARBQYk0ZIxgboxit4fz1Jj6qrSh7dGg,50601
jax/_src/profiler.py,sha256=aIUL_aJpq_MEB9zCko94SdZSbA5CgwJFP0tTZrM8RoQ,14812
jax/_src/public_test_util.py,sha256=rv6aBUJqR25KDKkOB8EvpBXaR_FDK09MDiBrpoiHwNU,10397
jax/_src/random.py,sha256=wa4Tos-igp2EQP5qnVu7BvYEcLY6ahK8uZJdC18iW8c,92843
jax/_src/sharding.py,sha256=T6r_pLVJ8mApjB0br4nNSWj7x45nWJc3FauyXOHgcd4,4921
jax/_src/sharding_impls.py,sha256=sSXthEwpreMsats2zwcukNOg77qlQ0zBZXGT6UuYTLk,51024
jax/_src/sharding_specs.py,sha256=UVydUNzQnQbKBx5sUTMdjr5Sk81DYWwt7swmsUS6o7Y,13532
jax/_src/source_info_util.py,sha256=Zu_cOpox8y_96y7gLBrjLeHnU2YbUhnpRSNL6p7ebVo,9346
jax/_src/stages.py,sha256=EK1oiBxKGjA7esUrxKjpePLbMhP-0eXtD28dtM2DppE,24494
jax/_src/test_util.py,sha256=fvDsZjPIC-RDIiJcHnx5eop9Rm64CeJGN2KDtHJf2_Q,46576
jax/_src/tpu_custom_call.py,sha256=agx4TwcWjF39ClVxQsCuCvQpVncttH6EqhXyqZ5mgMs,17137
jax/_src/traceback_util.py,sha256=_itagW8OvIgDeVHxQ8t8wODF8cgr0FlesvvhEjbG4_4,9037
jax/_src/tree_util.py,sha256=kwPpJRv6HVPkJ83r7VHyOzSs0rhmH7IANql11Fi7ceQ,35978
jax/_src/typing.py,sha256=DlTVTLKHqD4EQcQdEymtO1wQrhLgXQpMfvYVBMaECcQ,2919
jax/_src/util.py,sha256=DjbQJls6lK5CzdUTRpNu5jKy0kGH6M5qSe3RN4A9fGY,18011
jax/_src/xla_bridge.py,sha256=KL5LR_9wZWzMGx7EdxKbqSFZIU0rUneQEq-Iq8TwHlQ,34093
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/extend/random.py,sha256=riZ85VyieKSbGBB1_tqvATQQ5sfIEEDkg9KvD-xUD4Q,1249
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=uwSqVmfP4xI4dDlF6YpvGHTdRAJaXwaXEYyun2wpAfQ,33509
jax/_src/interpreters/batching.py,sha256=DlndWvnUh5f3Cc9Vw14rWmiLnSm302UmcZCGodDz6ww,50700
jax/_src/interpreters/mlir.py,sha256=vsKJwp26JHJeUtaPExocOOcWRs0Ag3GuJnpopQ1N0K0,111031
jax/_src/interpreters/partial_eval.py,sha256=RCAeQm9NHGZmGoltwgfp2mAjO_ucxMRaNTD3fsSBDFA,121835
jax/_src/interpreters/pxla.py,sha256=kXvdjGuvD0fDHqW-UDga9MMPM3N6EEUPgmr8Shq2a4A,127160
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=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=CAQSoUNKnIKMFBCUG7CWt7jAwugqYA62yH-1KXa73xk,10360
jax/_src/lax/lax.py,sha256=dl3sdDNibc6B9VnJM_0eUqcUm1ng1fS5Db_ye89lJBc,206748
jax/_src/lax/linalg.py,sha256=QE5t-gFja_A9gzc9Suj003nrGpiYi5CL61GvSThUExk,92309
jax/_src/lax/other.py,sha256=xiqAD3ZNXkIs0IzslMdhCoEOi_9xvsRzRRkUg0TAr7g,10554
jax/_src/lax/parallel.py,sha256=xo-ur0ZZLIOLHcFBd_dS5tinhaYlYAPC1cGGhiNGGuY,73479
jax/_src/lax/qdwh.py,sha256=dtvODeExB-lk69iL3qr_55Xlku5nNVQjFqYz-1EDa4c,8023
jax/_src/lax/slicing.py,sha256=cecZghCbUaUQC3eSngmDJsC4DZg4x0QDI3l3SkXOEww,111786
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=YYW_KsAK9Dtdhy4Rjlt36bZ9fKA1V-95VCQpN3LU_UQ,38180
jax/_src/lax/control_flow/for_loop.py,sha256=cXMp6k6k0QSdFyXMJ30mkeMzJZKCdqm10YXM3mKBlxE,37033
jax/_src/lax/control_flow/loops.py,sha256=w9EFAf5S7m8wENQEKYO7ANRZDJvc1Wh1FFo09pGZBBA,102243
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=i_Be3ahcF0XuumJrPBPO-Afv9-qIg6JLx3Ggb6aupaw,18434
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=q0FVjR-y-EF0Vf3TD5BEwHhMB-Kgp6gz2TEfiqMOJEU,31850
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=bthDWToBxex2c1IZOGmp-y77S9bqSEBTPHgylYhxSP8,206689
jax/_src/numpy/linalg.py,sha256=w3yvqLrxHDMo9QCPnPmRzsfzRaS4Ek0kjHomnOkJEiI,26587
jax/_src/numpy/polynomial.py,sha256=lQIcmiTjlC3XGVcaFbglX5zjsunI93PeUP3wxT6bfnc,12126
jax/_src/numpy/reductions.py,sha256=9NhJy_8KfWgI8Q3Z4bQMBMLydpZSkBlON4Q8JBSe59g,41135
jax/_src/numpy/setops.py,sha256=tPFFHXWt7fxGlNcq_SjWqPrZ8TvpSXrDNULIoxe_j-k,13736
jax/_src/numpy/ufunc_api.py,sha256=Cvv2hCSoca0IfqdLmd7NWbv8GehTvWugO625PUslB2Y,15230
jax/_src/numpy/ufuncs.py,sha256=wHJr1NktTtsESZ--_YP6BBcYS4GJ59xAZ6Nz0dWmy_w,27063
jax/_src/numpy/util.py,sha256=E0-Ctf3207eQKp8QycG14Rss5nhUNNCJozLk6O-1Tkg,17999
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=Qyej1i7x-y_xTmGhniu96GafWEQxa_5WqejFuO7EEI8,17462
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=2iLjakGcjyzLpOBYgFAhacKghDuA9l_2QtP014DPZiE,9104
jax/_src/pallas/indexing.py,sha256=tbt_T325TzkvLhUH901J5klSgR7paC66UNq_k4VN3MM,5746
jax/_src/pallas/pallas_call.py,sha256=hooIHEC9T6aKPQF7_6dy3NJhDodJZKNdrQxdOIOagUQ,17569
jax/_src/pallas/primitives.py,sha256=E9D3M_WHkOIONhkISbGbQRTYdWCHHaT6cIzQClJNNiU,15655
jax/_src/pallas/utils.py,sha256=lC5OPHzZRgHfeV_FQvepg66kXQeKf8FzgkDMX3Ordk4,3140
jax/_src/pallas/mosaic/__init__.py,sha256=p49OUB9rL95f_HqSUW9IWVnKNR7SWXGnwYGDHtMLU00,1949
jax/_src/pallas/mosaic/core.py,sha256=bjIZHpSGnQ91sxP_pVnPIvJSUg_WOIltp4lpXpWxpfk,7392
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=Uen_LhQtf3WlZ3xa-lk63ctTrySrTq_IY2EC725gj-c,58274
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=T_afkdeXLuMRiHQQBRrqmFnUIhVNfWMoMag2mhkUQLw,3417
jax/_src/pallas/mosaic/primitives.py,sha256=-yznClMvc1x4gGbAb7EzRA60KwiiadF63CmBrlCU704,11971
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/lowering.py,sha256=0Ej-rzBL5q02BPOX3rNBW3O-pzyU_rNhLW7lt4wxzlw,50983
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/scipy/stats/wrapcauchy.py,sha256=VflTPi7PHhWv44rjYHJ4UirzJYXAvMRlrp9-YkgOVzc,1410
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/discharge.py,sha256=6xkLmz64Toz3rZpFTX7HqpfDCWGX2MFrA_b5o8-B7TU,33482
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=Uf7ZCZ0Ary4Mk8Fntut294TvB-uRfbeMMg-tgR6U5f0,21394
jax/experimental/host_callback.py,sha256=ue_pM1_Sh4xZ4iE_MtAvmcI2Y6T_4k7hPbbjDodqHLk,79550
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=QjradelzbyvqxX4HTu3hnJSAetCq9AI4MvfY9RwLqac,78405
jax/experimental/topologies.py,sha256=IN2LBhwTIYzFZnph7cRnJUHCH1zXREZjLha3g8uVFe0,1864
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/array_serialization/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/array_serialization/serialization.py,sha256=68kx3wvtk4iQBhIeG1gzXjMBCagzYkkWtj1Yksi9KLY,21753
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=gcnFME07y5a_35xdYHOFH3b2x2WHMfrJSLRPrYGulQ8,58398
jax/experimental/export/shape_poly.py,sha256=noEZJYs6i7IlzlNwQEBEqJN8lvLxJ1Jc2BVvGj6Y1ts,60457
jax/experimental/jax2tf/__init__.py,sha256=7tXoLqrwpmFJsqrRnkqCg0EqbiIFJ9J0YOnlRGSXbHI,1085
jax/experimental/jax2tf/call_tf.py,sha256=nY_8NrXfO3TqeBkDfqtdNp5APyPNotr8D5f7X8Vk2-g,27398
jax/experimental/jax2tf/impl_no_xla.py,sha256=Ybue-3arwSWyVZIsoiE_qVlY5eEx3poyhkB3jdERslE,53993
jax/experimental/jax2tf/jax2tf.py,sha256=vVL8zDHXJm5LCQx2lyXtSphy-NMty0qYPaYEZJT1NxE,141004
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=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_test.py,sha256=6jw3EmAs54mtRG3XwDF4-nhqY3IlDy2A_WOyj5uOUM0,32201
jax/experimental/jax2tf/tests/back_compat_test_util.py,sha256=jj9fVrxfUNIdRJDuh8kLh_GEB2kjojTG64ad3nfuqb4,14952
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=aUC3suxV7kXM6o4UBa9jV2tqIg0NOfv4vNPvpfJ3-Uc,5346
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=jaPe8RMz8aPyh865HpwNJ-xz1E2mmG_IfW-jQ138siw,62919
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=YVFGwxLKzIZ2cxDtWLZhyM0_roPjY-vuPrlxmUpVJEw,58728
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=VOcl-h-uB4fi87tZydrpxkrQsmV0GW9fuz9-60kTX3s,69144
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=S7brHKZ0eVVEAFJDoB30eBfogDs5wNK00U9lwnOvoI8,6716
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=tmoDLvraYJijG_CiJE-inpWMRwKK48CjdlD5BLAAuQQ,5983
jax/experimental/jax2tf/tests/primitive_harness.py,sha256=PjSqCF_TxpUG8JfmOGA8TtlBZeHwFWYZV7wuc4l6q5g,124679
jax/experimental/jax2tf/tests/primitives_test.py,sha256=8S-4aqSTpbTK-c-8SvQhyqT0gWKhlw63EXll9sU3EsA,14077
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=0-9_nY8wSTu7IW1Coy5iQNURPhJxFxvFo06TjUXnmhk,149054
jax/experimental/jax2tf/tests/sharding_test.py,sha256=KlJda9jXDxJAB59Mk3BWC4k-aGIV6bU2rt4FDkDh4kk,30746
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=1yZhDedVL-L7B4BYMArgKwqJ6mOyEfgkNJpEHF5bJi0,20004
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=gar83kdE9ayr5AyA0pxzg7_253YgrXmxu7ArBS2Jpek,2217
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/tpu.py,sha256=FYUIpLJa8FSyRYWGJajqcCQ4OGnk9FkCGtrjnk3eYGI,1672
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
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=mI1NswSE0KGJLNWz4qCXrX8Zw82QgnqrnQwzIxH58sY,48826
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=-Q7z823t4ayjVf1EJbAmKn8Hb1rmgHKrPlbrPrEBIZ8,128160
jax/experimental/sparse/bcsr.py,sha256=8mHZeKepZmuV6RLMOIp6qE-x_WXJfgKALFgshmG-3UM,38199
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=caS40gmGNAUmK_gwXo1Oin_o7bmPm2mn08f4apTPjEw,36905
jax/experimental/sparse/util.py,sha256=7JtVLQkhjvJ6LXg1TpSrFT_tcAzpyL36Ee6GqFnPhXU,4384
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/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/ad.py,sha256=qVD0Ish5b9uGe6nTAsNQb6ZT8_IW1VvTdNqEbZpcaKk,3509
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=g7DuKIpB_PnkSd0oSz_I5XF4ha-xPYPRGbuuaYftQ2o,12808
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=dYanXY9KQ2WCi3PYmFdx5nc0gCK_gZ_FWWG5JzcJs4k,12203
jax/numpy/__init__.pyi,sha256=c_j61DtyXsdckvA9ooclThHBpJfWUSpYf3jtO0kIROU,34701
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=cg4pntviV7r1RHbnnpKPIysGR-b4f2kx9nTFF6h0gOY,1983
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/scipy/stats/wrapcauchy.py,sha256=aWOW5KZnRAFa2rgHxwTybH1c-gbzYE0KhGVdy6BNDbY,793
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/build_utils.py,sha256=ZSyEDWrSRwVP-XaJ3tQclCy1bpxbRMG0EZOw1zNGtWk,2963
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax-0.4.20.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.20.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.20.dist-info/METADATA,sha256=LzKknRExBt4Y0Kvvone3xuT0F6ad4dQwfSMIhGRJcYE,23495
jax-0.4.20.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
jax-0.4.20.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.20.dist-info/RECORD,,
