jax-0.4.23.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.4.23.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
jax-0.4.23.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.4.23.dist-info/METADATA,sha256=WTneMgjjf8U_9Bwdue1LgchPFMLJ3YGJegKor6MwzTU,24288
jax-0.4.23.dist-info/RECORD,,
jax-0.4.23.dist-info/REQUESTED,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax-0.4.23.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
jax-0.4.23.dist-info/direct_url.json,sha256=dDOGoy69fWVkifl66iuD6qk3ZMLVZGb4Mh5XdjlIBnA,134
jax-0.4.23.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax/__init__.py,sha256=TXCATqyRhKhnolBOke2n4tlUA814iIBKXNdFpk6fjkg,9543
jax/__pycache__/__init__.cpython-39.pyc,,
jax/__pycache__/ad_checkpoint.cpython-39.pyc,,
jax/__pycache__/api_util.cpython-39.pyc,,
jax/__pycache__/cloud_tpu_init.cpython-39.pyc,,
jax/__pycache__/collect_profile.cpython-39.pyc,,
jax/__pycache__/config.cpython-39.pyc,,
jax/__pycache__/core.cpython-39.pyc,,
jax/__pycache__/custom_batching.cpython-39.pyc,,
jax/__pycache__/custom_derivatives.cpython-39.pyc,,
jax/__pycache__/custom_transpose.cpython-39.pyc,,
jax/__pycache__/debug.cpython-39.pyc,,
jax/__pycache__/distributed.cpython-39.pyc,,
jax/__pycache__/dlpack.cpython-39.pyc,,
jax/__pycache__/dtypes.cpython-39.pyc,,
jax/__pycache__/errors.cpython-39.pyc,,
jax/__pycache__/flatten_util.cpython-39.pyc,,
jax/__pycache__/linear_util.cpython-39.pyc,,
jax/__pycache__/monitoring.cpython-39.pyc,,
jax/__pycache__/prng.cpython-39.pyc,,
jax/__pycache__/profiler.cpython-39.pyc,,
jax/__pycache__/random.cpython-39.pyc,,
jax/__pycache__/sharding.cpython-39.pyc,,
jax/__pycache__/stages.cpython-39.pyc,,
jax/__pycache__/test_util.cpython-39.pyc,,
jax/__pycache__/tree_util.cpython-39.pyc,,
jax/__pycache__/typing.cpython-39.pyc,,
jax/__pycache__/util.cpython-39.pyc,,
jax/__pycache__/version.cpython-39.pyc,,
jax/_src/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/__pycache__/__init__.cpython-39.pyc,,
jax/_src/__pycache__/abstract_arrays.cpython-39.pyc,,
jax/_src/__pycache__/ad_checkpoint.cpython-39.pyc,,
jax/_src/__pycache__/ad_util.cpython-39.pyc,,
jax/_src/__pycache__/api.cpython-39.pyc,,
jax/_src/__pycache__/api_util.cpython-39.pyc,,
jax/_src/__pycache__/array.cpython-39.pyc,,
jax/_src/__pycache__/basearray.cpython-39.pyc,,
jax/_src/__pycache__/cache_key.cpython-39.pyc,,
jax/_src/__pycache__/callback.cpython-39.pyc,,
jax/_src/__pycache__/checkify.cpython-39.pyc,,
jax/_src/__pycache__/cloud_tpu_init.cpython-39.pyc,,
jax/_src/__pycache__/compilation_cache.cpython-39.pyc,,
jax/_src/__pycache__/compilation_cache_interface.cpython-39.pyc,,
jax/_src/__pycache__/compiler.cpython-39.pyc,,
jax/_src/__pycache__/config.cpython-39.pyc,,
jax/_src/__pycache__/core.cpython-39.pyc,,
jax/_src/__pycache__/custom_api_util.cpython-39.pyc,,
jax/_src/__pycache__/custom_batching.cpython-39.pyc,,
jax/_src/__pycache__/custom_derivatives.cpython-39.pyc,,
jax/_src/__pycache__/custom_transpose.cpython-39.pyc,,
jax/_src/__pycache__/debugging.cpython-39.pyc,,
jax/_src/__pycache__/deprecations.cpython-39.pyc,,
jax/_src/__pycache__/dispatch.cpython-39.pyc,,
jax/_src/__pycache__/distributed.cpython-39.pyc,,
jax/_src/__pycache__/dlpack.cpython-39.pyc,,
jax/_src/__pycache__/dtypes.cpython-39.pyc,,
jax/_src/__pycache__/effects.cpython-39.pyc,,
jax/_src/__pycache__/environment_info.cpython-39.pyc,,
jax/_src/__pycache__/errors.cpython-39.pyc,,
jax/_src/__pycache__/flatten_util.cpython-39.pyc,,
jax/_src/__pycache__/gfile_cache.cpython-39.pyc,,
jax/_src/__pycache__/jaxpr_util.cpython-39.pyc,,
jax/_src/__pycache__/lax_reference.cpython-39.pyc,,
jax/_src/__pycache__/layout.cpython-39.pyc,,
jax/_src/__pycache__/lazy_loader.cpython-39.pyc,,
jax/_src/__pycache__/linear_util.cpython-39.pyc,,
jax/_src/__pycache__/logging_config.cpython-39.pyc,,
jax/_src/__pycache__/maps.cpython-39.pyc,,
jax/_src/__pycache__/mesh.cpython-39.pyc,,
jax/_src/__pycache__/monitoring.cpython-39.pyc,,
jax/_src/__pycache__/op_shardings.cpython-39.pyc,,
jax/_src/__pycache__/partition_spec.cpython-39.pyc,,
jax/_src/__pycache__/path.cpython-39.pyc,,
jax/_src/__pycache__/pickle_util.cpython-39.pyc,,
jax/_src/__pycache__/pjit.cpython-39.pyc,,
jax/_src/__pycache__/pretty_printer.cpython-39.pyc,,
jax/_src/__pycache__/prng.cpython-39.pyc,,
jax/_src/__pycache__/profiler.cpython-39.pyc,,
jax/_src/__pycache__/public_test_util.cpython-39.pyc,,
jax/_src/__pycache__/random.cpython-39.pyc,,
jax/_src/__pycache__/sharding.cpython-39.pyc,,
jax/_src/__pycache__/sharding_impls.cpython-39.pyc,,
jax/_src/__pycache__/sharding_specs.cpython-39.pyc,,
jax/_src/__pycache__/source_info_util.cpython-39.pyc,,
jax/_src/__pycache__/stages.cpython-39.pyc,,
jax/_src/__pycache__/test_util.cpython-39.pyc,,
jax/_src/__pycache__/tpu_custom_call.cpython-39.pyc,,
jax/_src/__pycache__/traceback_util.cpython-39.pyc,,
jax/_src/__pycache__/tree_util.cpython-39.pyc,,
jax/_src/__pycache__/typing.cpython-39.pyc,,
jax/_src/__pycache__/util.cpython-39.pyc,,
jax/_src/__pycache__/xla_bridge.cpython-39.pyc,,
jax/_src/abstract_arrays.py,sha256=ZpKoD3prFnp0yAECVMMAA99F4v9MJxnPOyC7Exkr_gY,3203
jax/_src/ad_checkpoint.py,sha256=3OyYHojuy1BqhICKYpiM8Yb2y5Ct0WY2ZRuuIAz89pE,34683
jax/_src/ad_util.py,sha256=BBNIshxskhvZydeapb6UCNhnkmWBnbJmPF88Y5T4jMQ,3966
jax/_src/api.py,sha256=mTJkaluNO2Jnyx8H4LuF0L-plMd9XOlrRXQQk57LtsQ,131527
jax/_src/api_util.py,sha256=oif-SXJTcQk5EK4CJudcgnEDMRXRpGQYZD8a2roOacI,26471
jax/_src/array.py,sha256=u-fGLENVK6_EqBnLFJOLgBu6CPfaeFbsHaVzRRs9fJQ,33369
jax/_src/basearray.py,sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A,4296
jax/_src/basearray.pyi,sha256=lO4QG8TKVdEZuzVnM_8HEI_3lrNNt8EkVcNh8ga4V0k,10581
jax/_src/cache_key.py,sha256=fA6bjaN_cvTxo6dRtmFOLX_VGfoNer_au1a43jR-ycI,8967
jax/_src/callback.py,sha256=y6GhYzA0H6PaZzVn83O-SEfKbdpTtNK4XvSUzAYx3rA,18282
jax/_src/checkify.py,sha256=GH6dQVAjO59ROLrUpxWkBzsz5YceHNY1aPf_qc_I3Xo,52834
jax/_src/cloud_tpu_init.py,sha256=HpJitsv4AwaP-hvmgwboEYoDmMXuXP9-CDRYHAEdPjE,2647
jax/_src/clusters/__init__.py,sha256=zGB6xLPiYJsM-zWSgDaofQTdP1Ot1vPXYk-JNzYXZ6w,1161
jax/_src/clusters/__pycache__/__init__.cpython-39.pyc,,
jax/_src/clusters/__pycache__/cloud_tpu_cluster.cpython-39.pyc,,
jax/_src/clusters/__pycache__/cluster.cpython-39.pyc,,
jax/_src/clusters/__pycache__/ompi_cluster.cpython-39.pyc,,
jax/_src/clusters/__pycache__/slurm_cluster.cpython-39.pyc,,
jax/_src/clusters/cloud_tpu_cluster.py,sha256=o97GoH5FSbLXdtBTKYfgBbDWV-02Rd7f93VpGNMpU24,5967
jax/_src/clusters/cluster.py,sha256=8fcnWnZVV6msrljrN8UmH-JZ9-7zZvhN5GQvUIZh3AQ,4760
jax/_src/clusters/ompi_cluster.py,sha256=FVzhwuactnX3un7bMtLggl8LaiFF2QHisjqVQq5WPDg,2268
jax/_src/clusters/slurm_cluster.py,sha256=V2J-mQj6Rf_bV4Q50-yc4EjPW66CABpFbdRdjs653r0,2271
jax/_src/compilation_cache.py,sha256=LI81iaNhTvQoOVmH-G3CHSO1MWZS58HbHQzmKqsbzcc,6925
jax/_src/compilation_cache_interface.py,sha256=xRXZ8RjI3kLnCDAxkBh8jvr_cN3e9Sk9E9CR1xkbrpo,831
jax/_src/compiler.py,sha256=rGg03nbpnLsxPlFgNtQ2zk1ii7SUswXsDi_mWI8MMIQ,15603
jax/_src/config.py,sha256=0SPmaRT16rYCsi405qZZ1cDRlYtbVSzKpozK2pohjzg,52841
jax/_src/core.py,sha256=OPjwZNtI4_VAkgvD2U0aEnXzv3C85C4uOYNm4hqvBbE,116984
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=qXKIOiS0ktOy1k2k6QWP7L5OjjwCQ1VXEmjARjd7AHQ,10498
jax/_src/custom_derivatives.py,sha256=p2Yg9DYXLhygYVf4N3TR4-QWHXbK6Wq9beZf5_X4oV4,58846
jax/_src/custom_transpose.py,sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E,8310
jax/_src/debugger/__init__.py,sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw,885
jax/_src/debugger/__pycache__/__init__.cpython-39.pyc,,
jax/_src/debugger/__pycache__/cli_debugger.cpython-39.pyc,,
jax/_src/debugger/__pycache__/colab_debugger.cpython-39.pyc,,
jax/_src/debugger/__pycache__/colab_lib.cpython-39.pyc,,
jax/_src/debugger/__pycache__/core.cpython-39.pyc,,
jax/_src/debugger/__pycache__/web_debugger.cpython-39.pyc,,
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/debugging.py,sha256=NtZqMeCJ97ilDemchzIX0pU3vba-SJoZUY_NMYjhRBo,25692
jax/_src/deprecations.py,sha256=Y2ID9gpK__WiYbwkM7buxpOaC3Yc1Ynls37vJr1wOn0,1763
jax/_src/dispatch.py,sha256=9CUqihsvD5e_g8qsD9HRPwK0bNwdICt7ZcS34QEfP8w,18067
jax/_src/distributed.py,sha256=Jt00nRrt8jlr9f2AgCtyN32_l9On3yciG1LA9lE6vbE,7670
jax/_src/dlpack.py,sha256=o2fHLqpUOTlKdrP8UMpOB9Gx0k-lYua2e3qAD1e4kvA,4097
jax/_src/dtypes.py,sha256=3KvsRMCO7LyMuBgi8KrcoasidL1_gw9myEly07uZ1pM,26458
jax/_src/effects.py,sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q,4682
jax/_src/environment_info.py,sha256=1aFtOHf6p62XEmy98E2uq-XW6xHNuQOPJH2kVWJL2nM,1962
jax/_src/errors.py,sha256=jJvt8vwEuR8Ie_hdqjBqSwuy5zD74nkIXvbT_ttPEso,23777
jax/_src/extend/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/extend/__pycache__/__init__.cpython-39.pyc,,
jax/_src/extend/__pycache__/random.cpython-39.pyc,,
jax/_src/extend/random.py,sha256=bAtdkmxevVfY9JfcwrWX6Ah8RXMoNelXg2I644ePHCM,1276
jax/_src/flatten_util.py,sha256=jzKbWVjG04I3Xwltjw7rOxg80FjuZLhZoSD7QftE18Q,3557
jax/_src/gfile_cache.py,sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY,1923
jax/_src/image/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/image/__pycache__/__init__.cpython-39.pyc,,
jax/_src/image/__pycache__/scale.cpython-39.pyc,,
jax/_src/image/scale.py,sha256=sWBhsUcujSz-pDPJe-gqiK-byJk_Dd2gISoT7in8dXg,13758
jax/_src/internal_test_util/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/internal_test_util/__pycache__/__init__.cpython-39.pyc,,
jax/_src/internal_test_util/__pycache__/deprecation_module.cpython-39.pyc,,
jax/_src/internal_test_util/__pycache__/export_back_compat_test_util.cpython-39.pyc,,
jax/_src/internal_test_util/__pycache__/lax_test_util.cpython-39.pyc,,
jax/_src/internal_test_util/__pycache__/test_harnesses.cpython-39.pyc,,
jax/_src/internal_test_util/deprecation_module.py,sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4,851
jax/_src/internal_test_util/export_back_compat_test_util.py,sha256=AYHQ_HZTFlb0MLer6qz-UnOfaQwZ4QPZPjqNI6NlRks,15288
jax/_src/internal_test_util/lax_test_util.py,sha256=LBych5KKJ1fYfxHTk9YyyiOuVGljYYyngz1Ckp__BNY,12148
jax/_src/internal_test_util/lazy_loader_module/__init__.py,sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs,705
jax/_src/internal_test_util/lazy_loader_module/__pycache__/__init__.cpython-39.pyc,,
jax/_src/internal_test_util/lazy_loader_module/__pycache__/lazy_test_submodule.cpython-39.pyc,,
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py,sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk,610
jax/_src/internal_test_util/test_harnesses.py,sha256=0q10eHQPsTgG1flHTZ1HHMhP2UVmPaCWrsG_pvG_9f0,124675
jax/_src/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/__pycache__/__init__.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/ad.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/batching.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/mlir.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/partial_eval.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/pxla.cpython-39.pyc,,
jax/_src/interpreters/__pycache__/xla.cpython-39.pyc,,
jax/_src/interpreters/ad.py,sha256=GDIZX3DRJe11KR27MMXarT9ZT1X8i-5RKJCtkkiGlIg,33648
jax/_src/interpreters/batching.py,sha256=0sdU1ao_gIhxRZLVjVgFLCg2SoWRcVKXZvSTuJsMvAs,50710
jax/_src/interpreters/mlir.py,sha256=w_yhe0XBpxizIL03ANr6cvlIkXxRzYstf8wtTmZwZKM,112537
jax/_src/interpreters/partial_eval.py,sha256=hKtjg0HENaFe3mSgDFVs0k1VIgCjwd2MlNHFeWD6Apk,122020
jax/_src/interpreters/pxla.py,sha256=VZeOWy217t1BZp_shgmjGHkUsCKFPVN-Qp551nuFR6Q,128203
jax/_src/interpreters/xla.py,sha256=Q6VGpfpqcmWU_FiEF5MCH0wq964jU668BS9p_L7xlcY,12067
jax/_src/jaxpr_util.py,sha256=4mB7jfzh0IhgqiqnStUnL4_msTyOljxgzG56pw1-x3s,6428
jax/_src/lax/__init__.py,sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI,690
jax/_src/lax/__pycache__/__init__.cpython-39.pyc,,
jax/_src/lax/__pycache__/ann.cpython-39.pyc,,
jax/_src/lax/__pycache__/convolution.cpython-39.pyc,,
jax/_src/lax/__pycache__/eigh.cpython-39.pyc,,
jax/_src/lax/__pycache__/fft.cpython-39.pyc,,
jax/_src/lax/__pycache__/lax.cpython-39.pyc,,
jax/_src/lax/__pycache__/linalg.cpython-39.pyc,,
jax/_src/lax/__pycache__/other.cpython-39.pyc,,
jax/_src/lax/__pycache__/parallel.cpython-39.pyc,,
jax/_src/lax/__pycache__/qdwh.cpython-39.pyc,,
jax/_src/lax/__pycache__/slicing.cpython-39.pyc,,
jax/_src/lax/__pycache__/special.cpython-39.pyc,,
jax/_src/lax/__pycache__/stack.cpython-39.pyc,,
jax/_src/lax/__pycache__/svd.cpython-39.pyc,,
jax/_src/lax/__pycache__/utils.cpython-39.pyc,,
jax/_src/lax/__pycache__/windowed_reductions.cpython-39.pyc,,
jax/_src/lax/ann.py,sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M,17623
jax/_src/lax/control_flow/__init__.py,sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw,2103
jax/_src/lax/control_flow/__pycache__/__init__.cpython-39.pyc,,
jax/_src/lax/control_flow/__pycache__/common.cpython-39.pyc,,
jax/_src/lax/control_flow/__pycache__/conditionals.cpython-39.pyc,,
jax/_src/lax/control_flow/__pycache__/for_loop.cpython-39.pyc,,
jax/_src/lax/control_flow/__pycache__/loops.cpython-39.pyc,,
jax/_src/lax/control_flow/__pycache__/solves.cpython-39.pyc,,
jax/_src/lax/control_flow/common.py,sha256=qYDAJ3mt93tjPUwVBxyP1wlVXqqFObQ3ORnu2oJgXTs,10292
jax/_src/lax/control_flow/conditionals.py,sha256=yW9oa-wQzCQBGn1oZjzjuTGB0YR2he7tWuNaZUs-cy4,43154
jax/_src/lax/control_flow/for_loop.py,sha256=F-LFL5g_GwiCYebEbBGGm9xSPr2teG985lFVXXDDYIU,37040
jax/_src/lax/control_flow/loops.py,sha256=D0R0cw31RUFZEU1WgMM3rCTLmH8HaqHqcXk3u9LXQeU,103813
jax/_src/lax/control_flow/solves.py,sha256=D3PEw6MsfWehqEV8bvW2Ol3zyFO-funjbMzaJR3ArJ4,18898
jax/_src/lax/convolution.py,sha256=snWKE70dQiJ7PNl6Q9S1p1uBFYpjDrVHqpHZM4T8Szw,44702
jax/_src/lax/eigh.py,sha256=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo,22205
jax/_src/lax/fft.py,sha256=izHxbi-S686onnDHQC3WRuu9PbKhDhYkZUtZPx6LxFI,10151
jax/_src/lax/lax.py,sha256=3cO6lWa2bR3vMaU7D-w_gEXwNCD8e5XZafajcPcJFks,206587
jax/_src/lax/linalg.py,sha256=_mTY_3pugjFpa_w571I_lTQDBb3UglLhmCqWBhomvMU,90456
jax/_src/lax/other.py,sha256=8o-YBA54msT4HNAsEmDOi9Wctha5ad1rd4fjSkyt5xM,10547
jax/_src/lax/parallel.py,sha256=ilaHYqPrrZihmBTvB8TLJPGzRBLlKHaUcRn_NLjxwUE,71621
jax/_src/lax/qdwh.py,sha256=f3QJwPMDDO_qC-VFaKrzSXUJ0njCSwSEjICsA5FckxU,8028
jax/_src/lax/slicing.py,sha256=uIYqhkjvFt9wJJ_2f0ltmkZ6X86NiSw_3sbBcStvjzM,111662
jax/_src/lax/special.py,sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY,26940
jax/_src/lax/stack.py,sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY,2560
jax/_src/lax/svd.py,sha256=ns1oFXqOUH0h6T3VXjuPT30mRCOodvJFJhGr1YgXaxs,9838
jax/_src/lax/utils.py,sha256=nbLjW7RfPRdBW3AXOMk2h7fza5Yx5wDynDzhnavA0lA,5006
jax/_src/lax/windowed_reductions.py,sha256=u_4VcO-emGWure71bXnKjlUSIzOSk8MHXyUMNQdd7g8,40045
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/lib/__init__.py,sha256=Houd6NOZ94PQeK1vZWXEvBCTjktaqeg6dxqU2v8LHHA,5453
jax/_src/lib/__pycache__/__init__.cpython-39.pyc,,
jax/_src/lib/mlir/__init__.py,sha256=vv85BXVE6lq8QZTUZtOIFPrwcoIWROkeiAqVa5a8SB0,885
jax/_src/lib/mlir/__pycache__/__init__.cpython-39.pyc,,
jax/_src/lib/mlir/dialects/__init__.py,sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs,1195
jax/_src/lib/mlir/dialects/__pycache__/__init__.cpython-39.pyc,,
jax/_src/linear_util.py,sha256=ZNPepQQYw3Ejbd1Kj9QdO6wUoIL763otW8BwAZBABHQ,13793
jax/_src/logging_config.py,sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY,2040
jax/_src/maps.py,sha256=CnQjzOWzLj2TE0h1KRtCIvyt2SgrJroBfIya-Im_9_c,86236
jax/_src/mesh.py,sha256=2QGHwooii-DZXMPRNO4kuAhneIGf85QfmOX_V-lYu7A,11190
jax/_src/monitoring.py,sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk,3888
jax/_src/nn/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/nn/__pycache__/__init__.cpython-39.pyc,,
jax/_src/nn/__pycache__/functions.cpython-39.pyc,,
jax/_src/nn/__pycache__/initializers.cpython-39.pyc,,
jax/_src/nn/functions.py,sha256=vPqYHflwWeFnA91rhApn7FVBNW2b7ESO4UlRSEYR1KA,18331
jax/_src/nn/initializers.py,sha256=ZP-ZtR4J51O4zR2znjfzG69sythZpkJQSPDXH96J3jE,25501
jax/_src/numpy/__init__.py,sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M,581
jax/_src/numpy/__pycache__/__init__.cpython-39.pyc,,
jax/_src/numpy/__pycache__/array_methods.cpython-39.pyc,,
jax/_src/numpy/__pycache__/fft.cpython-39.pyc,,
jax/_src/numpy/__pycache__/index_tricks.cpython-39.pyc,,
jax/_src/numpy/__pycache__/lax_numpy.cpython-39.pyc,,
jax/_src/numpy/__pycache__/linalg.cpython-39.pyc,,
jax/_src/numpy/__pycache__/polynomial.cpython-39.pyc,,
jax/_src/numpy/__pycache__/reductions.cpython-39.pyc,,
jax/_src/numpy/__pycache__/setops.cpython-39.pyc,,
jax/_src/numpy/__pycache__/ufunc_api.cpython-39.pyc,,
jax/_src/numpy/__pycache__/ufuncs.cpython-39.pyc,,
jax/_src/numpy/__pycache__/util.cpython-39.pyc,,
jax/_src/numpy/__pycache__/vectorize.cpython-39.pyc,,
jax/_src/numpy/array_methods.py,sha256=ysUvvdSCRmGOJ3vLX-OOp3itLMq_K5IRJJqWiCNGu4M,32619
jax/_src/numpy/fft.py,sha256=3uVzZYxW8_KmLutMO9CGxLUkXmwu4otudm9hP8F7ZSA,11203
jax/_src/numpy/index_tricks.py,sha256=gwT09d0GkwQ8WblRDT-nBfvoWx-o0DgHsHh2fUsvIjU,9908
jax/_src/numpy/lax_numpy.py,sha256=_TW4Nb2uTuwQ2PiSL5Zx5SkVvKpkeN3rYZWG8mljmpQ,212732
jax/_src/numpy/linalg.py,sha256=cj6RcitzFp6CO4F-5Ox0tLWeOOAyGwDMCzaTMbDlc88,27445
jax/_src/numpy/polynomial.py,sha256=3rCKlC2Q-_CvpHU8aVUu5aoFvq8QmRlj9vYtaP6A59s,12111
jax/_src/numpy/reductions.py,sha256=aXzvuB6zQA3RNiOq9Tb1lkTBHXjQQ60xTS7W7sIUPC0,41203
jax/_src/numpy/setops.py,sha256=vQafNBVvH6zHh00C7P1E6NFiIr3lRgvngnQIgD6yfcM,13687
jax/_src/numpy/ufunc_api.py,sha256=3ZW5FrJg3kaEfDItXEpDyuK_uKX1-rK3eKzqLiJqv_4,15232
jax/_src/numpy/ufuncs.py,sha256=3tAt9ZKn80ngAwN2B6CgQ0OHTAjFCLGKlSp8uTDo_EA,27088
jax/_src/numpy/util.py,sha256=hP8_mEPsQegTLB1mdZ1gwa4BIPZuY9_pd3qCA7JXYrI,18734
jax/_src/numpy/vectorize.py,sha256=QNSCyH2KENwzApKTy6ciS1KKABl1LhbL3e3tj4hBLGE,12527
jax/_src/op_shardings.py,sha256=mb_s-ThVgw-rgM_ZZVH1P6Lpgl1RtMauk__Qn2g9XqM,3895
jax/_src/ops/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/ops/__pycache__/__init__.cpython-39.pyc,,
jax/_src/ops/__pycache__/scatter.cpython-39.pyc,,
jax/_src/ops/__pycache__/special.cpython-39.pyc,,
jax/_src/ops/scatter.py,sha256=bWdrF9eSPPBvP4rH2e1Lo7__zFcGqnXkwFGaRz3XeyQ,17438
jax/_src/ops/special.py,sha256=ZlDyAxWecAhj0pV5Ey9IngMi-9_mDf7Lb1GG25KIjIU,4367
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/__pycache__/__init__.cpython-39.pyc,,
jax/_src/pallas/__pycache__/core.cpython-39.pyc,,
jax/_src/pallas/__pycache__/indexing.cpython-39.pyc,,
jax/_src/pallas/__pycache__/pallas_call.cpython-39.pyc,,
jax/_src/pallas/__pycache__/primitives.cpython-39.pyc,,
jax/_src/pallas/__pycache__/utils.cpython-39.pyc,,
jax/_src/pallas/core.py,sha256=Zn14pysOV3ucrif7MzyKGlN_GUCQSIN4FU_9JtWbqMY,9131
jax/_src/pallas/indexing.py,sha256=1xp2_LEsHyeI6oqurqc71hH42DRUA6tVFeYDF1uw5OU,5732
jax/_src/pallas/mosaic/__init__.py,sha256=SX4gmShGQPi7iVxmSc4JnTTY8PAIc09u1vXzSZL_oAY,2017
jax/_src/pallas/mosaic/__pycache__/__init__.cpython-39.pyc,,
jax/_src/pallas/mosaic/__pycache__/core.cpython-39.pyc,,
jax/_src/pallas/mosaic/__pycache__/kernel_regeneration_util.cpython-39.pyc,,
jax/_src/pallas/mosaic/__pycache__/lowering.cpython-39.pyc,,
jax/_src/pallas/mosaic/__pycache__/pallas_call_registration.cpython-39.pyc,,
jax/_src/pallas/mosaic/__pycache__/primitives.cpython-39.pyc,,
jax/_src/pallas/mosaic/core.py,sha256=MgPwhasoL5Efw8Ihlb8jmHIkOJ5b4X9mm-z9PKrOE4I,7569
jax/_src/pallas/mosaic/kernel_regeneration_util.py,sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8,1995
jax/_src/pallas/mosaic/lowering.py,sha256=pSP6OJxMpf2u2p8NX4BbQuJYRNtQIM773NIcCFbbSq4,64281
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=NvNC3HeqV-08_P7h6U7eCOvreM4wtKvtDuc2ztna8dE,3659
jax/_src/pallas/mosaic/primitives.py,sha256=0Z3VBmn4aNjwYrnmP6r-1yDrd0i76nwQGAlFh1ewN2M,12818
jax/_src/pallas/pallas_call.py,sha256=JoT94df5HT9h-YjvgZbsA3nSwkx1ngoMXNHcQzHklBU,20576
jax/_src/pallas/primitives.py,sha256=E9D3M_WHkOIONhkISbGbQRTYdWCHHaT6cIzQClJNNiU,15655
jax/_src/pallas/triton/__init__.py,sha256=orEv3BdlkA4qlE2bmoIAcnOm85p_x4ZjfpYJDGUwtdA,822
jax/_src/pallas/triton/__pycache__/__init__.cpython-39.pyc,,
jax/_src/pallas/triton/__pycache__/lowering.cpython-39.pyc,,
jax/_src/pallas/triton/lowering.py,sha256=O43jyuYef-7g44gOhmlFG-XPgt5sLBC09mVJBKDKFlw,52126
jax/_src/pallas/utils.py,sha256=MzAZ8ZYNBGlPPSXkUTlUL1MhRI_dz0Vse4YqIpwd56M,3115
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=jlOJyPIw2a6X46z3gsVKsx6kGz3s1gda3WgmMcjwPSA,93114
jax/_src/pretty_printer.py,sha256=nHoaspMnW9Rac9GvfWLNgG0chuprsbY_9aWmxwM4xGo,13055
jax/_src/prng.py,sha256=Gka0S1cTxobyTYsVWsR3XrEgGfq0BLOMgSHVch-LHXA,51572
jax/_src/profiler.py,sha256=lsRQTKMbCS5uX4ka-_R6Zi10QAKfJurZ2jca5_qMU64,14507
jax/_src/public_test_util.py,sha256=rv6aBUJqR25KDKkOB8EvpBXaR_FDK09MDiBrpoiHwNU,10397
jax/_src/random.py,sha256=48WZLE1i4_rJbYYlruFlwCMI89oTn94BM3-CWdFqiNc,99722
jax/_src/scipy/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/__pycache__/fft.cpython-39.pyc,,
jax/_src/scipy/__pycache__/integrate.cpython-39.pyc,,
jax/_src/scipy/__pycache__/linalg.cpython-39.pyc,,
jax/_src/scipy/__pycache__/ndimage.cpython-39.pyc,,
jax/_src/scipy/__pycache__/signal.cpython-39.pyc,,
jax/_src/scipy/__pycache__/special.cpython-39.pyc,,
jax/_src/scipy/cluster/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/cluster/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/cluster/__pycache__/vq.cpython-39.pyc,,
jax/_src/scipy/cluster/vq.py,sha256=OxDujGx3RvogMoqXKWWfpxsKigzBwxG2AYPCAsqLh4g,1771
jax/_src/scipy/fft.py,sha256=J7pJA1lzlOxc6WPgPlyCp7H-SxwptVKF972iSl1MpCw,6228
jax/_src/scipy/integrate.py,sha256=kHJe7Bmki7VIg5iauSwQtreNyoDt45eIqhOoK9hy4rY,1553
jax/_src/scipy/interpolate/__init__.py,sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8,581
jax/_src/scipy/interpolate/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/linalg.py,sha256=4Fg7JiR5d5gTU5lIRUvf9p67SIYbMJ0ECL5pFvQD048,41748
jax/_src/scipy/ndimage.py,sha256=uydh28hKMyyxjHQC_J5nEHDC-lYRJ8wNMHU_fBDHxDY,5354
jax/_src/scipy/optimize/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/optimize/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/optimize/__pycache__/_lbfgs.cpython-39.pyc,,
jax/_src/scipy/optimize/__pycache__/bfgs.cpython-39.pyc,,
jax/_src/scipy/optimize/__pycache__/line_search.cpython-39.pyc,,
jax/_src/scipy/optimize/__pycache__/minimize.cpython-39.pyc,,
jax/_src/scipy/optimize/_lbfgs.py,sha256=Io1xjWLc90YFTCPiGMz-fP9_FMWlpNuymTIVIKJXjZQ,7790
jax/_src/scipy/optimize/bfgs.py,sha256=fJ5GdaoI79pGHpFlX0USGF-1ooGa_-bKjMVvAaqqu1U,5582
jax/_src/scipy/optimize/line_search.py,sha256=JF16fsbL1kGJnVRRXumssJzXmsV7cQGeDFInmxYnlG0,13229
jax/_src/scipy/optimize/minimize.py,sha256=miFHZRaf8KpiDd4XsEUAKHUGrGodKYCOiKAITye-BIY,4979
jax/_src/scipy/signal.py,sha256=pMZayR1-43VaP-8z8VlRdcLv1vp6yAmE1NrkkmT2xbg,27508
jax/_src/scipy/sparse/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/sparse/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/sparse/__pycache__/linalg.cpython-39.pyc,,
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/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/spatial/__pycache__/transform.cpython-39.pyc,,
jax/_src/scipy/spatial/transform.py,sha256=Qru_dm510f7D2RP-cv1sXC38FXDOghEgKbQongL8XbQ,15155
jax/_src/scipy/special.py,sha256=PRcDBIi9V1LIFDoG8rW18mEV8ngolnCkfMyUUQtCJCw,63558
jax/_src/scipy/stats/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/_src/scipy/stats/__pycache__/__init__.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/_core.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/bernoulli.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/beta.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/betabinom.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/binom.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/cauchy.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/chi2.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/dirichlet.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/expon.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/gamma.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/gennorm.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/geom.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/kde.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/laplace.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/logistic.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/multinomial.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/multivariate_normal.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/nbinom.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/norm.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/pareto.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/poisson.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/t.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/truncnorm.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/uniform.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/vonmises.cpython-39.pyc,,
jax/_src/scipy/stats/__pycache__/wrapcauchy.cpython-39.pyc,,
jax/_src/scipy/stats/_core.py,sha256=8DqmMWWAGceMjtw975bvl4MP6ErnCgh4JIQZbc9RUfo,4971
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/sharding.py,sha256=T6r_pLVJ8mApjB0br4nNSWj7x45nWJc3FauyXOHgcd4,4921
jax/_src/sharding_impls.py,sha256=NR7qqD8PdWdqv_IH8FNQWpvq-wwxtQ-V8JrOQJIvpCM,48437
jax/_src/sharding_specs.py,sha256=KpFlpBX_puQRCXIado7TT0t2zmLFYPw3UMrBMv4rwIY,13549
jax/_src/source_info_util.py,sha256=j8L8uGsyv8_qdqd6nE1TMrPf0yrvedySAreFtvxRJd8,9348
jax/_src/stages.py,sha256=xsPwfUBgHK-SQwanw2stYbREG76cfR6liUOdg1-6OJw,25829
jax/_src/state/__init__.py,sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU,866
jax/_src/state/__pycache__/__init__.cpython-39.pyc,,
jax/_src/state/__pycache__/discharge.cpython-39.pyc,,
jax/_src/state/__pycache__/primitives.cpython-39.pyc,,
jax/_src/state/__pycache__/types.cpython-39.pyc,,
jax/_src/state/__pycache__/utils.cpython-39.pyc,,
jax/_src/state/discharge.py,sha256=zxhT04yPCWvZNKrI8drSbhUPeCMrZ6y8Zk3gxxFwXpQ,33486
jax/_src/state/primitives.py,sha256=Ozvz9bE56X8hFeL1PxYOIoZLc1lSFTJ4ylV7bK8scb8,21696
jax/_src/state/types.py,sha256=dqkFz4nXGxmHfTVJmRxk_wKc8PD5adiKYP2w-ZgrpgA,5738
jax/_src/state/utils.py,sha256=fn8NOU_WBT04Rd21eWmzecHiPKtBTaX5OqTqgeri5V8,2340
jax/_src/test_util.py,sha256=oX3pVlTkApDx12CCpbKOnZnZtUtlcl8Cd4AEg_eLFbw,47016
jax/_src/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/__pycache__/__init__.cpython-39.pyc,,
jax/_src/third_party/numpy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax/_src/third_party/numpy/__pycache__/__init__.cpython-39.pyc,,
jax/_src/third_party/numpy/__pycache__/linalg.cpython-39.pyc,,
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/__pycache__/__init__.cpython-39.pyc,,
jax/_src/third_party/scipy/__pycache__/betaln.cpython-39.pyc,,
jax/_src/third_party/scipy/__pycache__/interpolate.cpython-39.pyc,,
jax/_src/third_party/scipy/__pycache__/linalg.cpython-39.pyc,,
jax/_src/third_party/scipy/__pycache__/signal_helper.cpython-39.pyc,,
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=VqLSblEWCItAViXc6lIXRbF-FT83l9305aHIZP9B788,3081
jax/_src/tpu_custom_call.py,sha256=n-xsaGHdgUcRUeisxd2iK-UTdreN098_6uKUqq2pA10,17200
jax/_src/traceback_util.py,sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA,9060
jax/_src/tree_util.py,sha256=dTifj_ddFklcu-Rwe_YnN2w_pqAylQPKzrh0M2XYmP0,36036
jax/_src/typing.py,sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI,2892
jax/_src/util.py,sha256=IOz5XdLUMNB9zoLg7rOuhVQKEhYQ-AbGfOmGnzhKXqY,18095
jax/_src/xla_bridge.py,sha256=5YnPy3-hQH3ZKmcppqBnb-oQBmSJYz_TXGJfVqxAzOM,35292
jax/ad_checkpoint.py,sha256=N3-hWts5BdssIjnoNm1NnGzACdpQmtjol7fd_priwKo,802
jax/api_util.py,sha256=JnZFSuqh5WBglJwerrWxcIDslBaXYZQrCv6j0jPfu7M,769
jax/cloud_tpu_init.py,sha256=nNdPdZZffdRGb38rEx1YOLXNpymSNt05ZVWSm8lhZ8c,633
jax/collect_profile.py,sha256=exKxJOhMrkMnXcseOR2VwV4m3ZNw00DsMmXAR-9Qr1w,4721
jax/config.py,sha256=B6dnRzeKmU8XTeZLlKAT06VQDwoCLw8c7fJoSsKYOuM,1119
jax/core.py,sha256=_p4_J6R8maJi4cR2waN8zViGJjgx0Ynbyvv7VdMt95o,10180
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/example_libraries/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/example_libraries/__pycache__/__init__.cpython-39.pyc,,
jax/example_libraries/__pycache__/optimizers.cpython-39.pyc,,
jax/example_libraries/__pycache__/stax.cpython-39.pyc,,
jax/example_libraries/optimizers.py,sha256=224Ba6KPYLs3zlF3g7dlubbQkIkvoko-WGqDx4svBzE,20559
jax/example_libraries/stax.py,sha256=cMt8-39Q-o-tWwquR00RAlbbiWaitHLK4P03Xj0Wa3M,13849
jax/experimental/__init__.py,sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4,877
jax/experimental/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/__pycache__/checkify.cpython-39.pyc,,
jax/experimental/__pycache__/custom_partitioning.cpython-39.pyc,,
jax/experimental/__pycache__/host_callback.cpython-39.pyc,,
jax/experimental/__pycache__/jet.cpython-39.pyc,,
jax/experimental/__pycache__/layout.cpython-39.pyc,,
jax/experimental/__pycache__/maps.cpython-39.pyc,,
jax/experimental/__pycache__/mesh_utils.cpython-39.pyc,,
jax/experimental/__pycache__/multihost_utils.cpython-39.pyc,,
jax/experimental/__pycache__/ode.cpython-39.pyc,,
jax/experimental/__pycache__/pjit.cpython-39.pyc,,
jax/experimental/__pycache__/profiler.cpython-39.pyc,,
jax/experimental/__pycache__/rnn.cpython-39.pyc,,
jax/experimental/__pycache__/serialize_executable.cpython-39.pyc,,
jax/experimental/__pycache__/shard_map.cpython-39.pyc,,
jax/experimental/__pycache__/topologies.cpython-39.pyc,,
jax/experimental/__pycache__/x64_context.cpython-39.pyc,,
jax/experimental/array_api/__init__.py,sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA,5364
jax/experimental/array_api/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_array_methods.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_constants.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_creation_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_data_type_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_dtypes.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_elementwise_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_fft_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_indexing_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_linear_algebra_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_manipulation_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_searching_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_set_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_sorting_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_statistical_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_utility_functions.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/_version.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/fft.cpython-39.pyc,,
jax/experimental/array_api/__pycache__/linalg.cpython-39.pyc,,
jax/experimental/array_api/_array_methods.py,sha256=PmHaY46xTbmBcytrjWDfHHxFeA6FWEDIudNefHNYmJQ,1776
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=UysxZ2jT2K5yPM10z2pj9QhiQkMt6DfujijrcoxAbCc,6070
jax/experimental/array_api/_manipulation_functions.py,sha256=naJzREfLuGR6izxtMV8cvxkTdQqciQYqE8P2b2RHEwk,3019
jax/experimental/array_api/_searching_functions.py,sha256=VhWWD3rH1lQi4EfoGzKqC74bnNmCB4pmByk0WqTG3ek,1483
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/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/array_serialization/__pycache__/serialization.cpython-39.pyc,,
jax/experimental/array_serialization/__pycache__/serialization_test.cpython-39.pyc,,
jax/experimental/array_serialization/serialization.py,sha256=6oZUZg_F_3v0y_5SYydnGq_dHSRNuLXCtgWCorf8a_4,21753
jax/experimental/array_serialization/serialization_test.py,sha256=uhoAlXU_T81ILGqDbL8Uv3LvAnOriZjbaaIbWPBXfns,13018
jax/experimental/checkify.py,sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA,1213
jax/experimental/compilation_cache/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/compilation_cache/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/compilation_cache/__pycache__/compilation_cache.cpython-39.pyc,,
jax/experimental/compilation_cache/compilation_cache.py,sha256=FuNWsnrwmZgC00X2-K27s-D1bKIR7SC7gkMGLQ1WxmM,731
jax/experimental/custom_partitioning.py,sha256=9y7790jhaipum8dnH4LTJvl_l7__ToMjoLkBkMyK3NY,21800
jax/experimental/export/__init__.py,sha256=2cJ-OgrtfVES902JD29-I0UJH22lXrSoUepUwWoEYrU,682
jax/experimental/export/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/export/__pycache__/export.cpython-39.pyc,,
jax/experimental/export/__pycache__/serialization.cpython-39.pyc,,
jax/experimental/export/__pycache__/serialization_generated.cpython-39.pyc,,
jax/experimental/export/__pycache__/shape_poly.cpython-39.pyc,,
jax/experimental/export/export.py,sha256=1QoeqnARmopenKE8lWW2s7F6L14eIBl5In5oW95mLJk,58432
jax/experimental/export/serialization.py,sha256=_1I9ZjF0Ubc79TBmYLXDDtMOZWn9dm7TGPVuNVWa57M,16515
jax/experimental/export/serialization_generated.py,sha256=eiwhirTvzjyHq67oPRwQff9G1A1Lc6_BpR8KsPe68ds,25780
jax/experimental/export/shape_poly.py,sha256=JCZwTv5UXT56A5kjcDV2U5UJR_GUvTtbt-rz4oSzlMc,63429
jax/experimental/host_callback.py,sha256=VsWfdyVP78wYfXGay4AH6eenNdQxTs2KV0_LHt6LJqo,79581
jax/experimental/jax2tf/__init__.py,sha256=zAjffo3sksJB_KI-bMB4rYEQd7UZdkM61gUaoWeoeL4,925
jax/experimental/jax2tf/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/jax2tf/__pycache__/call_tf.cpython-39.pyc,,
jax/experimental/jax2tf/__pycache__/impl_no_xla.cpython-39.pyc,,
jax/experimental/jax2tf/__pycache__/jax2tf.cpython-39.pyc,,
jax/experimental/jax2tf/call_tf.py,sha256=34M1dxO3kt7EpuIDxwvCEIGMs6yL3XEmuk4kP-5vNOI,27425
jax/experimental/jax2tf/examples/__init__.py,sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw,581
jax/experimental/jax2tf/examples/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/keras_reuse_main_test.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/mnist_lib.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_lib.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main.cpython-39.pyc,,
jax/experimental/jax2tf/examples/__pycache__/saved_model_main_test.cpython-39.pyc,,
jax/experimental/jax2tf/examples/keras_reuse_main.py,sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is,2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py,sha256=1FBpLLwCpm9v4udaL0tEN3umiqF4a1lXSr8k483R9Eg,1616
jax/experimental/jax2tf/examples/mnist_lib.py,sha256=LftkJ-7__c0JpPxUWdx9uv3QiTGbOKRXzLhc1ixkev8,11520
jax/experimental/jax2tf/examples/saved_model_lib.py,sha256=GiDv959KnxbH_u6iOVKGhO0tU_KlTYO8Ymk6loaCDPw,7119
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/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/jax2tf/examples/serving/__pycache__/model_server_request.cpython-39.pyc,,
jax/experimental/jax2tf/examples/serving/model_server_request.py,sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8,4952
jax/experimental/jax2tf/impl_no_xla.py,sha256=NvmEkUv39qYOGL1CSgpChGKNDRa37qFKdmxumnCWrng,54017
jax/experimental/jax2tf/jax2tf.py,sha256=xifqlluRBuuigFDewuWN9Q2f-w9-uofTX4molcqlMTs,141925
jax/experimental/jax2tf/tests/__init__.py,sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg,581
jax/experimental/jax2tf/tests/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/back_compat_tf_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/call_tf_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/control_flow_ops_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/converters.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/cross_compilation_check.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_limitations.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax2tf_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/jax_primitives_coverage_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/model_harness.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/models_test_main.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/primitives_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/savedmodel_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/shape_poly_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/sharding_test.cpython-39.pyc,,
jax/experimental/jax2tf/tests/__pycache__/tf_test_util.cpython-39.pyc,,
jax/experimental/jax2tf/tests/back_compat_test.py,sha256=Gh9XRYkih5yKZw8JvCHZ7jTf9xuvbcfpKyI9FbSJjzE,32366
jax/experimental/jax2tf/tests/back_compat_tf_test.py,sha256=9FGx-2zxn89IzuQ6OYhaqA0zII0OQ6L6jktQvHzSqY0,5412
jax/experimental/jax2tf/tests/call_tf_test.py,sha256=8EqjRWiYd9H1RZEzRUFLoa2jzJ9vMyvcJ0IGaVc4Dc4,62904
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=mnuni8_Kwueia-gIq_sbqi1l1OHeHscO5rXWNgzT9iI,7421
jax/experimental/jax2tf/tests/jax2tf_limitations.py,sha256=fSAjcdd1d6-NnsIgw6GlLyvarjUeG6RHrzI_OxBROro,58570
jax/experimental/jax2tf/tests/jax2tf_test.py,sha256=RdT5h-HY62Oysk0vMCd-giMX6stXwokdyTgRcSR5DlI,67581
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py,sha256=zabPZ6x7eN4ZQ8HIL4TQtdHf-4OH8_r3mc2lky8I82I,6718
jax/experimental/jax2tf/tests/model_harness.py,sha256=1lzDFrtZV9XmDkkIxoQO0VtNPlmzptr1RUuf15mBjQ4,13951
jax/experimental/jax2tf/tests/models_test_main.py,sha256=JjtOP-nWEB2V8CDOhv5-w-Ob5_axXqtOb4qwBGZyK5E,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=jfFoKoha529Rpb4265D755gcwVvdUS8YtA60eCWpSiw,133842
jax/experimental/jax2tf/tests/sharding_test.py,sha256=P_0S0WYDlXzeJm0sI5xilAcXv9_8QmZFE-0vo_WnsgY,31365
jax/experimental/jax2tf/tests/tf_test_util.py,sha256=bjEogoTYFpaLFJmnga-C_WjBvAdLs7DwOi4h4xHwdZs,20137
jax/experimental/jet.py,sha256=4Lk7tOMSThQDy5zoH3QJ72Qo-s8B6jvEfUUgw2eQle0,26899
jax/experimental/key_reuse/__init__.py,sha256=t4C0gG8ht16IHAJZkE36_NyNIeO25ojiZan81y88MSo,1552
jax/experimental/key_reuse/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/key_reuse/__pycache__/_common.cpython-39.pyc,,
jax/experimental/key_reuse/__pycache__/_core.cpython-39.pyc,,
jax/experimental/key_reuse/__pycache__/_forwarding.cpython-39.pyc,,
jax/experimental/key_reuse/__pycache__/_simple.cpython-39.pyc,,
jax/experimental/key_reuse/_common.py,sha256=-VjPkYCZOOLR1y4EEY8-7TAfiBTCSQpgUWRXW_fWIpc,2348
jax/experimental/key_reuse/_core.py,sha256=S5i0koJ3xrz75u44fWYM8GyM4zRypOvtiO2GL9qP2vY,1737
jax/experimental/key_reuse/_forwarding.py,sha256=Dxtcs7oaHhYwciI2K7I-JRCBMadmZouFe_OUINjE7Aw,13298
jax/experimental/key_reuse/_simple.py,sha256=B5rSIcxpa_HOYnisNYlutB4YJXEz000hjP27uVaM5mQ,11895
jax/experimental/layout.py,sha256=yzQFzFbz1XVt6pwOHEDEFJndisvA7u06OrUJkYJyKAU,668
jax/experimental/maps.py,sha256=yjycDuNpT_aNQJK_XUBqhDDfY7qHUKeBpfi7w_zwSAs,964
jax/experimental/mesh_utils.py,sha256=6FHGYfdWskRrcgfMaIpTRBq_jxrTRA6EQi0m-PwSG8g,15823
jax/experimental/mosaic/__init__.py,sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0,770
jax/experimental/mosaic/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/mosaic/__pycache__/dialects.cpython-39.pyc,,
jax/experimental/mosaic/dialects.py,sha256=4f2cGGXxVUjHh_Jv1Q3hGOTfx8EyN6CWdwznrHwgbIk,791
jax/experimental/multihost_utils.py,sha256=nkwEgyKmwziWlW_6sgICadmUuG2_P8Web-Isk44ZWFo,16114
jax/experimental/ode.py,sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M,10659
jax/experimental/pallas/__init__.py,sha256=gar83kdE9ayr5AyA0pxzg7_253YgrXmxu7ArBS2Jpek,2217
jax/experimental/pallas/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/pallas/__pycache__/gpu.cpython-39.pyc,,
jax/experimental/pallas/__pycache__/tpu.cpython-39.pyc,,
jax/experimental/pallas/gpu.py,sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU,897
jax/experimental/pallas/ops/__init__.py,sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234,965
jax/experimental/pallas/ops/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/pallas/ops/__pycache__/attention.cpython-39.pyc,,
jax/experimental/pallas/ops/__pycache__/layer_norm.cpython-39.pyc,,
jax/experimental/pallas/ops/__pycache__/rms_norm.cpython-39.pyc,,
jax/experimental/pallas/ops/__pycache__/softmax.cpython-39.pyc,,
jax/experimental/pallas/ops/attention.py,sha256=Czxf_V3n6qMn_tUgh3jCIod21HqCQ33F7SXMS4cHBug,18891
jax/experimental/pallas/ops/layer_norm.py,sha256=H43nnnYXGjkc5j8EHguacLtQC3PBm258gaoabhXLjH0,10967
jax/experimental/pallas/ops/rms_norm.py,sha256=yE5gUqbjOQQfUzSJEryTKrHi-F1EN9NXPh4A4JU-OP4,9860
jax/experimental/pallas/ops/softmax.py,sha256=1DN8CrgJU2KZhFKf-VrXp5pbH64Yk-ocxkZmYUOWHDE,2647
jax/experimental/pallas/ops/tpu/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/experimental/pallas/ops/tpu/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/all_gather.cpython-39.pyc,,
jax/experimental/pallas/ops/tpu/__pycache__/flash_attention.cpython-39.pyc,,
jax/experimental/pallas/ops/tpu/all_gather.py,sha256=7QywIkmmDXtqcCCBYK2YXvvMAbrI15AM9LKDX1jcTUQ,5536
jax/experimental/pallas/ops/tpu/flash_attention.py,sha256=Ie6viezAca8s4Np6l2eC0cOLZRTxHOENcEFDnTuUk_k,48878
jax/experimental/pallas/tpu.py,sha256=0ilsv4z-VTA82HebQ-L_AD8miM3AXrofynONEHcxJ8M,1729
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=-ytQl3LPU31nJnxHC-eZEJcHx724XiCm9-ThYj2qaUk,3041
jax/experimental/shard_map.py,sha256=5JKq7L4hapb-uXHmQEustJ6AJusaJJ8yIlXUhZ3ZEZU,80045
jax/experimental/sparse/__init__.py,sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM,11186
jax/experimental/sparse/__pycache__/__init__.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/_base.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/_lowerings.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/ad.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/api.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/bcoo.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/bcsr.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/coo.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/csr.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/linalg.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/random.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/test_util.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/transform.cpython-39.pyc,,
jax/experimental/sparse/__pycache__/util.cpython-39.pyc,,
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=EzkW6zoTzy4t0qakVROQCoHy_QOMfmoq3dQqWsBWJYE,7578
jax/experimental/sparse/api.py,sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho,6631
jax/experimental/sparse/bcoo.py,sha256=SHYtb8Bln5QI3AE0LhnIk1eIGfSOsxNeBpACwmpHRjk,128191
jax/experimental/sparse/bcsr.py,sha256=OUfQcM1pEzJLdlh_TMOz6tzcsdiYLmQzXFdRV9bwoG8,38224
jax/experimental/sparse/coo.py,sha256=MKH3YpJvSroCUItkDlUwHZxkJnyG1AXPlLsO47_ooGY,24122
jax/experimental/sparse/csr.py,sha256=j7ib2vljbZl4gBJ8ijpLUO-cixg_4A4mK0c1emGcYgo,24422
jax/experimental/sparse/linalg.py,sha256=TViHHFbabrLUaLfCvaA1tRRNbwubh5JuHOvyBWV6D6A,23300
jax/experimental/sparse/random.py,sha256=HZeJZE9eqEr0d0gAwTQQZJKZMTFitnXSFm0zoIIsXEs,3888
jax/experimental/sparse/test_util.py,sha256=41Hvh5w84LiGxrZ1TPZWTQs1yiTSTCrhv5LP0E-89Kc,9597
jax/experimental/sparse/transform.py,sha256=MXD6sJrBLio729YHj2csDL9LSkNmuOW4uvCW9IsSg28,36919
jax/experimental/sparse/util.py,sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y,4381
jax/experimental/topologies.py,sha256=7WTF50_uUjV8J3TSK8aJyayBumliu5CIYraSwDOuoOQ,1897
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/extend/__init__.py,sha256=67qhpZPLPofDNC0vKaWo6f1j3K2kF6nZj2e9X8UrWUY,1225
jax/extend/__pycache__/__init__.cpython-39.pyc,,
jax/extend/__pycache__/core.cpython-39.pyc,,
jax/extend/__pycache__/linear_util.cpython-39.pyc,,
jax/extend/__pycache__/random.cpython-39.pyc,,
jax/extend/__pycache__/source_info_util.cpython-39.pyc,,
jax/extend/core.py,sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs,781
jax/extend/linear_util.py,sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8,986
jax/extend/mlir/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/__pycache__/__init__.cpython-39.pyc,,
jax/extend/mlir/__pycache__/ir.cpython-39.pyc,,
jax/extend/mlir/__pycache__/passmanager.cpython-39.pyc,,
jax/extend/mlir/dialects/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/extend/mlir/dialects/__pycache__/__init__.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/arith.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/builtin.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/chlo.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/func.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/math.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/memref.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/mhlo.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/scf.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/sparse_tensor.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/stablehlo.cpython-39.pyc,,
jax/extend/mlir/dialects/__pycache__/vector.cpython-39.pyc,,
jax/extend/mlir/dialects/arith.py,sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E,643
jax/extend/mlir/dialects/builtin.py,sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ,645
jax/extend/mlir/dialects/chlo.py,sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A,642
jax/extend/mlir/dialects/func.py,sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk,642
jax/extend/mlir/dialects/math.py,sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ,642
jax/extend/mlir/dialects/memref.py,sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w,644
jax/extend/mlir/dialects/mhlo.py,sha256=rkdPjP1q9tyRkG7JUoC65jjhBH3KVOx_pDe7ScEa-qI,642
jax/extend/mlir/dialects/scf.py,sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI,641
jax/extend/mlir/dialects/sparse_tensor.py,sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE,651
jax/extend/mlir/dialects/stablehlo.py,sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE,647
jax/extend/mlir/dialects/vector.py,sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI,644
jax/extend/mlir/ir.py,sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4,631
jax/extend/mlir/passmanager.py,sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI,640
jax/extend/random.py,sha256=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU,1083
jax/extend/source_info_util.py,sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI,1215
jax/flatten_util.py,sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY,645
jax/image/__init__.py,sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c,1027
jax/image/__pycache__/__init__.cpython-39.pyc,,
jax/interpreters/__init__.py,sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko,690
jax/interpreters/__pycache__/__init__.cpython-39.pyc,,
jax/interpreters/__pycache__/ad.cpython-39.pyc,,
jax/interpreters/__pycache__/batching.cpython-39.pyc,,
jax/interpreters/__pycache__/mlir.cpython-39.pyc,,
jax/interpreters/__pycache__/partial_eval.cpython-39.pyc,,
jax/interpreters/__pycache__/pxla.cpython-39.pyc,,
jax/interpreters/__pycache__/xla.cpython-39.pyc,,
jax/interpreters/ad.py,sha256=qVD0Ish5b9uGe6nTAsNQb6ZT8_IW1VvTdNqEbZpcaKk,3509
jax/interpreters/batching.py,sha256=LaCZQ2bg8xB3NdlsxkdI9DmPH0D5KJGrzSl6OYr6E2I,2688
jax/interpreters/mlir.py,sha256=t2-CXXKvIbtAJxqB0hxSDrcQzjQa-OUNk83PVunEpBY,2717
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=HrbfVvWEwOWJsCP810j9mfbk-Yf3XzH3B4iBlrp0f5s,12886
jax/lax/__pycache__/__init__.cpython-39.pyc,,
jax/lax/__pycache__/linalg.cpython-39.pyc,,
jax/lax/linalg.py,sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY,1016
jax/lib/__init__.py,sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA,767
jax/lib/__pycache__/__init__.cpython-39.pyc,,
jax/lib/__pycache__/xla_bridge.cpython-39.pyc,,
jax/lib/xla_bridge.py,sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw,840
jax/linear_util.py,sha256=vH-b8qwOfaImoYX2kFiJlVfoLFixU9Q9W1_0b-FlR-Y,2453
jax/monitoring.py,sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs,1191
jax/nn/__init__.py,sha256=o23x0ZOvhizFah-idwif160uoE8XivJvphR2-YuXXZs,1866
jax/nn/__pycache__/__init__.cpython-39.pyc,,
jax/nn/__pycache__/initializers.cpython-39.pyc,,
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/__pycache__/__init__.cpython-39.pyc,,
jax/numpy/__pycache__/fft.cpython-39.pyc,,
jax/numpy/__pycache__/linalg.cpython-39.pyc,,
jax/numpy/fft.py,sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk,1084
jax/numpy/linalg.py,sha256=o7XJ7Q8xurHhsetAp41P10TDu8L6AUimRfl2bMne7-o,1239
jax/ops/__init__.py,sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ,870
jax/ops/__pycache__/__init__.cpython-39.pyc,,
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/scipy/__init__.py,sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c,1474
jax/scipy/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/__pycache__/fft.cpython-39.pyc,,
jax/scipy/__pycache__/integrate.cpython-39.pyc,,
jax/scipy/__pycache__/linalg.cpython-39.pyc,,
jax/scipy/__pycache__/ndimage.cpython-39.pyc,,
jax/scipy/__pycache__/signal.cpython-39.pyc,,
jax/scipy/__pycache__/special.cpython-39.pyc,,
jax/scipy/cluster/__init__.py,sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI,768
jax/scipy/cluster/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/cluster/__pycache__/vq.cpython-39.pyc,,
jax/scipy/cluster/vq.py,sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM,776
jax/scipy/fft.py,sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE,809
jax/scipy/integrate.py,sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ,777
jax/scipy/interpolate/__init__.py,sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM,760
jax/scipy/interpolate/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/linalg.py,sha256=pd7kF8mgpRnaCxrY_DI2koCGleKYwl4tBSMIs4NMlQg,2012
jax/scipy/ndimage.py,sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ,788
jax/scipy/optimize/__init__.py,sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg,822
jax/scipy/optimize/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/signal.py,sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo,975
jax/scipy/sparse/__init__.py,sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4,757
jax/scipy/sparse/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/sparse/__pycache__/linalg.cpython-39.pyc,,
jax/scipy/sparse/linalg.py,sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc,810
jax/scipy/spatial/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/scipy/spatial/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/spatial/__pycache__/transform.cpython-39.pyc,,
jax/scipy/spatial/transform.py,sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro,802
jax/scipy/special.py,sha256=jdGeirmuJrppj9EFta_eeXa_YEPK8Ma5OGNZMNVlY6E,1568
jax/scipy/stats/__init__.py,sha256=cg4pntviV7r1RHbnnpKPIysGR-b4f2kx9nTFF6h0gOY,1983
jax/scipy/stats/__pycache__/__init__.cpython-39.pyc,,
jax/scipy/stats/__pycache__/bernoulli.cpython-39.pyc,,
jax/scipy/stats/__pycache__/beta.cpython-39.pyc,,
jax/scipy/stats/__pycache__/betabinom.cpython-39.pyc,,
jax/scipy/stats/__pycache__/binom.cpython-39.pyc,,
jax/scipy/stats/__pycache__/cauchy.cpython-39.pyc,,
jax/scipy/stats/__pycache__/chi2.cpython-39.pyc,,
jax/scipy/stats/__pycache__/dirichlet.cpython-39.pyc,,
jax/scipy/stats/__pycache__/expon.cpython-39.pyc,,
jax/scipy/stats/__pycache__/gamma.cpython-39.pyc,,
jax/scipy/stats/__pycache__/gennorm.cpython-39.pyc,,
jax/scipy/stats/__pycache__/geom.cpython-39.pyc,,
jax/scipy/stats/__pycache__/laplace.cpython-39.pyc,,
jax/scipy/stats/__pycache__/logistic.cpython-39.pyc,,
jax/scipy/stats/__pycache__/multinomial.cpython-39.pyc,,
jax/scipy/stats/__pycache__/multivariate_normal.cpython-39.pyc,,
jax/scipy/stats/__pycache__/nbinom.cpython-39.pyc,,
jax/scipy/stats/__pycache__/norm.cpython-39.pyc,,
jax/scipy/stats/__pycache__/pareto.cpython-39.pyc,,
jax/scipy/stats/__pycache__/poisson.cpython-39.pyc,,
jax/scipy/stats/__pycache__/t.cpython-39.pyc,,
jax/scipy/stats/__pycache__/truncnorm.cpython-39.pyc,,
jax/scipy/stats/__pycache__/uniform.cpython-39.pyc,,
jax/scipy/stats/__pycache__/vonmises.cpython-39.pyc,,
jax/scipy/stats/__pycache__/wrapcauchy.cpython-39.pyc,,
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/sharding.py,sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94,1184
jax/stages.py,sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg,1272
jax/test_util.py,sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU,835
jax/tools/__init__.py,sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ,581
jax/tools/__pycache__/__init__.cpython-39.pyc,,
jax/tools/__pycache__/build_utils.cpython-39.pyc,,
jax/tools/__pycache__/colab_tpu.cpython-39.pyc,,
jax/tools/__pycache__/jax_to_ir.cpython-39.pyc,,
jax/tools/build_utils.py,sha256=3wYMEgwJrTspNXN4LqvTBzDw__Y0Rj7LS5IYDzB1aFQ,3300
jax/tools/colab_tpu.py,sha256=M8F2u_rcvDdTKHB502zhticEZEqNGE8j5dzgPwoELgo,1553
jax/tools/jax_to_ir.py,sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU,8684
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=IktXqEnzwyoY0JKe-wiKrzh6rdCrZTlKeyiplE-IiBg,5583
