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=KJmRViYn41aCmOlgMfD00EIwk-N2mycDtfKN2tipgSA,9885
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=DvCl0rR_YX_4g7xIDEHmeVZ-gE6-pAtGCYAI0g7BhEM,5583
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=KFbGopi-BeCbGjseAnpJA6QGYmZ9gJzjWrXSvZ3tqqw,131506
jax/_src/api_util.py,sha256=WSU09HHe55wMCZ3QOr7HkF41_1StskzCVL6J8Ec0asI,26498
jax/_src/array.py,sha256=7yB5rJZsFAEnsxMU70JPR_g2w7SoA344c8yq7Ojv3uE,33047
jax/_src/basearray.py,sha256=JKXT8RVPJGp68f8kf0DEQPFTKFk41aukOjbsCPQGKP4,4235
jax/_src/basearray.pyi,sha256=lO4QG8TKVdEZuzVnM_8HEI_3lrNNt8EkVcNh8ga4V0k,10581
jax/_src/cache_key.py,sha256=HSC0Aws0kmBiKIs6hYZKrK5yRdQF--cb79Hq7ic77_k,14668
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=DKxl3RBk5nH1GVYBSxIIBzETON4Vg2t4wO_FzVya9t4,6412
jax/_src/compilation_cache_interface.py,sha256=xRXZ8RjI3kLnCDAxkBh8jvr_cN3e9Sk9E9CR1xkbrpo,831
jax/_src/compiler.py,sha256=oszgaxDbLctSFy0AM_epAQBXChERkfgzd90jgL5TXDI,16384
jax/_src/config.py,sha256=dU7CpIqjvb1DwZSVxPD151ji1aeHNk3bw9BBbJr8hMA,52732
jax/_src/core.py,sha256=l8yVZ9HNIb7_F7_7Hx2BpbyMvPt4kmbH4wQP6Nc3xeI,113932
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=o-tZ0s4og_yrN4ExkpWGcrh-8juDGEEaqM1c3pxTmsk,25250
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=hxOCTCGjEPvx7CsIhuhnIXnPSh7Q6v6yxtrEzUMEyOo,17585
jax/_src/distributed.py,sha256=BSqO7JuyZU3vrTOx4ALdN7Y4vhhHvR3gtXJxkX7ThXw,7699
jax/_src/dlpack.py,sha256=o2fHLqpUOTlKdrP8UMpOB9Gx0k-lYua2e3qAD1e4kvA,4097
jax/_src/dtypes.py,sha256=5ksWAj2UITljBKUdCYdDTG74K25G7AF0RFdyd5qhfPo,26501
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=5EhHSvYNWZy9lBjzc6c3AR-Q5H44kcrlWAfqgG-6PnI,6532
jax/_src/lax_reference.py,sha256=jAUWKbRz4u2JkMgkUgFHwykAeKBt_Y1ZNWmJrSpfjf4,17312
jax/_src/layout.py,sha256=ga5-18Pz5gAV2ktKpJJzEPlfD_LRDMvuGtfZSloqNi4,1585
jax/_src/lazy_loader.py,sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo,1333
jax/_src/linear_util.py,sha256=ZNPepQQYw3Ejbd1Kj9QdO6wUoIL763otW8BwAZBABHQ,13793
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=D62EuQeYkARFDG2OBoknUxUdPF3Blx_TU2a6WlcXf74,86268
jax/_src/mesh.py,sha256=ZuHaLBUPdqQ0uG068Q_Qi0URVeLBP_T40BTTHmNyAhM,10926
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=M-6AMKTFoUYiWT-AUS9pn6jucm13quZDzxFoYgiebKo,92859
jax/_src/pretty_printer.py,sha256=6cHfrLAvpyz1O26zfEJWJaUV8U9EqNwGiAoPFPQ0epM,13088
jax/_src/prng.py,sha256=g00XSESyAKqnmB89s2WKgi95o5efx2DO-qlDpNlkTjc,51485
jax/_src/profiler.py,sha256=00mtRO1KOg_pdSsXSjKLNWuRLyzL-1h0sJfDu_35i1E,14493
jax/_src/public_test_util.py,sha256=rv6aBUJqR25KDKkOB8EvpBXaR_FDK09MDiBrpoiHwNU,10397
jax/_src/random.py,sha256=LZCZgM1q_ftd4InZbfOoVZGmR7VCYjNurvjZNGvmT9A,99864
jax/_src/sharding.py,sha256=T6r_pLVJ8mApjB0br4nNSWj7x45nWJc3FauyXOHgcd4,4921
jax/_src/sharding_impls.py,sha256=ZqdBDkihyws_Xc5xtd4AbI2OTAIxX0C8b9-LvEJGCt8,48239
jax/_src/sharding_specs.py,sha256=UVydUNzQnQbKBx5sUTMdjr5Sk81DYWwt7swmsUS6o7Y,13532
jax/_src/source_info_util.py,sha256=0XimzbdwEE8XUwM6l56S6CZHgnSXk8q4v3pA2j5C7sM,9405
jax/_src/stages.py,sha256=xsPwfUBgHK-SQwanw2stYbREG76cfR6liUOdg1-6OJw,25829
jax/_src/test_util.py,sha256=AE1pUnBK42jFuxh_Pcv59J2by8WQNTaTLCxgB1FJe1s,46960
jax/_src/tpu_custom_call.py,sha256=wRWK1rdtQmtsePSAszgsCA-kO148sFuX1_NC3xRaBDI,17112
jax/_src/traceback_util.py,sha256=_itagW8OvIgDeVHxQ8t8wODF8cgr0FlesvvhEjbG4_4,9037
jax/_src/tree_util.py,sha256=NviihrbPpEjIvDrJ-Qa0qBAPH4p8OKYqvDXnF_DC_8o,35971
jax/_src/typing.py,sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI,2892
jax/_src/util.py,sha256=yMD2rMszGCSit2f6bUpA8MHSU7noTtybMEqG555MDOQ,18087
jax/_src/xla_bridge.py,sha256=VusEKpydQI84oZ1TnTb4HWtNrgYXUzE7I-EPiCaiww4,34616
jax/_src/clusters/__init__.py,sha256=zGB6xLPiYJsM-zWSgDaofQTdP1Ot1vPXYk-JNzYXZ6w,1161
jax/_src/clusters/cloud_tpu_cluster.py,sha256=2gu-EHealloJGZi6TXUfp38f2C5iP0aSQ6gHL9zj7LU,5833
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/export_back_compat_test_util.py,sha256=3XxJE9a4U6IyyfB6Hb76drY5rCBOBOT8kVUvdcL63-Q,15252
jax/_src/internal_test_util/lax_test_util.py,sha256=MKeoUyHd2ecj5dYutoRo_Jy_kQPjDCkqah6hwP3t9XU,12054
jax/_src/internal_test_util/test_harnesses.py,sha256=dKnfweFBPf3-nn3Ardq_1LsywTDzR48ou-qZXSqQKJ0,124673
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=RIpF5FST_k5L4B6IVf8ODq7-VP-WMJBVhBhvFIbPdAI,33638
jax/_src/interpreters/batching.py,sha256=DlndWvnUh5f3Cc9Vw14rWmiLnSm302UmcZCGodDz6ww,50700
jax/_src/interpreters/mlir.py,sha256=K83Et1Jl32XGP6oWGB0UgdR8QdxiZutgVBsdOCdGgQE,110840
jax/_src/interpreters/partial_eval.py,sha256=IJ69CCZ5_VDTXvoranKQASfSQ-9yvNAHlAkldUO9VOQ,122072
jax/_src/interpreters/pxla.py,sha256=1XqRK4s_9fDMeXqtuQfDRiFxXZ5AkGxeT8HUlQjTLjw,127839
jax/_src/interpreters/xla.py,sha256=U0JkJytgmkx0W6APAc5bXx6XWuvdzt1OM2CvyjKkwYg,12099
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/ann.py,sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M,17623
jax/_src/lax/convolution.py,sha256=xRVZbJwTkgEP4kM8C3nQmD2XhJH_rG4gosTPOCiTLUw,44715
jax/_src/lax/eigh.py,sha256=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=HcW5vHSsW4tOgIKuIUYFCVgG2bslUi1Eg1JQhqB5p4c,10149
jax/_src/lax/lax.py,sha256=noDa5s8pMP2AGHHA_1X7Ya82HwXyiCqhsMZ1CZofAW8,206746
jax/_src/lax/linalg.py,sha256=KH9oTc1KyEucxj8Wbr4cnyi67_PwIntKkivF63IBVgg,90455
jax/_src/lax/other.py,sha256=xiqAD3ZNXkIs0IzslMdhCoEOi_9xvsRzRRkUg0TAr7g,10554
jax/_src/lax/parallel.py,sha256=jSLuivJirRj20BsF4Ak_t8vdSEIFzkf0A0UiLasxJBk,71616
jax/_src/lax/qdwh.py,sha256=dtvODeExB-lk69iL3qr_55Xlku5nNVQjFqYz-1EDa4c,8023
jax/_src/lax/slicing.py,sha256=lBR1QBCaAveqDiLapgIAEOvtPxeHv0fnPZBNeAa6lB0,111816
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
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=YlAu8wpZM1CN3oYrjE0ggeVx1_lw_IaKdtvqsIdEW7c,40071
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/common.py,sha256=qxOTlGoIJfo0qpgCDsXWsNwL_8loTFlU03y8UW168t8,10272
jax/_src/lax/control_flow/conditionals.py,sha256=yW9oa-wQzCQBGn1oZjzjuTGB0YR2he7tWuNaZUs-cy4,43154
jax/_src/lax/control_flow/for_loop.py,sha256=cXMp6k6k0QSdFyXMJ30mkeMzJZKCdqm10YXM3mKBlxE,37033
jax/_src/lax/control_flow/loops.py,sha256=eWJfDLWuESHumYhJMxnP2fbDXbAgKZNLtAZ5GIf2n6c,102277
jax/_src/lax/control_flow/solves.py,sha256=D3PEw6MsfWehqEV8bvW2Ol3zyFO-funjbMzaJR3ArJ4,18898
jax/_src/lib/__init__.py,sha256=qXUkQEeihb8AwEi_ldBk9Ve4_07AnTsDAG0jEFTbVzI,5464
jax/_src/lib/mlir/__init__.py,sha256=KyKFigc6V_REcOMXB95auEAFQaOuFbfnxlxeyLcqfJk,676
jax/_src/lib/mlir/dialects/__init__.py,sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs,1195
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/functions.py,sha256=q8zFneK6f7U8xpIWjZ3_LXKCIzVKFAZqu8BXyR3rQVs,18402
jax/_src/nn/initializers.py,sha256=hIsY5n-jWmOfj_1i3SjXM5vdrg1gwNdsGv5OrXcRRbM,25594
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/array_methods.py,sha256=dTDs3MMtbdvGpu0nPFnvodX1QPDWzUE7Ria4Epujq0Y,32621
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=FeUA8lZk12dlMKfWPsbJRxmbdWPYdquL3p1FOb-NReo,212771
jax/_src/numpy/linalg.py,sha256=w3yvqLrxHDMo9QCPnPmRzsfzRaS4Ek0kjHomnOkJEiI,26587
jax/_src/numpy/polynomial.py,sha256=lQIcmiTjlC3XGVcaFbglX5zjsunI93PeUP3wxT6bfnc,12126
jax/_src/numpy/reductions.py,sha256=etdjefeSfDudsN0_ZjTHh2dvhGNLEgUPoOJY-1zP1-Q,41465
jax/_src/numpy/setops.py,sha256=jvm34fvxNPRUD74KsVPOCxOMfHmSohmV1JEIXJ2H2UI,13707
jax/_src/numpy/ufunc_api.py,sha256=ED6Ys4nk53u4bDSh9yvCCzT35IsAL-6E00qDYGXbonM,15246
jax/_src/numpy/ufuncs.py,sha256=JFpG6mKygQBDWbTgiHWBBWDhI5zYj5vyE56vo1GVxag,27083
jax/_src/numpy/util.py,sha256=NIpRsin_NOxiV8oUNpmgYqDRGTfRVb3RXDF96vKMwUE,18759
jax/_src/numpy/vectorize.py,sha256=QNSCyH2KENwzApKTy6ciS1KKABl1LhbL3e3tj4hBLGE,12527
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=W8rZmHfAHj9OEqpgRM0d0AsZExNjtxeGWEiihEzRUv8,5739
jax/_src/pallas/pallas_call.py,sha256=G8nAterQ5PulNzFrwwMQGq06L7NOspkHYfOLDTT58MM,18747
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=SX4gmShGQPi7iVxmSc4JnTTY8PAIc09u1vXzSZL_oAY,2017
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=3hRozTPO6b9LkjxzjCFug8xpV7BgdRY1oj4DFzKOMXA,62572
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=-p_hH0yrWPdlOxCi4i0elYIiBsse_GSyGxZYFajByNY,3504
jax/_src/pallas/mosaic/primitives.py,sha256=9eGNiI8OoiXJ5kCXGr_BbmfnNT-Q2aWF4hQLP_OOtqE,12524
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/lowering.py,sha256=jv31cJh3yMJgmh3j8tYYthqBK5dLNfK8vYkM4Zijagw,51988
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/fft.py,sha256=peYWlVuk9hepPlZ-_yiKPUnZl8wLObNOvQ8ERP8L4ig,6253
jax/_src/scipy/integrate.py,sha256=kHJe7Bmki7VIg5iauSwQtreNyoDt45eIqhOoK9hy4rY,1553
jax/_src/scipy/linalg.py,sha256=CbxEopYF26wckLVvYuCMnHx0gW8rZyfCA-XVDiHOm_o,41910
jax/_src/scipy/ndimage.py,sha256=uydh28hKMyyxjHQC_J5nEHDC-lYRJ8wNMHU_fBDHxDY,5354
jax/_src/scipy/signal.py,sha256=wnVKw8vYTCAhNGsOrjYJFR8efUvgeCSSZ3wn1oiFTq0,27554
jax/_src/scipy/special.py,sha256=hlgzSo2H-4AFbyUEgDym1N70fXRg0pkh0MoXTFdkx_M,63253
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=VCjQWXeUaf_9xqTCTIiGTVkMjaa9IvEwKgLrhinavEw,27326
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=hBJtU9D5tX5z2sYb5rOVqZYbN9WMfPvbsca4EU1kPi4,21539
jax/experimental/host_callback.py,sha256=ue_pM1_Sh4xZ4iE_MtAvmcI2Y6T_4k7hPbbjDodqHLk,79550
jax/experimental/jet.py,sha256=4Lk7tOMSThQDy5zoH3QJ72Qo-s8B6jvEfUUgw2eQle0,26899
jax/experimental/layout.py,sha256=yzQFzFbz1XVt6pwOHEDEFJndisvA7u06OrUJkYJyKAU,668
jax/experimental/maps.py,sha256=yjycDuNpT_aNQJK_XUBqhDDfY7qHUKeBpfi7w_zwSAs,964
jax/experimental/mesh_utils.py,sha256=fY7iAGYD9kwnRQTxddi6Lv5kEf61jXcMuSTThF3BFFQ,15572
jax/experimental/multihost_utils.py,sha256=HG5RnIlsKXbtIHgbi7GjhUnZj2eolcxJDxzwwFyb7TY,16081
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pjit.py,sha256=QgNOqRbjr8SR_O7bs2ntcLqTk75Jbt7mjDjrGsWj1WI,1438
jax/experimental/profiler.py,sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus,1424
jax/experimental/rnn.py,sha256=HL4GGWdBbp5T-4KH6Jwq_FCqtbslppjzf8eCbl85cGU,18684
jax/experimental/serialize_executable.py,sha256=8wx9DBxOd2SxDYvDUSJQ6WmRc8Bsls9xUIcCOYCU2ik,3014
jax/experimental/shard_map.py,sha256=qaC6BaTRpzvkzgFFhynPfsN-WHNrBPW76E6zatS01kU,78700
jax/experimental/topologies.py,sha256=IN2LBhwTIYzFZnph7cRnJUHCH1zXREZjLha3g8uVFe0,1864
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/array_api/__init__.py,sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA,5364
jax/experimental/array_api/_array_methods.py,sha256=ZMMl4a2gXv_xZOvOmrhJhX13UBc9PwGkatPc63OfbJI,1785
jax/experimental/array_api/_constants.py,sha256=Q8eoyHE5GDoeBUgbur2mF3_RzUW6OO1PhzneUzmnH_c,669
jax/experimental/array_api/_creation_functions.py,sha256=RdV4316PQaXg5sf4wVrWWw_JQ0rf6akXgYj26XC0Kdk,2496
jax/experimental/array_api/_data_type_functions.py,sha256=cmVQeYvTMcaVnXuhdL0sTGbiQDkrPh3xQMBjQ3hLM4o,6303
jax/experimental/array_api/_dtypes.py,sha256=4itd8mk4z4SpohYVHfELDpGQF87fu2ScT61nBD3tjG4,968
jax/experimental/array_api/_elementwise_functions.py,sha256=Z0iZB20H80XzoIxUEPC5eeA_WdRtQgszwvQsOk1Whic,14860
jax/experimental/array_api/_fft_functions.py,sha256=okg_ZQy62JTj2_CVfOkvrwn6XyCBWwX7ScBdLDnVWNg,3096
jax/experimental/array_api/_indexing_functions.py,sha256=gemN8vpsGfjy1GDilIQbtes-NHIvE9A0x18mqxZR5mk,675
jax/experimental/array_api/_linear_algebra_functions.py,sha256=tRA4XxsDqIrVvT7Dgo5sLB5nomNhvukVAadcMNCbpx0,6056
jax/experimental/array_api/_manipulation_functions.py,sha256=S08_RVIPqjzNpfdiTuMQt7SMNOLmTgvjkVejDj53d74,3031
jax/experimental/array_api/_searching_functions.py,sha256=sl77E7ZutUWrZwtotTs8kTmhowXJEcKD-wEGZHsPEHg,1376
jax/experimental/array_api/_set_functions.py,sha256=ZxvroMAnG4mpmGKGdQyCUu5Va3jziNrQIy-CiKrCjHY,2243
jax/experimental/array_api/_sorting_functions.py,sha256=1fAQCzYEOx-5Ce2vLi6au9EPfWPP7fBkI5d6a-t-mCs,1275
jax/experimental/array_api/_statistical_functions.py,sha256=0xcB70zRpzfhTqOZyKEkfvFmXYSB8gBQPInuX_ai5T4,2003
jax/experimental/array_api/_utility_functions.py,sha256=Nx0zBTWs_uXSL65mo3rqnoQHAbFatpfkQCsv4UsFCuQ,975
jax/experimental/array_api/_version.py,sha256=fVbhvRAhSrwJXVDwqRQEt5vZWd-PGzRQkxJ1zmfiMFA,616
jax/experimental/array_api/fft.py,sha256=1N17z4OpG086sZne714AEJrfmZ3OnagMyMRYcLQ48aA,906
jax/experimental/array_api/linalg.py,sha256=-i-ZfmrMECazmyk0NFHEnsKV5isPeZNmvmX_mDBuJqE,1167
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=uhoAlXU_T81ILGqDbL8Uv3LvAnOriZjbaaIbWPBXfns,13018
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=HbfdbB5-lRbLmckZcvFyBKuhq4uQkosVY5AEDuujSPA,58543
jax/experimental/export/shape_poly.py,sha256=vSb7pFScPzJ7gUHncWQtzRtjPCQP-S-YL3lbP9s5E3g,62719
jax/experimental/jax2tf/__init__.py,sha256=7tXoLqrwpmFJsqrRnkqCg0EqbiIFJ9J0YOnlRGSXbHI,1085
jax/experimental/jax2tf/call_tf.py,sha256=hXL-N20I7ybIcQuRnXadRbXFGLqS8mUi24vzrZT8kUw,27391
jax/experimental/jax2tf/impl_no_xla.py,sha256=Ybue-3arwSWyVZIsoiE_qVlY5eEx3poyhkB3jdERslE,53993
jax/experimental/jax2tf/jax2tf.py,sha256=cAc-xU8TD8tN4zniwFfsKKzCDYLoWHrXocAJHfuncxA,141793
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=Oy9GmL8qMZ2QkJgvk1ebTFFEmZ2EiimXOiJrlxXGBJs,32314
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=IGP519uzgTvzN2eXywy6873kapfgZnpksu0EanwKMVQ,5382
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=TKCkl_Vyo91V7pIIFpPPnLNUdBJ_7-4U0Oe9ITjkARc,58561
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=9RJ_cbVtHWPA3cDJb1yQbMKzhLQsi80X5gengNCvfoc,67596
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=WqYkw_GotGycsZZpHsiFqz2tftwkDo85rdVM5Q_W2HE,6666
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/primitives_test.py,sha256=RuQJk4pCGOsum-y5vqA5CQproY-4ycwOJjdAQFrfhpA,14048
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M,12815
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=rt83YM2U9kgKhtSzoaK3qxAAIPrKHA995LZIDkfAUKE,133867
jax/experimental/jax2tf/tests/sharding_test.py,sha256=P_0S0WYDlXzeJm0sI5xilAcXv9_8QmZFE-0vo_WnsgY,31365
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=G4YtsCS7rfBY6oBnbgJ41bkl_owWdQ8YmQO1St6_KNQ,20075
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/dialects.py,sha256=4f2cGGXxVUjHh_Jv1Q3hGOTfx8EyN6CWdwznrHwgbIk,791
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=0ilsv4z-VTA82HebQ-L_AD8miM3AXrofynONEHcxJ8M,1729
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
jax/experimental/pallas/ops/attention.py,sha256=QBloFG8BDuhD0Z-ndH-1CiCeslqN7Sa6dJrPpv8_W5k,18900
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=LI2C0CLu2Lc3nxLcyp9PySBqc2lnEMb8DYRAxIkUICY,3206
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=SHYtb8Bln5QI3AE0LhnIk1eIGfSOsxNeBpACwmpHRjk,128191
jax/experimental/sparse/bcsr.py,sha256=ICHMKqhfQHuKz2udHHH9Ngn9R8qIxyx0BAmb6ppeGHI,38227
jax/experimental/sparse/coo.py,sha256=hXDUKoeQIWmkFM2SgifNcAM-hOG3z_pmcsvuja4XJVQ,24125
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=MXD6sJrBLio729YHj2csDL9LSkNmuOW4uvCW9IsSg28,36919
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/extend/__init__.py,sha256=67qhpZPLPofDNC0vKaWo6f1j3K2kF6nZj2e9X8UrWUY,1225
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/random.py,sha256=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU,1083
jax/extend/source_info_util.py,sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI,1215
jax/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=-lae_76QV7l1wXU29P2nIzhiEx822yekIKSM0HLZwzo,12852
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA,767
jax/lib/xla_bridge.py,sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw,840
jax/nn/__init__.py,sha256=o23x0ZOvhizFah-idwif160uoE8XivJvphR2-YuXXZs,1866
jax/nn/initializers.py,sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs,1469
jax/numpy/__init__.py,sha256=Xb0CSG1YvzBfegxCQUVhg0ddkth07DKoetl9fEYb4Ik,12225
jax/numpy/__init__.pyi,sha256=3JB-JKtpfS-nukvTwkj8AGOPZuw1Ohhr5Y3y8S8APeo,34997
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=hipvIwK9UB4FvQCDk9I9rBxrZ6QDZxs8TW9-MKQ7h4I,1542
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=_Hei7RMSv9ptzQwj_z0ezdJqvxYHz4dE4QBpdLfE84A,3296
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
jax-0.4.21.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.21.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.21.dist-info/METADATA,sha256=iO6B0KWqQ0ek-LD-24akI986_F09Iyc2eORo5P4ts2w,23495
jax-0.4.21.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
jax-0.4.21.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.4.21.dist-info/RECORD,,
