jax

View on PyPIReverse Dependencies (1220)

0.5.0 jax-0.5.0-py3-none-any.whl

Wheel Details

Project: jax
Version: 0.5.0
Filename: jax-0.5.0-py3-none-any.whl
Download: [link]
Size: 2270365
MD5: ee40769645faa46c394131761981ad22
SHA256: b3907aa87ae2c340b39cdbf80c07a74550369cafcaf7398fb60ba58d167345ab
Uploaded: 2025-01-17 17:40:30 +0000

dist-info

METADATA

Metadata-Version: 2.2
Name: jax
Version: 0.5.0
Summary: Differentiate, compile, and transform Numpy code.
Author: JAX team
Author-Email: jax-dev[at]google.com
Home-Page: https://github.com/jax-ml/jax
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.10
Requires-Dist: jaxlib (<=0.5.0,>=0.5.0)
Requires-Dist: ml_dtypes (>=0.4.0)
Requires-Dist: numpy (>=1.25)
Requires-Dist: numpy (>=1.26.0); python_version >= "3.12"
Requires-Dist: opt_einsum
Requires-Dist: scipy (>=1.11.1)
Requires-Dist: jaxlib (==0.5.0); extra == "minimum-jaxlib"
Requires-Dist: jaxlib (==0.4.38); extra == "ci"
Requires-Dist: jaxlib (<=0.5.0,>=0.5.0); extra == "tpu"
Requires-Dist: libtpu-nightly (==0.1.dev20241010+nightly.cleanup); extra == "tpu"
Requires-Dist: libtpu (==0.0.8); extra == "tpu"
Requires-Dist: requests; extra == "tpu"
Requires-Dist: jaxlib (==0.5.0); extra == "cuda"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.5.0,>=0.5.0); extra == "cuda"
Requires-Dist: jaxlib (==0.5.0); extra == "cuda12"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.5.0,>=0.5.0); extra == "cuda12"
Requires-Dist: jaxlib (==0.5.0); extra == "cuda12-pip"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.5.0,>=0.5.0); extra == "cuda12-pip"
Requires-Dist: jaxlib (==0.5.0); extra == "cuda12-local"
Requires-Dist: jax-cuda12-plugin (==0.5.0); extra == "cuda12-local"
Requires-Dist: jaxlib (==0.5.0); extra == "rocm"
Requires-Dist: jax-rocm60-plugin (<=0.5.0,>=0.5.0); extra == "rocm"
Requires-Dist: kubernetes; extra == "k8s"
Provides-Extra: minimum-jaxlib
Provides-Extra: cpu
Provides-Extra: ci
Provides-Extra: tpu
Provides-Extra: cuda
Provides-Extra: cuda12
Provides-Extra: cuda12-pip
Provides-Extra: cuda12-local
Provides-Extra: rocm
Provides-Extra: k8s
Description-Content-Type: text/markdown
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
License-File: LICENSE
License-File: AUTHORS
[Description omitted; length: 20734 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.8.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
jax/__init__.py sha256=1hi3Rd3-wtSdX1G1_y5PXVZZSeiq8TSrbE6Ql_BPhNI 10083
jax/ad_checkpoint.py sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U 932
jax/api_util.py sha256=VhdFXkIFfQbWc8RM-eNBSBmil1vHsAFPctXKWrrRJIw 951
jax/cloud_tpu_init.py sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs 651
jax/collect_profile.py sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk 4730
jax/core.py sha256=7a9brtV50J4mrTqfdAkH6Flt1ty7hQISyUplzAJZ4WU 12266
jax/custom_batching.py sha256=MEoypZfGgYqvZrBKtnvL5sQjl8YrjVm11hiFWoTCtV8 691
jax/custom_derivatives.py sha256=-yrA5kDE7gDErJFdartoj43ELYeogp6PI8kjP8p_TLk 1453
jax/custom_transpose.py sha256=vGj3GjRJQpxl-MhvYztIedauNBiWMj9nD_aJHQiNaYc 664
jax/debug.py sha256=HXdNZlaM5LxtEYS6fsH3J4LRgiDVL-5W6AXcimbFS5o 1083
jax/distributed.py sha256=D9-SeP4MWzq9jw2itQrlm10cn5BtR-9Zjn2YGzEXBDU 673
jax/dlpack.py sha256=TWzMqAIedhARYUrnreJ35tEYt0SepzrpvTQtmnSdhEw 710
jax/dtypes.py sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I 1188
jax/errors.py sha256=MaZbeSxeRyWRMfq4cj6P0GA9UFykLgX3aURPAKhqbts 1363
jax/export.py sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA 1537
jax/ffi.py sha256=1bu9CI_NynECrQ83ps09-hNJt3MSuAjOtePAOuYeLG8 958
jax/flatten_util.py sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY 645
jax/monitoring.py sha256=U87O_UHrN3UhWxY91BJi7r26ZsGPi0zoJODNVjIv5K0 1321
jax/profiler.py sha256=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c 1119
jax/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/random.py sha256=TG_JPGzvJzNvfy2OB_24N-5m3e9U7lcwgyQ5U03jhAY 9416
jax/sharding.py sha256=tCEKtRHY6fAij2jyyyYJDfbqgQJ3EJGsI69NEBWdnLE 1686
jax/stages.py sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218 1314
jax/test_util.py sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs 835
jax/tree.py sha256=tI9yJ2PIEhAFiSBbVKnxV1Wfly6mb3b3mJXECzugJs0 1078
jax/tree_util.py sha256=ZCuOkjx2V6A0EVV3e9sp7JtvhI5sxHF7u1-Vqpak-es 3125
jax/typing.py sha256=CrQpi9UTB9MYfg-zEsqP3n2qe28hlY_QEsgY-9mrtAs 3331
jax/util.py sha256=OXU0bu3n2-y901BwEhxd1CvOeGoEBaXPdvYaVNHOT3E 1135
jax/version.py sha256=pK-MCVzioPoUsrrAfTsI_FARKDUUDw-i58hm2FZSer8 6009
jax/_src/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/abstract_arrays.py sha256=oCNOkD0OfTHrteBJa1SjR2QbgM1PUVFGPqsqNdaM0Cw 2834
jax/_src/ad_checkpoint.py sha256=bvDcC_v92DJArsyA1qzS_CgRxrU1C-emZmy6AYmQ4GI 39083
jax/_src/ad_util.py sha256=n14jlpCC3f7itcVEL8ijSM_C4ofyB8wbKKTzwWzxbr0 4164
jax/_src/api.py sha256=vkAl8nqHljLvbaLVC45SwRLBAuoyUv8mbI7SjEnQNII 121612
jax/_src/api_util.py sha256=aDJ23YheNkE83HADg_UtWpjCnO8t_9PmYQACliVyaD0 28210
jax/_src/array.py sha256=iZZR4Tfm2CVtwDPVszz60bR4OCLXujo6Uo9X-Hxry8Y 46524
jax/_src/basearray.py sha256=12hYSzFDsq0LVpPX8nQ-gH5mLE253PBWRSnYvQsuVwQ 6503
jax/_src/basearray.pyi sha256=D9Q7W2_DF0LkP5BpXRFZROcV2VzNBBgAMCo_QLax2ZE 13146
jax/_src/blocked_sampler.py sha256=rYk-BNwg0fIXWqgMXG5v3kjJw66Bod7vjNSQTa8atuw 5666
jax/_src/cache_key.py sha256=8ytVe0xpceRekMKyEUxC0T3GaEKAv_JgreT7xPuCB5M 12883
jax/_src/callback.py sha256=CB9sEZwErI5o_aFyDGTZspFOQhbf__h7DMbEJ5YBoTc 24618
jax/_src/checkify.py sha256=CrpCwsdNd8OoNZEXxcM84JTiyF6manovjdx25iwA4Uw 56193
jax/_src/cloud_tpu_init.py sha256=SVyU-xfB2YxNfZtlv2ebpEgoSnp64WUlnyJuutz_oag 3519
jax/_src/compilation_cache.py sha256=m2pr7zVMcYgCY2UsyaajGp5jTEZ1tztrh4YKCiYeZM0 11226
jax/_src/compilation_cache_interface.py sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo 887
jax/_src/compiler.py sha256=xmbeRCHdTRMD17tvGtFG8PqUYlDiayN9iZYfj4QRapY 28408
jax/_src/compute_on.py sha256=GYS3AbPWjfRgw9SsCFa7R7q2dpTluQeejqTjnQ3iVTk 2074
jax/_src/config.py sha256=UFNH_zNvArzW-uPOOCK8bG974KCppdiX1VCXE5uZSZA 67040
jax/_src/core.py sha256=SMoYSheXXhO0M2Js8ZgZMElfldCiQYqUHrSO__GOpHM 110662
jax/_src/custom_api_util.py sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw 876
jax/_src/custom_batching.py sha256=GEjpbvKYWgsOYJzqaCPgvFz9zrKI4KLMN6ueq00vzsI 14295
jax/_src/custom_derivatives.py sha256=KRG82zCJskH-Nj6nVaKwdek_e1r_KJaHuvK3sbxuNS8 70662
jax/_src/custom_partitioning.py sha256=78UBGhe6ze22mbZZAzTuK0IuPyhRA0RbybNnM-W-0Hs 25122
jax/_src/custom_partitioning_sharding_rule.py sha256=KmL6kqH6drwwh1EgLcxIJyOkNK3poBV60dM4eYnqRu4 16227
jax/_src/custom_transpose.py sha256=8UOwMc39nsG90p1jDBw50_QZ8YX5a_qFP1CsFYFoV0c 7788
jax/_src/debugging.py sha256=MqC9zpSlxQ35OZ4yoXwhKgtTYDVTsvfvD1xRhD6vNJM 26805
jax/_src/deprecations.py sha256=FO0YxHbuhn-h2kFyzUH3Aj93PdQUSFqWyMjFRuyyFiQ 4784
jax/_src/dispatch.py sha256=PvzMW8MWFn7iHVJrIga1LUwh8QG6Wnco897QwUbE87A 23034
jax/_src/distributed.py sha256=FRYsUdt6u8i_7Dd6YpTh7ih53_CzvS8ngrSGncCbPDE 12502
jax/_src/dlpack.py sha256=pzQdxNalm0JLOOZJyHaLYSAT5DfP_UTPHFl5lBSX9SU 11544
jax/_src/dtypes.py sha256=ZMFmO2hBBDheBvD7tZmnKbjvJWYYb4ccTI9k6Jlw56A 34409
jax/_src/earray.py sha256=3xAOp69wfWoKhBSu1pBIWlXKfO9CR72NPApGdVQCTTY 4399
jax/_src/effects.py sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q 4682
jax/_src/environment_info.py sha256=TLwjyscnAh8rrQWePe1b14-IeSc-z-XP51Y92dMQeGI 1873
jax/_src/errors.py sha256=s3zDLKcfvcCVSrJgNON0YF7bb7RB8Gp-xWinVHcvsNU 24741
jax/_src/ffi.py sha256=E03VNdsJwkniUwyD0cdtmnK0iqYFbG1mXfnVASqASck 24751
jax/_src/flatten_util.py sha256=t0PwBrSY7U22mLp_-xz_UJju-SBsCgtKDJwCRyPQllE 3556
jax/_src/hardware_utils.py sha256=L03Dapnam8MdnSyBGq2d8nio5gPF51xd_Ufxw2pVjCE 1728
jax/_src/jaxpr_util.py sha256=Jv3fcRlWFNUt-WQwHYfmLeXOSKAYka9RKEwrkYl-BkQ 6439
jax/_src/lax_reference.py sha256=YPZTCI52PssuXAEnMn4r44ZZ2jDHB0wYj8d_fJDrCnw 19282
jax/_src/layout.py sha256=Km3wPwPjWyBE37S4V0prIvE0FqI56Nq1u0TCVhuL3tE 5375
jax/_src/lazy_loader.py sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw 1842
jax/_src/linear_util.py sha256=9CPjcxRPyefG-j_xxtYWJ-EXQ1rh1FAmgp1cirjPHSw 13885
jax/_src/logging_config.py sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk 4452
jax/_src/lru_cache.py sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s 6661
jax/_src/mesh.py sha256=1rgkxOx5CClLEjfaefJZ7rM_WbWngeNydTnhYt9LMaU 18412
jax/_src/mesh_utils.py sha256=AYtLuCPor_YRW3jAqS-NmiAKRCaoOmpDnOioLX7Le2U 32394
jax/_src/monitoring.py sha256=Z5ol4haHZeaj11ZHKFNgBOUXXSjeOHYmg-xS9ZNQpZw 5188
jax/_src/op_shardings.py sha256=gsygmZDpugleC2AkgHEoq6JIDo-ACOK-O7BoJjmyWO8 3894
jax/_src/partition_spec.py sha256=qnecyMIbJJjWABMiZXzjXu_3WSB3TFF5ZL5Kwitw7J0 2186
jax/_src/path.py sha256=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A 1452
jax/_src/pickle_util.py sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ 2312
jax/_src/pjit.py sha256=H0G80p1fEjV_o6w1HSxhAEAeFa6CAmsaNVVQF-3rq-o 127316
jax/_src/pretty_printer.py sha256=iRZHOftfSZUYXoFmBxLWM58CXEuwvlDVhJk0Njhc_QY 15460
jax/_src/prng.py sha256=lEsvN_n4RaDRko1-G_qhhuZOgrh9aj6TMD0OZxSgSAo 44596
jax/_src/profiler.py sha256=278hQdUi8B7FVXo-YKhY9jf1qXSaBNem50ZAeB6F4DI 16195
jax/_src/public_test_util.py sha256=SZxqWnOEobpLnMpXA18FAFrhdU9YG0y-JvTZDTE2p_g 11152
jax/_src/random.py sha256=-0fcrzyHs4w9dNgahL_hpTDou-5x6OuCA5kNhBpRsak 100275
jax/_src/shard_alike.py sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg 3989
jax/_src/sharding.py sha256=frnT8rcK_erouiHIyySG20lFYlZXjfzmh3B9WtXEzMw 8102
jax/_src/sharding_impls.py sha256=CgbyAVKGTDGB3h4McnPFKgVSeou_ro6z7dkQ2QHk_Gg 69767
jax/_src/sharding_specs.py sha256=ql8VcvciCOMmrc2TXnb4khTatU_in3BOAc90QpJDMl0 8646
jax/_src/source_info_util.py sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE 10140
jax/_src/sourcemap.py sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y 6640
jax/_src/stages.py sha256=ydC3LteN1CBJsL-pv1EOP-vCIe6NZ4lzdnMTccczUI4 29971
jax/_src/test_util.py sha256=Rd5QjqjbaWxl6ni2uH1abN5jYdYxD_gL_SjpJkKtUXc 84347
jax/_src/test_warning_util.py sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE 4074
jax/_src/tpu_custom_call.py sha256=FUWVvoit4lV9aPwIw7yBOSFy8fCiq68a533hQ8X8Xbc 27331
jax/_src/traceback_util.py sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo 9103
jax/_src/tree.py sha256=SIwfh6UL_xEuEMq-ky4bmOmoq1LllYa0ZUhmh4l4-gM 12592
jax/_src/tree_util.py sha256=vf7NEirISobznTDhWaj2tp5PatvtcG3PVre_1HZURsc 46800
jax/_src/typing.py sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw 3287
jax/_src/util.py sha256=29srTUV7fDiMZfgvcoWFdvY85EJh2XogJuLdwiQcA74 20244
jax/_src/xla_bridge.py sha256=WT0k1-tQ55QvJ5_S_h5XKzMZeg1-AczSAByUztKMD4U 47143
jax/_src/xla_metadata.py sha256=IvJL8QejbnCGogGcaTCMMcseAWsKW0JouZrCRzUxV9c 1519
jax/_src/clusters/__init__.py sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94 1283
jax/_src/clusters/cloud_tpu_cluster.py sha256=xpeiJ-pVH6uNypUfIoxwNVaRuATiEtIqGVtNrvYN_lE 8405
jax/_src/clusters/cluster.py sha256=4_heeOPzwH7NHumJKZspEfCReABCJC9SDfW8z3ORZp0 5856
jax/_src/clusters/k8s_cluster.py sha256=TzUXU4ksY11X2f6iYjhb5LEGZfr6r5WZM5tc9DQhvBQ 4188
jax/_src/clusters/mpi4py_cluster.py sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg 2608
jax/_src/clusters/ompi_cluster.py sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ 2317
jax/_src/clusters/slurm_cluster.py sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0 2321
jax/_src/cudnn/__init__.py sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A 631
jax/_src/cudnn/fused_attention_stablehlo.py sha256=O9xGbgwTjSk1sBIWF0XkHQ5ZjQYVAujYWMdehI5UiS8 65649
jax/_src/cudnn/fusion.py sha256=3ifqe61DuTyoOXQL7Nelacz2KRdcJdPRI__ub4s_vI8 2808
jax/_src/debugger/__init__.py sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q 899
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=WilcRMqpzssKa5YL_P7SFtKaUVUaXqDutxO_N-ZRnp0 8593
jax/_src/debugger/web_debugger.py sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w 3433
jax/_src/export/__init__.py sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU 581
jax/_src/export/_export.py sha256=Vjx84UCtpfGJ7QsRgdfh4t_d3CxLHuQJ6_a50A25lXs 66094
jax/_src/export/serialization.py sha256=WLEwoP7dnrQSBnz73KzXQXPs88Cl3jdrtjnpLiY4s9w 19203
jax/_src/export/serialization_generated.py sha256=hd6uuEJUksQMgq5_SMVN06lfVaC41PPnMR0HVhngqVM 27772
jax/_src/export/shape_poly.py sha256=prgFrctrTTs4b0L5AJoNZ8Sk07s3q8qc6Wx1qjQ7rLY 85937
jax/_src/export/shape_poly_decision.py sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc 20476
jax/_src/extend/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/extend/random.py sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw 1258
jax/_src/image/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/image/scale.py sha256=_Hezs_YE3kCj6_cTn6XvMuU26sPiedsiJR8-VVhKEDA 13740
jax/_src/internal_test_util/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/internal_test_util/deprecation_module.py sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4 851
jax/_src/internal_test_util/export_back_compat_test_util.py sha256=S8wDJb98vYYeIq-omRo_b8jErOBa2DJAoX9mO1V-cI8 15336
jax/_src/internal_test_util/lax_test_util.py sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo 12140
jax/_src/internal_test_util/test_harnesses.py sha256=d3BdtWFyEKoB8v8vWTbp01UfhQkzHagQkpSptra28zc 126010
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=Jz1t4Zbm1Ed56aehf36GRRcB-63ykATN-jeZjD3Uu08 48887
jax/_src/interpreters/batching.py sha256=esSUzj-TcBvjMSQbVxhx56fQEf8nfMoALVKR8NAUH6c 47526
jax/_src/interpreters/mlir.py sha256=fLtNZ2Jb4JjGG-KIXZ44HpufF0Wf11ctxJk35KWZnG4 135647
jax/_src/interpreters/partial_eval.py sha256=yZ00yHhKG-3WoX49KoIgH9iYzsYjgSaGs2cl13vzaMg 115041
jax/_src/interpreters/pxla.py sha256=rztL-j6uuq9RMD2ylAC8_2huWK8dRjji2a5dlyP-wBs 133962
jax/_src/interpreters/xla.py sha256=h7Sbg8U01o0vXNCMN-cX7TBX89i7cl01tXt-xLAM8Ys 5424
jax/_src/lax/__init__.py sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI 690
jax/_src/lax/ann.py sha256=53ap20vxBD-hV6L1JZsLNfzyACtK8qkE2fmPqsJivvk 16761
jax/_src/lax/convolution.py sha256=kxGQ5J18irVz3S1sxqoqlTW6otJO_5g3zPCHi09FRhY 44551
jax/_src/lax/eigh.py sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ 22235
jax/_src/lax/fft.py sha256=fIrp5WrJCZM0FYZpZnmC67Y7rsWCthbOwK3GXryoEqQ 7258
jax/_src/lax/lax.py sha256=IbucuO3UBvldke8INUVIob1l5JzpBMTdqyD8BniGTbU 276914
jax/_src/lax/linalg.py sha256=oKNi8k4seAYfggRu1m7tUzMnnGWRE8Du4pTWl7UuslQ 112605
jax/_src/lax/other.py sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc 12497
jax/_src/lax/parallel.py sha256=ZhXlmszqPHjDj4yAFJV7vavV9yDn3F08rlAc7t1z5pY 72887
jax/_src/lax/qdwh.py sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo 9391
jax/_src/lax/slicing.py sha256=lKwyNH5axa0xT05GbdWniCWWC9CtiAxfDDtfPbU6wWc 129293
jax/_src/lax/special.py sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY 26940
jax/_src/lax/stack.py sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY 2560
jax/_src/lax/svd.py sha256=3ol5rMgm_uy40yZzKs-9BoDwFn4oUqbcDWbKcvvVok0 8826
jax/_src/lax/utils.py sha256=smrb7YyChN3oUAbv1kvV0YYbWQ8ORKONuZRqQpt_sDU 4742
jax/_src/lax/windowed_reductions.py sha256=WtfWaXuWCxCCYrKtRqwtDS0x2if8wl7NWbbJlwpN-Ac 42337
jax/_src/lax/control_flow/__init__.py sha256=SUQsNr-Ctcs2Vgmb4HIzqVidxEpAHFe1J9WsM0aMegw 2200
jax/_src/lax/control_flow/common.py sha256=xA0fYk0uaF6lbCn4hjK_lxE1BdYM0aLCOiXR2CoacsE 10985
jax/_src/lax/control_flow/conditionals.py sha256=ke5dmbheCscj_r6fv9SMtsxv_PNbRrtsAiT-nmipWoE 43321
jax/_src/lax/control_flow/for_loop.py sha256=Ph-U0L_ih9aIzwUOuDZ4FHnjkHaoKI9zQOqh4qZRbDM 35763
jax/_src/lax/control_flow/loops.py sha256=tatbiLmoSIIbn5L1qy05wH3ie3qhvVgrvJyHDj_pDbg 112104
jax/_src/lax/control_flow/solves.py sha256=TPPv-SJvh44F3cUI8PfM6UtR6rvLn0PncHnJdMpVXCw 18383
jax/_src/lib/__init__.py sha256=bUYdjvw8d2XkBpIYLe2i31-v8yu5BEMlM7MpdY4uBa4 6807
jax/_src/lib/mosaic_gpu.py sha256=CebeF4-0kxDoMd2t_JH6JZSy_2iJyvjrcwskHK9jbhs 986
jax/_src/lib/triton.py sha256=6zzcpqutI1cZW4sG9tLwPFyPDw35xhpPoNH8wkG-VbE 662
jax/_src/lib/mlir/__init__.py sha256=UoirBDcNhqLa8Gb1Qrxd4wYxScz_p-1tb-ewbTsK1jM 733
jax/_src/lib/mlir/dialects/__init__.py sha256=6KUUW3fBmKZn0eM7VlIFh5yiSLRQ9PNC1Cl9tAxoF2o 2032
jax/_src/nn/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/nn/functions.py sha256=A-_85Ox3bfFrjaTmYYPQ-n0xcR5oBczLcSW9M-3xVKc 36635
jax/_src/nn/initializers.py sha256=TOGKwbExnXQvMai1amFpEH2PFaZxiVz95KGBVaopnmA 25450
jax/_src/numpy/__init__.py sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M 581
jax/_src/numpy/array_api_metadata.py sha256=je15NAJj_CFlgqeDqo5iSg9EzmN9IwCkZ_E8TtStKcg 3347
jax/_src/numpy/array_methods.py sha256=87w0psHs65RwZ0Wwf-AKiKIVFyTE4OofHjsmDs82OjI 46147
jax/_src/numpy/fft.py sha256=PBeG_3pGNRqDW1RqgmJYvZ56JOhKRHLyYgBZLHv1gGw 52614
jax/_src/numpy/index_tricks.py sha256=aRFc9So43wmgzk-eVGBeN16BI4Ny7eXYFr7yEtrgDqM 9992
jax/_src/numpy/lax_numpy.py sha256=Orx5dRn9RF19U5KMHiux76RZc4BqUYMN0zrBO46kxGA 486731
jax/_src/numpy/linalg.py sha256=zZCc13Hu6hwY3XXguJdvfyl1vAJH4-Qsl4-WvKRrFWE 77762
jax/_src/numpy/polynomial.py sha256=LU6iRWNBBHG2BmYztPVH4tE-5ZWTnCGVzTNNTXBHxqs 27913
jax/_src/numpy/reductions.py sha256=TOJCHM-X8B9hW0p9snD1O2Nl89TqyZ-bEH6HaS8ybIU 108968
jax/_src/numpy/setops.py sha256=Uptcxj3KK04zB5P4wTrr0Qoj0INwDl44d9zEQmvXl8E 47405
jax/_src/numpy/ufunc_api.py sha256=VXdn5cXraEXPjBXUTTU_jhWFGrdOF5NlrDEXHBhyn0w 24204
jax/_src/numpy/ufuncs.py sha256=i8KwYXArPAPHt9N9DsvHo3N357vv-6D5L2hRX1LJWLM 116554
jax/_src/numpy/util.py sha256=fwLeCdCAbhkjiOlNCHIwklKr4fWaox9otNRXwBNcKok 11953
jax/_src/numpy/vectorize.py sha256=lugYJ7HrEpG-L0yNd9UD9mEa6ASQlaaACdb72PFtat4 14018
jax/_src/ops/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/ops/scatter.py sha256=wsV-jnH9epkp-GjIy8AG9xgpVoBUrPEmaj6h0eXhBis 17625
jax/_src/ops/special.py sha256=tm7RbrE3fHxkBi3eU9UjYUP0zm69aQYQ_3Mql6wYnRg 4206
jax/_src/pallas/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/core.py sha256=pG7VHnsPPdj0afBhHpAKBMJwZJ1vw8V8DDY7vuWkhTw 40125
jax/_src/pallas/cost_estimate.py sha256=RtYcdNqOTQx9NQbnR_rxZY2HHXbEo2xzuQa_Uj3vGyM 8283
jax/_src/pallas/helpers.py sha256=DNwbVLgsI9mzJIix6VlW0mnkCQ4qxde9T81yRzaP2rg 1504
jax/_src/pallas/pallas_call.py sha256=RiC3-7C8S9Awy9NxfTdR5GdA1kAb3WkYhZWXdl4LAo4 74741
jax/_src/pallas/primitives.py sha256=EdLlbqfdacTjKIXiVTwAtpcfNeM_koE-zT2cjI25K3U 32583
jax/_src/pallas/utils.py sha256=tNjoP20oRWC4A-ns5aSUB5LjRCS11fy4E1SH4rw-8-M 14067
jax/_src/pallas/mosaic/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/mosaic/core.py sha256=XsuZsozi87im_osSLyNBEKAl35zr871DFOXmKLNY9w4 8707
jax/_src/pallas/mosaic/error_handling.py sha256=nmjLsqhnE9-foi7lwMAWdWNQMdHd2jOMqJtDxhBhr1o 5480
jax/_src/pallas/mosaic/helpers.py sha256=4Q4uzzivCTnSwvYCTM-3y4g4bUFWWQjCse2xAhBRtLY 1775
jax/_src/pallas/mosaic/lowering.py sha256=Il5ybXGnO0TTIJMuKkfSoxJQUqAPbEWBlBMwdaGAQjg 124828
jax/_src/pallas/mosaic/pallas_call_registration.py sha256=sGETuDHKZQEqCVOluCDymrKEC1H1k9_KR0i2Kr7bUik 9239
jax/_src/pallas/mosaic/pipeline.py sha256=8ShG9SBvCySphlB-GC2glH8nDyU3euVwXt9_rH4H9Ag 44060
jax/_src/pallas/mosaic/primitives.py sha256=iP7ROK4kwpcXWoNMozzYO_uZCrl8VK2igDvKI4auE4E 30418
jax/_src/pallas/mosaic/random.py sha256=yBP4tvNPUodwWMSssHDUJkwW92uU3K2wtvziRg00CpM 7989
jax/_src/pallas/mosaic/verification.py sha256=fQAVzbETThqBWA_yp2jLbeX_5meoVShh3flYgQWiF9g 24540
jax/_src/pallas/mosaic_gpu/__init__.py sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU 581
jax/_src/pallas/mosaic_gpu/core.py sha256=6xederx6OkOkbZcMOHI0sSTyZvnJ1wv-zrgWqkgKNgs 17873
jax/_src/pallas/mosaic_gpu/lowering.py sha256=7b5dO4bV-H_QvQKufvp4mAVg_AMqseSAE0_0ADHsFOI 66628
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py sha256=ldMJecjZB80KVXZC9pq9ZM0i4WqGXYz-DrbHDGgIV6U 3463
jax/_src/pallas/mosaic_gpu/pipeline.py sha256=BLTeaPMSKCWUx8AedqV-c5EJRJmx0v83fTiASBsMoTM 21788
jax/_src/pallas/mosaic_gpu/primitives.py sha256=-fUCUHzilQYjUYq7XvR9XNoaJSwZYTSouGsQ15IoL54 24980
jax/_src/pallas/triton/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/triton/core.py sha256=5u6hubCM5N7aSYK5iExZpffEGrmlpqR6Op1xd3dOhDk 1374
jax/_src/pallas/triton/lowering.py sha256=eZfbXa9Slq5rd3FSVTIctFw3GnTxo037uWiiO6DCK40 89570
jax/_src/pallas/triton/pallas_call_registration.py sha256=RBttMRLdzpSTvZe0SoXQ_43rQYZHEsYD57KlRod-wLg 4395
jax/_src/pallas/triton/primitives.py sha256=_UjvgpOsEaRzul3xizwzPdP1blZ1-NzChzZ4bsVmSaA 4228
jax/_src/scipy/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/fft.py sha256=NFy_bDIzx44Lpkw1-74fBm2jxeXmzIUp8LTVh3-6Z5s 15281
jax/_src/scipy/integrate.py sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0 2346
jax/_src/scipy/linalg.py sha256=dAiqOEVVJamlBhQtk25NnEyNoyVs_FClaWOTOLsR4nY 77382
jax/_src/scipy/ndimage.py sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc 7092
jax/_src/scipy/signal.py sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs 47453
jax/_src/scipy/special.py sha256=-F5ApAboZ3LyQFwzWWOUBtEdt3tgm-ZIXOs8VhVzpBY 82636
jax/_src/scipy/cluster/__init__.py sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8 581
jax/_src/scipy/cluster/vq.py sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc 3043
jax/_src/scipy/interpolate/__init__.py sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8 581
jax/_src/scipy/optimize/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/optimize/_lbfgs.py sha256=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw 7769
jax/_src/scipy/optimize/bfgs.py sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE 5609
jax/_src/scipy/optimize/line_search.py sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc 13235
jax/_src/scipy/optimize/minimize.py sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs 4979
jax/_src/scipy/sparse/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/sparse/linalg.py sha256=IGX8lZ2il77S3WA50Lz7MdBiw7lqdD8mHgWRaCUHX6k 27245
jax/_src/scipy/spatial/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/scipy/spatial/transform.py sha256=yD9HOfgQpQO3-eACTh5zW8qzV-WLcrTHY_5ZlsnBoSk 17525
jax/_src/scipy/stats/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/stats/_core.py sha256=Gq8H86uI_wWodhyvdEy5GiNulBfW3kJZvm8Z0Qw6Vdc 10653
jax/_src/scipy/stats/bernoulli.py sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo 4541
jax/_src/scipy/stats/beta.py sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME 7988
jax/_src/scipy/stats/betabinom.py sha256=dmhOAUk6py5Fnu-jR95KnwJ6suvvfwNF9CXNHz73CSc 3447
jax/_src/scipy/stats/binom.py sha256=E7a3go8bCM5bSiC72l1pDQEuuzSpXLmgfG40iPF90wA 2825
jax/_src/scipy/stats/cauchy.py sha256=26eHaNIfxwh1og3kTqnPYgivH0GTIuWVwElai4VJwv8 8884
jax/_src/scipy/stats/chi2.py sha256=vPFrhpbLRPDheYenmHQ7QvZrRt8cAyRtOpU9VmemNIY 8008
jax/_src/scipy/stats/dirichlet.py sha256=_wWyx07z62_EH25i3J4jS5q53oj0UH5A3-XgQzapwbY 3198
jax/_src/scipy/stats/expon.py sha256=8q5nUo_N8Rl8BjMVfFKv2ydB-x7-TkKtIdZi5FVrbl4 2268
jax/_src/scipy/stats/gamma.py sha256=W5B8z6LPD3Zi3cbcWHDB-RU_OuKjXopJIspf_VKZMw0 6987
jax/_src/scipy/stats/gennorm.py sha256=kp22L5H-rR5n7sn9rpeWsWta-F76yZaXkO0Ovvf9zc8 2989
jax/_src/scipy/stats/geom.py sha256=2hQ4LoaqNHFBahdRTGS9oyWMYdFQufJb2ByqOw9709s 2243
jax/_src/scipy/stats/kde.py sha256=Mi9nMB6q3eueSbyNMfs3DokLhsjte0lYsSIdAZ9Birc 10141
jax/_src/scipy/stats/laplace.py sha256=9PlK08r9wRHelqSIuKkv8KGO0oPV2AArI0Fs6rICFj0 3269
jax/_src/scipy/stats/logistic.py sha256=HB3Brjwwl-yyGB_9K1lpdiQlmitsHwNLfgl0g4owFSM 6271
jax/_src/scipy/stats/multinomial.py sha256=CVgpOPCVcr_wTNuF1vXcUjgsENe2mVO7gYBYTSdCOwI 2469
jax/_src/scipy/stats/multivariate_normal.py sha256=Dei7nhj3j5gB1lBi9q_ZpdMHi1UY7XCr3GTzNuaWF1E 3433
jax/_src/scipy/stats/nbinom.py sha256=kr6FLGYROqMMtfLfWLI1fumHBl_tSX1_Ys2icp59Hdw 2603
jax/_src/scipy/stats/norm.py sha256=pUS-JhrjrFtdbfO178IeOJizZBU_WjXU6aUr8W67tQg 8401
jax/_src/scipy/stats/pareto.py sha256=WcRAchdj198RF_W3egR8Bteok1dXE0PHv9ygw-pQYtM 2596
jax/_src/scipy/stats/poisson.py sha256=Y43e2hY6L89go0Vc1HJCDrf61-vDZcQVIC6zVJdaYyQ 3348
jax/_src/scipy/stats/t.py sha256=I32sQ9TVaes4keK5AIK6I-vX--nvl4qtZctuc0zIhG0 3221
jax/_src/scipy/stats/truncnorm.py sha256=V4_Y2OEMYPepPyWVqMZKO0hRV8HwwnI6_djK9xYwdDA 9239
jax/_src/scipy/stats/uniform.py sha256=pR3NYr6ZHTs6xhw-LECsRJqp5l_iBbsw689QHxr9Puc 4361
jax/_src/scipy/stats/vonmises.py sha256=yxvmgBI_HWQn8U3u7wtLv9PjqOk7hRX8I4gWKfqRZKg 2451
jax/_src/scipy/stats/wrapcauchy.py sha256=SpHIjzS9j3TqgAMTHXxtDz73aBfezqYUaNSsGYCky5I 2371
jax/_src/state/__init__.py sha256=TU0qQETOXKzVAd1oVaIEtOJTC2OMj1PdFhsExMHFx2g 988
jax/_src/state/discharge.py sha256=VpQcyNOSch6sQ2QpoNygcYfDDjmrYw4y4ycjlOoAzLQ 44838
jax/_src/state/indexing.py sha256=p8oEIfj8oIv1M3VhQBoQI2gkG4AJ19MiPayqlDZC6xw 9264
jax/_src/state/primitives.py sha256=iofaTUrRSb3sWyJcJSGqpBjXkDOtBA8s43lpOd0BWXI 25203
jax/_src/state/types.py sha256=0Zro4YUM4K7_cy_t4hSuHOHYBaIP2OEDkLK-goBDv70 12462
jax/_src/state/utils.py sha256=7hHiQUpzt9uEyu3yh-l3PkzWV0-_65JtTcg0WuAkIyw 4078
jax/_src/third_party/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/_src/third_party/scipy/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/_src/third_party/scipy/betaln.py sha256=TT77MW5HcPt-C6m-jHT43mGPLuAKO9ANcoEsXjl8JDE 2373
jax/_src/third_party/scipy/interpolate.py sha256=5C9-SwIUrwlpBUgAKYskaSLgTbOtpwESJRPv-6cjj0Y 6307
jax/_src/third_party/scipy/linalg.py sha256=SzmpCoi4kSnLTflDsRa5QdoBcijKHuwmtN9xac2TWlw 3786
jax/_src/third_party/scipy/signal_helper.py sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU 3495
jax/_src/third_party/scipy/special.py sha256=ojE5A5mClaexr_qca69NYpBTgvBEdvVaVMRmptVh0yE 9155
jax/example_libraries/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/example_libraries/optimizers.py sha256=T2D3qDPWZ50kwTyDe2IXPPLbOUu04cQTdf51oHdXVvU 20426
jax/example_libraries/stax.py sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8 13862
jax/experimental/__init__.py sha256=OQS_hX2SykwFBXHl4t0VHv5WKOsqpLjuVL5wXg0CZik 1012
jax/experimental/attrs.py sha256=COmuC5-6hKH5XG9QGyzVzBeSAkQvsUD5qh6zxFMF7Xw 9633
jax/experimental/checkify.py sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0 1213
jax/experimental/compute_on.py sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE 644
jax/experimental/custom_partitioning.py sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g 1054
jax/experimental/host_callback.py sha256=cuZhGDILVkhFgEFjgwSRH4C__ySPwxeJjxvPVNXf-b8 1207
jax/experimental/jet.py sha256=EwfrJ_W-QCjhUItWt3aR6fWlOk7BdLYCFtIKKdPjZ8s 27079
jax/experimental/layout.py sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY 675
jax/experimental/mesh_utils.py sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY 930
jax/experimental/multihost_utils.py sha256=miTiKIBU_DwUZJQHNZyNy1yf4uPJs3r5cay-b1hQ_Jc 18438
jax/experimental/ode.py sha256=zvQAZHbxgGHA9fAhhq7KVxGefqO9-E0kB726eXc18Pc 10646
jax/experimental/pjit.py sha256=2Qi4A4tKn1p6pjxysiCY5UAsfQq_KVZ7QpMX3gRJw30 749
jax/experimental/profiler.py sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus 1424
jax/experimental/rnn.py sha256=cYiMNj6ZANhtjnpdGWsKz4pnKXlC42-JaDQzwmpQcjw 20268
jax/experimental/serialize_executable.py sha256=DYfargD1RSL8Uc65iFWOjm1MU1WArjt4Q4t5AZTTz_0 3006
jax/experimental/shard_alike.py sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M 646
jax/experimental/shard_map.py sha256=CLORcCWvR6vJg_DY69mT_ER0KaHw_rLPG0hSpPxpXtk 93695
jax/experimental/topologies.py sha256=Lxxb2W6nT6AEhM2ekkNUKm6W1tgh5i-_49454NuxCkQ 2101
jax/experimental/x64_context.py sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30 1710
jax/experimental/xla_metadata.py sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk 658
jax/experimental/array_serialization/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/array_serialization/serialization.py sha256=Fv3JGT287UWmx9YD9qGJBsZL9yAcamDcbiBRmOI2m-0 26066
jax/experimental/array_serialization/serialization_test.py sha256=t9v6BZXYAQTlpHJGf0Vi-apf-5wiklWuj-yh1AlRoS8 24295
jax/experimental/colocated_python/__init__.py sha256=bpuYP80GTKaGnrDoSTwjYRb_Faoi2Kz7E0il-40CVvw 926
jax/experimental/colocated_python/api.py sha256=AnaV-mxxfaxPiG9_pst54n3QCV4rFgu6bYmKNtloFLg 2018
jax/experimental/colocated_python/func.py sha256=OusSq72YbwE3wpP9BDOXL3LOhA81g1JDfAcnoWbpj_0 17447
jax/experimental/colocated_python/func_backend.py sha256=lXN-FRbWIWezsICyIc3QYwjpF9-tYgZV1Ut7TBN_dpE 1373
jax/experimental/colocated_python/serialization.py sha256=w8pKcncrWWKZtezgWhVb7BzLuSeUAm67b-8PPqbBU0s 8635
jax/experimental/compilation_cache/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/compilation_cache/compilation_cache.py sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As 820
jax/experimental/jax2tf/__init__.py sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw 944
jax/experimental/jax2tf/call_tf.py sha256=5_UtMy3cx45NHQ2OIVXRZTJCaFKStDZoYGA1UfPcC0Y 28243
jax/experimental/jax2tf/impl_no_xla.py sha256=7ztEeEhJycNm_yA263efIAa-FsuYeOI2cvvgrqmmgNY 54478
jax/experimental/jax2tf/jax2tf.py sha256=db3WL0UmyH_9gnXa5hFC0Y18pcoeEl-rzbvoMDExQ7A 144593
jax/experimental/jax2tf/examples/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/jax2tf/examples/keras_reuse_main.py sha256=AN1R_wzL4NyCgMYEb6ab8fos2vruCL0PGj_GYAEWNEs 2936
jax/experimental/jax2tf/examples/keras_reuse_main_test.py sha256=0nnk1mDlssNRhOIG4u1_WSx6KXF9C7F5Eih6f5hFWjw 1670
jax/experimental/jax2tf/examples/mnist_lib.py sha256=c-Bno2RVxfY4aVgMRK9sjrCInqWVpTIBIQH3C9nPyIQ 11512
jax/experimental/jax2tf/examples/saved_model_lib.py sha256=czppig2onwKslN38JDp6qx3z7iHzFqL_gAcIVmfrAN8 7031
jax/experimental/jax2tf/examples/saved_model_main.py sha256=wuf8oOJ5QNr3R71ltZyAci6o7oKKMpvyn_l96ISMm6U 8003
jax/experimental/jax2tf/examples/saved_model_main_test.py sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g 2467
jax/experimental/jax2tf/examples/serving/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/jax2tf/examples/serving/model_server_request.py sha256=De9TWLSyC5EbkQRg00S9bvVOo17HJuS_M4cztRFjOa4 4912
jax/experimental/jax2tf/tests/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/experimental/jax2tf/tests/back_compat_tf_test.py sha256=xOP6CFs9sD3yKiI2vkk0W4rPKdB5T5C_4IA1E5-H09g 5360
jax/experimental/jax2tf/tests/call_tf_test.py sha256=HxoQ9V7DL4LiI6aUeXBHrwW8x2zfN7ZUmACbfwSzWRE 66938
jax/experimental/jax2tf/tests/control_flow_ops_test.py sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg 10108
jax/experimental/jax2tf/tests/converters.py sha256=uYSiDOPUNaxcgVwLExAgquiK6M0VFH7j55OnbruOyaM 1998
jax/experimental/jax2tf/tests/cross_compilation_check.py sha256=holjJ2qso1R1VD_QSmKj-HnSoh_dkJgkOVLVqZqPIcA 7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py sha256=FWkCsZeRHZq93XP2No2CtiO4uLq24AcLbh6kLns1Frw 60812
jax/experimental/jax2tf/tests/jax2tf_test.py sha256=Qo_6eyZuKlpn2L6s8gWKFUD7ml8hOglt_tNproLt5jY 67012
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py sha256=2ouDpIkKBvLLomEsmYU8jMKIm4iy4xKjONdlEG1Fy-s 6645
jax/experimental/jax2tf/tests/model_harness.py sha256=zhPlq2VnostLmvDfkPdE74A6UPec7wLRS6c_0oFwaLA 13934
jax/experimental/jax2tf/tests/models_test_main.py sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac 8928
jax/experimental/jax2tf/tests/primitives_test.py sha256=58KkYZk0Gay81FNOxySKcF8JsLBnspJT0nDh5eCNu8E 14539
jax/experimental/jax2tf/tests/savedmodel_test.py sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0 13067
jax/experimental/jax2tf/tests/shape_poly_test.py sha256=mt-L7CsMtEA0b0J6S_n9ZYDjsbxHj426gPJwP4XaPVk 126322
jax/experimental/jax2tf/tests/sharding_test.py sha256=6BzliayHqYimzNPsJOJJvHk7Fjmvu_E1BmMziI0qyos 25810
jax/experimental/jax2tf/tests/tf_test_util.py sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg 19962
jax/experimental/key_reuse/__init__.py sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ 1688
jax/experimental/key_reuse/_core.py sha256=NVFNMCQZg0UOZhHD0RpiDqlXum6A48s2z79ETOP9rb0 22781
jax/experimental/mosaic/__init__.py sha256=-MnUIUUmkSaNkO5bOpPdXrcTnKVWt9ModYX4cnlq9LY 883
jax/experimental/mosaic/dialects.py sha256=eYxSlw_cu_8usVXcEA3f-cqmlxDUwcTiqLiqNAgWWfs 769
jax/experimental/mosaic/gpu/__init__.py sha256=CF31GHoctzRGoU0O49900tcu7xIU5clvp12pVj_34DA 3143
jax/experimental/mosaic/gpu/core.py sha256=YU3apA1Ru4gU-5FYm16dsioYzxDvCE7idIR0THdsWxM 23217
jax/experimental/mosaic/gpu/dialect_lowering.py sha256=BVhXSblhFlgqF26nHf7T4g51RrGsEeChywsCMeRg6X8 9917
jax/experimental/mosaic/gpu/fragmented_array.py sha256=OO8AkceUjaRt6HPXAwwzj00LyFIPGWDwJvwsK9o-fHM 77399
jax/experimental/mosaic/gpu/launch_context.py sha256=JjIy-aB4k0EMt5aob9Y0wlWV2_O4MqoNEOrNIn0Sf5c 21564
jax/experimental/mosaic/gpu/layout_inference.py sha256=eEKwSQR9H_QqVmeQWEk2vMk3nQhn65-pGTBpj-_VhEg 12698
jax/experimental/mosaic/gpu/layouts.py sha256=Hp4oerxY_X_IGTVAQ87B7EgtpBp25K9QnFqVnsVukk8 4373
jax/experimental/mosaic/gpu/profiler.py sha256=BklKwpTilcghIvKnImA0qwHjMsfO02CoK-ZYJW5KqNM 14495
jax/experimental/mosaic/gpu/utils.py sha256=F3FvlofnQEMMB5kBMZULobkeDQnvxEkQ4QuSPkvDPec 38164
jax/experimental/mosaic/gpu/wgmma.py sha256=kDsIBYlSViDyP8TsNbKP3vYoPXq4Px1S081c-OJtciU 16152
jax/experimental/pallas/__init__.py sha256=QUJ_nWiTxkpUrksKeazlFnwjpTjRo3iL8fnnUi2sYWg 3471
jax/experimental/pallas/gpu.py sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I 865
jax/experimental/pallas/mosaic_gpu.py sha256=FbA96Z4GmD06pcAjhDiGxiow5AFGObNFEf9eFiyNYCw 2823
jax/experimental/pallas/tpu.py sha256=7OnDrWx_2dvWvcv-7NmjhBt17tDbxO16jSyYdtTqvSs 3957
jax/experimental/pallas/triton.py sha256=8FEMurTT9d-X8tipH-706ETuSLjFkT0LGWiOHjp8zfM 948
jax/experimental/pallas/ops/__init__.py sha256=vC8qlznuwgbYeFDJqP8pDUIf2gEKFrCHg3EZYdTkiwM 764
jax/experimental/pallas/ops/gpu/__init__.py sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU 581
jax/experimental/pallas/ops/gpu/attention.py sha256=u7C3fYnWEVr8Df1L7cHFK8CEFYiHoV3mdi_Q2_e6fz0 19616
jax/experimental/pallas/ops/gpu/attention_mgpu.py sha256=ZU7GJd1WKtkZHpt_8gyOx3_iynUqOmiDDrIaQGeBUDg 12165
jax/experimental/pallas/ops/gpu/decode_attention.py sha256=X_FCW76hLD98IY5QcwlgT1heF7SWe3Bc65I5vFmE2h8 16482
jax/experimental/pallas/ops/gpu/layer_norm.py sha256=OAdBY-Ijtai5IOUfXafJAtudlOUrli0g-_Qhc-t9jQU 11142
jax/experimental/pallas/ops/gpu/rms_norm.py sha256=hNREA4EYeqPbF3RAijhEvtFk4jP8IiNy5LhPGcQldwU 10033
jax/experimental/pallas/ops/gpu/softmax.py sha256=ydajdpaW_oAOwwdGJUx-t2TmIx9QfHhnXX_lNdtmVL4 2774
jax/experimental/pallas/ops/tpu/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/experimental/pallas/ops/tpu/all_gather.py sha256=oLZefmIUQ1Ag_C3kz5h114E2l4OiW9-Nn6DkpB4AC8I 5581
jax/experimental/pallas/ops/tpu/example_kernel.py sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM 802
jax/experimental/pallas/ops/tpu/flash_attention.py sha256=2g9jECYvE3S-HnVPmABhsQgFsiw6JU3wacYhYIGFAWs 50787
jax/experimental/pallas/ops/tpu/matmul.py sha256=jfpCbL3yQ37So4yOtb9pF6K7PtA0WmV65yrzQ8yp4cs 2726
jax/experimental/pallas/ops/tpu/megablox/__init__.py sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w 650
jax/experimental/pallas/ops/tpu/megablox/common.py sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38 1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py sha256=vNNgQIirDK7pyDViP2foec7GC8EhJuxNjR1mp3kHwjU 27249
jax/experimental/pallas/ops/tpu/megablox/ops.py sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw 2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY 700
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py sha256=f6z0ukhHjObUOYoWsLc8Wxvhv8g3UVpjGxUixdnf_UE 21960
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss 2556
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs 2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py sha256=Ra3ZkXny78WrKMPvorsP6-BdjOOZcIj6Htm1DTgWOlk 73759
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py sha256=o-9hXrMgy6vq7nq8SiUFPceTGUdvvAsXNHR1bnjPa5A 14554
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py sha256=OJLEky-Sfna6TGiBjLd1NLhzVJ1Q6AVp1XpQvsRinxM 32846
jax/experimental/roofline/__init__.py sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw 1260
jax/experimental/roofline/roofline.py sha256=hobbkBeBhgGGzSDPWiR-X6f_7V_amtlPd0jl65k9S8k 9964
jax/experimental/roofline/rooflines.py sha256=_1MGFjZaRvcZlN-E_C6nLFVfq6oEPN4slTQ1Dm-K8DQ 8169
jax/experimental/sparse/__init__.py sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs 11186
jax/experimental/sparse/_base.py sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g 3239
jax/experimental/sparse/_lowerings.py sha256=Wgpt5l3eCB6Vv8PKOrNUh6A_7i4JJ4xyhZxeI7zMvgk 7666
jax/experimental/sparse/ad.py sha256=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE 7784
jax/experimental/sparse/api.py sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g 6596
jax/experimental/sparse/bcoo.py sha256=rp4XUNXy_QNlTr3FvF6OP7TniuDDXk7gHFii4zs-7LY 129026
jax/experimental/sparse/bcsr.py sha256=4mT2jU1EKlftB2x44kzLI8Wz1MUQ5d5qVQSsoFxKQoA 38298
jax/experimental/sparse/coo.py sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg 24143
jax/experimental/sparse/csr.py sha256=bBNOl1KR8WwAW0MtiBNKfebV3W2qTW58mA7lsH6Kotg 24396
jax/experimental/sparse/linalg.py sha256=MOtslQMaEgK7Ky7kktDOCjltE8J5vYtbtBQAMIAUQnA 23442
jax/experimental/sparse/nm.py sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I 8568
jax/experimental/sparse/random.py sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA 3882
jax/experimental/sparse/test_util.py sha256=4JlLtN_KEl_bzF-ts9KRSbMNbzUsjEB_6IyQAhw7WQA 9428
jax/experimental/sparse/transform.py sha256=DAdD2G_2mI8yNtWgf-af8CEj2f8McKVFVjh0UN1ky-A 36917
jax/experimental/sparse/util.py sha256=li_VLh2WCr728hXZ18xVRDpL3qo6tpen-LR7CvruB9I 4332
jax/extend/__init__.py sha256=cL0QjoXSmMpM_j6_QBlKN6dtl30zMbvHqYA3eoHsd4g 1630
jax/extend/backend.py sha256=ODjEVf5_5U37PZk5WVp8wI8thqOZO8XAdRV6Rortr8E 1054
jax/extend/ffi.py sha256=31OhFUi5t8Msov6VnqpTj8U8bpDe9V746tMb1enGh9w 1783
jax/extend/ifrt_programs.py sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc 801
jax/extend/linear_util.py sha256=Ie3ii7eVvtHFWsbtlOVu3o_T4jzP_JiXcKDnl-QKm6E 1080
jax/extend/random.py sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ 1083
jax/extend/source_info_util.py sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU 1215
jax/extend/core/__init__.py sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM 1036
jax/extend/core/primitives.py sha256=Ja7N3kkrO4Qf1HOj5_KSLy9GUfgQfmpthYNEvdaKGGA 6422
jax/extend/mlir/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/extend/mlir/ir.py sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4 631
jax/extend/mlir/passmanager.py sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI 640
jax/extend/mlir/dialects/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/extend/mlir/dialects/arith.py sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E 643
jax/extend/mlir/dialects/builtin.py sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ 645
jax/extend/mlir/dialects/chlo.py sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A 642
jax/extend/mlir/dialects/func.py sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk 642
jax/extend/mlir/dialects/math.py sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ 642
jax/extend/mlir/dialects/memref.py sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w 644
jax/extend/mlir/dialects/scf.py sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI 641
jax/extend/mlir/dialects/sdy.py sha256=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc 746
jax/extend/mlir/dialects/sparse_tensor.py sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE 651
jax/extend/mlir/dialects/stablehlo.py sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE 647
jax/extend/mlir/dialects/vector.py sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI 644
jax/image/__init__.py sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI 1027
jax/interpreters/__init__.py sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko 690
jax/interpreters/ad.py sha256=p0gJFj9VvrsZbZavI-xg00Y-PzEu5Rw_6d3upq5u92Y 2759
jax/interpreters/batching.py sha256=QqKOP4uX4F1PSf96ZrXi6x3DOIrU8wyWqXGTlt1nC2g 2574
jax/interpreters/mlir.py sha256=9Kg6EyCFqLqVIxvAo6euAo_o3UYVmZxJcfWPgcFSrBk 2924
jax/interpreters/partial_eval.py sha256=cbb61jx9P0L1g2pWfn3QCggaWgv-8_Lhnrkoim9pNCI 4083
jax/interpreters/pxla.py sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE 1801
jax/interpreters/xla.py sha256=4s2i2F0NnxUevbTeW3BHHM7Nz7yHV7Wkjx9lXUGEIcA 2038
jax/lax/__init__.py sha256=cejK9TONReoAkN_8xvpnrdAOzrksZ8ianj3ifZkU3bk 10840
jax/lax/linalg.py sha256=xvbvXnAOPZ_z97hjoaYzqlyM9verSaMqaL5ZrEsulK0 1397
jax/lib/__init__.py sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M 774
jax/lib/xla_bridge.py sha256=SrvAZ1XaonXyv1MmJv6JT4_eg9q5aa8Gk8Gj2qwH24w 1557
jax/lib/xla_client.py sha256=jvsBGY7rUMEOSsv10CS8Ehh4NX7zqS4A1ROOUyRXULc 3893
jax/lib/xla_extension.py sha256=xuSR42eY6codcchsSZx8rJZ2MdokxbJDNZwW9Xldz6o 1942
jax/nn/__init__.py sha256=4LyqLzosWMtCv_zCd-9Gt9XmfCohbduCQrN_cRPqvVg 1562
jax/nn/initializers.py sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g 1469
jax/numpy/__init__.py sha256=I2cC5hf3hBIzn8sR1l8Qm8dl-y0qRe-4-ZxvgUsZ_kc 12420
jax/numpy/__init__.pyi sha256=7CzHCH8D-J8PQSYHTBfumIoTPHoSvg5MfWGt6SZKql0 43666
jax/numpy/fft.py sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM 1084
jax/numpy/linalg.py sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI 1419
jax/ops/__init__.py sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc 870
jax/scipy/__init__.py sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58 1474
jax/scipy/fft.py sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE 809
jax/scipy/integrate.py sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y 777
jax/scipy/linalg.py sha256=jLoDaaxGZTcRu94dbVgk7qa4Nm-CHu_L8rMsvEIwzWI 1350
jax/scipy/ndimage.py sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU 788
jax/scipy/signal.py sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710 975
jax/scipy/special.py sha256=u7tHwfOThzXNgH7N0wSI_ApWNhifGfa9i4BukWG4Wq8 2386
jax/scipy/cluster/__init__.py sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E 768
jax/scipy/cluster/vq.py sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0 776
jax/scipy/interpolate/__init__.py sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM 760
jax/scipy/optimize/__init__.py sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s 822
jax/scipy/sparse/__init__.py sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo 757
jax/scipy/sparse/linalg.py sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk 810
jax/scipy/spatial/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/scipy/spatial/transform.py sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o 802
jax/scipy/stats/__init__.py sha256=ssiwAm2AdZ81MPMM11015yMNogj4DGsNfWfq507yhlo 1995
jax/scipy/stats/bernoulli.py sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg 819
jax/scipy/stats/beta.py sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs 851
jax/scipy/stats/betabinom.py sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0 792
jax/scipy/stats/binom.py sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs 659
jax/scipy/stats/cauchy.py sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg 881
jax/scipy/stats/chi2.py sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo 851
jax/scipy/stats/dirichlet.py sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE 792
jax/scipy/stats/expon.py sha256=D4a9KT9p2u0vRjpBS6sQgZ95vuzyZLCh-Mon8HiNq4U 788
jax/scipy/stats/gamma.py sha256=TvP4BE8MROMxQL6YNDQIHSaZCfxumHkjGbCXw9O2iUo 852
jax/scipy/stats/gennorm.py sha256=hqwz7ueAgdtzs1v0LHTTro2PLKRDjWxTFuu9jXwXOyI 804
jax/scipy/stats/geom.py sha256=pdp2tCdXOa3CUWyKJCyrFrMR7TBX0La5tVI4ALAETI4 787
jax/scipy/stats/laplace.py sha256=7Lap3E3_neHYpG5mUFlYxZnwAT1J8mTrNPnn8BSEGrA 804
jax/scipy/stats/logistic.py sha256=IoHe_TMB0CH5EMAXjmqRMP8LjR4lTS_x9Zf4K7K5e2g 845
jax/scipy/stats/multinomial.py sha256=hfRYQL_GeszF4sJXx5Y416JgXoXVw5vxtQIt1NmkCDU 794
jax/scipy/stats/multivariate_normal.py sha256=YYqtLLZmjsLWeCIoDpxUk-rHde-g9unfo6ofg44nSBQ 802
jax/scipy/stats/nbinom.py sha256=hWMMt7D_rLRtiGjlpHcbOXJF03BfA3v9Cu972twYKCY 660
jax/scipy/stats/norm.py sha256=qNMz53OSiWOZGt6TZkJZvWAFocdUMUu-DsysM9kg2K4 879
jax/scipy/stats/pareto.py sha256=xOqP9JMLAYdVXMFpCfCoRqxJNXVeWfJXkEqpOsrUjHc 789
jax/scipy/stats/poisson.py sha256=Guu0WOaGAwtujTTWVlBZiBg2fLu4JOjx0fumX4m4Ems 804
jax/scipy/stats/t.py sha256=kPmOWYVodYyD6cTxzX98qNxiFeWpDHBM1WwrdK5Ztoc 784
jax/scipy/stats/truncnorm.py sha256=q5aSLcidwhZqFi9sNjqvFF9RxT_YLcYBMW2DofDoE5M 855
jax/scipy/stats/uniform.py sha256=KCmUJttnFPbr-_ECG7O5xd-fy8CqvysIhepltRj_6SU 818
jax/scipy/stats/vonmises.py sha256=NYGQbIKES93vjVumCkF6IihydEhrhH7JMaJFmGDfCwM 791
jax/scipy/stats/wrapcauchy.py sha256=pcs6O0cdJTyhsN6LGTOxBm4TyuKkm3HRIdxfZVoz8r4 793
jax/tools/__init__.py sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ 581
jax/tools/build_utils.py sha256=nu6hxCk0o1Z06Tydm8msAqSmoZnjMOq6kY58qpCfHwk 4393
jax/tools/colab_tpu.py sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo 890
jax/tools/jax_to_ir.py sha256=Z8LkCMaXfqjYFSHPSTS3L5imebmYtV3NPLy7V1WMLPU 8776
jax/tools/pgo_nsys_converter.py sha256=BCxXmHnvcorM62adEbOFNYcsnTyv-Hkqnf4K7lo_wt4 3413
jax-0.5.0.dist-info/AUTHORS sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY 313
jax-0.5.0.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax-0.5.0.dist-info/METADATA sha256=u6AWUBoJyy-b3FKlQG9C2n02PrW474y-EuG5S1kRO64 22927
jax-0.5.0.dist-info/WHEEL sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8 91
jax-0.5.0.dist-info/top_level.txt sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE 4
jax-0.5.0.dist-info/RECORD

top_level.txt

jax