Wheelodex » Projects » jax

jax

View on PyPI — Reverse Dependencies (824)

0.4.26 jax-0.4.26-py3-none-any.whl

Wheel Details

Project: jax
Version: 0.4.26
Filename: jax-0.4.26-py3-none-any.whl
Download: [link]
Size: 1856888
MD5: d8e85f1da9c9d8204e9f28cafdd1b7c7
SHA256: 50dc795148ee6b0735b48b477e5abc556aa3a4c7af5d6940dad08024a908b02f
Uploaded: 2024-04-03 21:45:34 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: jax
Version: 0.4.26
Summary: Differentiate, compile, and transform Numpy code.
Author: JAX team
Author-Email: jax-dev[at]google.com
Home-Page: https://github.com/google/jax
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Requires-Dist: ml-dtypes (>=0.2.0)
Requires-Dist: numpy (>=1.22)
Requires-Dist: opt-einsum
Requires-Dist: scipy (>=1.9)
Requires-Dist: importlib-metadata (>=4.6); python_version < "3.10"
Requires-Dist: numpy (>=1.23.2); python_version >= "3.11"
Requires-Dist: numpy (>=1.26.0); python_version >= "3.12"
Requires-Dist: scipy (>=1.11.1); python_version >= "3.12"
Requires-Dist: protobuf (<4,>=3.13); extra == "australis"
Requires-Dist: jaxlib (==0.4.25); extra == "ci"
Requires-Dist: jaxlib (==0.4.26); extra == "cpu"
Requires-Dist: jaxlib (==0.4.26+cuda12.cudnn89); extra == "cuda"
Requires-Dist: jaxlib (==0.4.26); extra == "cuda12"
Requires-Dist: jax-cuda12-plugin (==0.4.26); extra == "cuda12"
Requires-Dist: nvidia-cublas-cu12 (>=12.1.3.1); extra == "cuda12"
Requires-Dist: nvidia-cuda-cupti-cu12 (>=12.1.105); extra == "cuda12"
Requires-Dist: nvidia-cuda-nvcc-cu12 (>=12.1.105); extra == "cuda12"
Requires-Dist: nvidia-cuda-runtime-cu12 (>=12.1.105); extra == "cuda12"
Requires-Dist: nvidia-cudnn-cu12 (<9.0,>=8.9.2.26); extra == "cuda12"
Requires-Dist: nvidia-cufft-cu12 (>=11.0.2.54); extra == "cuda12"
Requires-Dist: nvidia-cusolver-cu12 (>=11.4.5.107); extra == "cuda12"
Requires-Dist: nvidia-cusparse-cu12 (>=12.1.0.106); extra == "cuda12"
Requires-Dist: nvidia-nccl-cu12 (>=2.18.1); extra == "cuda12"
Requires-Dist: nvidia-nvjitlink-cu12 (>=12.1.105); extra == "cuda12"
Requires-Dist: jaxlib (==0.4.26+cuda12.cudnn89); extra == "cuda12-cudnn89"
Requires-Dist: jaxlib (==0.4.26+cuda12.cudnn89); extra == "cuda12-local"
Requires-Dist: jaxlib (==0.4.26+cuda12.cudnn89); extra == "cuda12-pip"
Requires-Dist: nvidia-cublas-cu12 (>=12.1.3.1); extra == "cuda12-pip"
Requires-Dist: nvidia-cuda-cupti-cu12 (>=12.1.105); extra == "cuda12-pip"
Requires-Dist: nvidia-cuda-nvcc-cu12 (>=12.1.105); extra == "cuda12-pip"
Requires-Dist: nvidia-cuda-runtime-cu12 (>=12.1.105); extra == "cuda12-pip"
Requires-Dist: nvidia-cudnn-cu12 (<9.0,>=8.9.2.26); extra == "cuda12-pip"
Requires-Dist: nvidia-cufft-cu12 (>=11.0.2.54); extra == "cuda12-pip"
Requires-Dist: nvidia-cusolver-cu12 (>=11.4.5.107); extra == "cuda12-pip"
Requires-Dist: nvidia-cusparse-cu12 (>=12.1.0.106); extra == "cuda12-pip"
Requires-Dist: nvidia-nccl-cu12 (>=2.18.1); extra == "cuda12-pip"
Requires-Dist: nvidia-nvjitlink-cu12 (>=12.1.105); extra == "cuda12-pip"
Requires-Dist: jaxlib (==0.4.20); extra == "minimum-jaxlib"
Requires-Dist: jaxlib (==0.4.26); extra == "tpu"
Requires-Dist: libtpu-nightly (==0.1.dev20240403); extra == "tpu"
Requires-Dist: requests; extra == "tpu"
Provides-Extra: australis
Provides-Extra: ci
Provides-Extra: cpu
Provides-Extra: cuda
Provides-Extra: cuda12
Provides-Extra: cuda12_cudnn89
Provides-Extra: cuda12_local
Provides-Extra: cuda12_pip
Provides-Extra: minimum-jaxlib
Provides-Extra: tpu
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: AUTHORS
[Description omitted; length: 20262 characters]

WHEEL

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

RECORD

Path Digest Size
jax/__init__.py sha256=dQlccp1SwjK0S0m-3JyzU4WDe5FSwJaW8GlRS2AKHJU 10107
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=xx0PZUvoNtaxkxXQImoymJlY6I7J1oSbnkgCigp1nMg 1040
jax/core.py sha256=POA3yaxLJnbb_Oj699lSD6epH0S20VO3tSXFuqQZkf4 7340
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=zIyWVOrlGmSSVF5-50dOxgZLfgPeZU7VNeQx6i6wFpc 1272
jax/flatten_util.py sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY 645
jax/monitoring.py sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs 1191
jax/profiler.py sha256=tMxGkMNYoug4DQGLE-OKx0z_nhlR-gBSs6vFpXd3jzY 1119
jax/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/random.py sha256=DLfiGlzcPAOi-DCoo8wIqn39c7ZyEyXmBNgMykTs7ZU 8990
jax/sharding.py sha256=UgnvqjbgLSbXzUHLJUzA1d9KgFba-1Nzq1sMAecUB94 1184
jax/stages.py sha256=fxDMZsTC4Ehzedq5af92xB0RMrU0gfLlLCKE4MWioOg 1272
jax/test_util.py sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU 835
jax/tree.py sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA 956
jax/tree_util.py sha256=z9oGFs_ZJOhnsc9_ljiQ_rbkkwVTooz6XOxUunAAq80 2971
jax/typing.py sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk 3331
jax/util.py sha256=hG_kZcqiOEO2HjTehzSyEFVPctNZneOIORqfC9hwzyA 1091
jax/version.py sha256=3fDfFKASNW5eJ-EDpGV3YntNwKg9FETdthdnchVrs2s 6132
jax/_src/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/abstract_arrays.py sha256=r5HI4QdiNeLvvysAx4HoMR7MqrEzEbypFIzc8549OKQ 2680
jax/_src/ad_checkpoint.py sha256=FFRmP7llpRq_KBKMGhr26gsw_gUFJ9iuEvQ41T7RCSM 36532
jax/_src/ad_util.py sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY 3773
jax/_src/api.py sha256=HMfaxRovlmBVJftzTDAS89Yub5aj0UVnJl6-uhzJg5s 130113
jax/_src/api_util.py sha256=y9Hk8Lto3mameVddN7iz2HdFKwYqpqIXKVcWPSTO2Pw 28383
jax/_src/array.py sha256=muWlC9ygk-uFe_lrByHf22o6HetmPqC9Jz6pqxz1_vE 37113
jax/_src/basearray.py sha256=8le5M8SR-a15Z2ewqJuPxiJN483G5kCz0v70JWius1A 4296
jax/_src/basearray.pyi sha256=2gzeRPRuIcLJuSQZL3zOBVY1izUIkPonPhEjAma6RiI 10732
jax/_src/cache_key.py sha256=tdlwLLhNmNl9VOeZfGAJrX6eOgHbEDMMPvfLnk_uu88 10498
jax/_src/callback.py sha256=PI9maf7onxINRRhmPI4fxR9jLVSw-Ad0GxLRQPdqHak 18988
jax/_src/checkify.py sha256=GKISKwXgUnAxgMO9UPrqDG80x2r83unDD2nw4tyOq4w 52918
jax/_src/cloud_tpu_init.py sha256=0IY467hAWhiTXeL4PXM49RCrci2298bSurPiQ55paXM 2371
jax/_src/compilation_cache.py sha256=45YR_4oGGUkvFm5-vDt-Vm1dHxAEb6R68_YwfhwsQoE 8842
jax/_src/compilation_cache_interface.py sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c 863
jax/_src/compiler.py sha256=p5bcWfLgMtwD---p8ndwdNtRXrRIkdtPbbgKs4PBhJU 20648
jax/_src/config.py sha256=dtrumUo8XH7iiFDSE7iqCuVpfF9hvYkU5-wLiwiwpwU 61646
jax/_src/core.py sha256=maebiZGHNjyuVpGiyP2voeX63gO3Jcuqxs2bl2Yms6s 118878
jax/_src/custom_api_util.py sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw 876
jax/_src/custom_batching.py sha256=x0zm1OkE1g-bBo6w_W4ul3zGnDx7ouUMTLRLz09-6ss 10502
jax/_src/custom_derivatives.py sha256=NDSZvq1PdL2u4nXp_Wx-jwCvfD_o2QQ14uPvdEtuykg 60484
jax/_src/custom_transpose.py sha256=65gQw3u5hG3vHavcbjx9gvN_visilcXoKzpYEpfzF4E 8310
jax/_src/debugging.py sha256=oKvllSijZ6zbDpjy7d424qWEfoxEsgTIlg4xGlvR6tU 24309
jax/_src/deprecations.py sha256=c2fclD9ZpSKNQHNTndSiV0nR4dsmZS4qG3Tfnaa4JKE 3064
jax/_src/dispatch.py sha256=TxHT8sTirhKfBTyP1nYiNJIjciVp40W9mwSjKZt3Yds 17641
jax/_src/distributed.py sha256=OrDKJreVZYWombPwWWlOO4f6Xm4-w_g2Ttad11t9ha0 8865
jax/_src/dlpack.py sha256=aCfyJPRz6IsWE7Eh2kIVO9NLiyD4l1byoxZPnQXBPXE 5309
jax/_src/dtypes.py sha256=hcLTsa5nCSyAwhA9KHABHQ4aHLTLAtLVAhNHnecfzAU 28551
jax/_src/effects.py sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q 4682
jax/_src/environment_info.py sha256=9Gwd-KeYyMlORpCWQR6a8ruZgnJkom8qT_1RMsjBacw 2010
jax/_src/errors.py sha256=2yIUoXYA4XLYehScnEoerQIu6geOQskNqQV-_6jerG0 24766
jax/_src/flatten_util.py sha256=-9ozNDsKw2QRDf5J4UjIlhTPOv3pGloUMKffS9aGteY 3554
jax/_src/gfile_cache.py sha256=yfHAo5Jl-Knt-P1GWRaaB4Ucxw3t7g66jD-rGti9KwY 1923
jax/_src/hardware_utils.py sha256=s0UuPPWP6eB7oqQ8o02dL3RqzQTYTsMKcNCukGoO5zI 1714
jax/_src/jaxpr_util.py sha256=4mB7jfzh0IhgqiqnStUnL4_msTyOljxgzG56pw1-x3s 6428
jax/_src/lax_reference.py sha256=hZ9Pm3Dx5iJa5yWm0oSvkrAbyuaEartugDi8uIuW454 17330
jax/_src/layout.py sha256=f5y8-wOswaysEWfpc_KKLdgfz1PIH-NlSFP4ZgWl2HE 1231
jax/_src/lazy_loader.py sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo 1333
jax/_src/linear_util.py sha256=9s_-v7r432MAQc6r-juktBEyIu2-tOYh0j9orLJRLs4 13946
jax/_src/logging_config.py sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY 2040
jax/_src/maps.py sha256=2BE9tfqqdOzmOQzwzzrOEvkj_P1AZWnyPZXLfCvKiO8 86657
jax/_src/mesh.py sha256=I2YfKPwdLK9NRunSqs6ImaAJyXS3JGvYtOFunO_bzpk 11314
jax/_src/monitoring.py sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk 3888
jax/_src/op_shardings.py sha256=4FR-A5ldSwoP5oLU0SW_kjmOPr3rc6usnZ56B11wl98 3897
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=jkbmUevVHf2NJBJMoAHIbE8tbrtlKjT9y0gzEx3bMjg 103341
jax/_src/pretty_printer.py sha256=gy89qxIgK7ZG7P9cglSmzrtFFjPS1uXWdcUbUjH1iFo 13077
jax/_src/prng.py sha256=fX5ba58BCyFyuEi32H22EXDby5V_H2Es8ByTivmD748 49241
jax/_src/profiler.py sha256=twgR0_2CzcOobgt8kDAoF_Sf89h5fHNWRXbWYwoZE4I 14525
jax/_src/public_test_util.py sha256=Z4p1XzuocBDSXwI2HBFkssqPrIGpO3R3VETa5gyi2bY 10451
jax/_src/random.py sha256=o9E2RqCsSsgD6pu_E2V39SpVeHxT67N7zuTUY_54bt8 100218
jax/_src/shard_alike.py sha256=sdSYS0jagWfqS8pkdLEqQg6tJBr2QoU92dEgG0Hv4nM 3978
jax/_src/sharding.py sha256=fB6sGov0DDVy1eCAJ3MrwypVzQ7Rho85-0QNXv6kvO0 5070
jax/_src/sharding_impls.py sha256=nxvhLRMPmH28uEE_1UAOJTbauloMb6pe5OU6Gjkzgig 50872
jax/_src/sharding_specs.py sha256=9s9WfSdGriLgmreTp_2xlqQtkf1TmSxCWRmtT-q--1U 9207
jax/_src/source_info_util.py sha256=AjLwjmZXBhYFM13j4kMBWKK0Ch1A9Q07aKwyt3Wcefc 9912
jax/_src/stages.py sha256=asm4WlL5rFZ6Q4s-XYSTI4xElsGgETO1UuXy4FH4aVQ 26931
jax/_src/test_util.py sha256=YbzZlcccPLbfIdCI5N62MIu58x53cFkfUPloqTXLIS8 62652
jax/_src/tpu_custom_call.py sha256=heMk1vjYPSuvy_MbQBMWLGVEDroWHjcZbp6-Mi7-W5k 18012
jax/_src/traceback_util.py sha256=bo0vQtIPZNxSjYeR6TXNTN0Hf4XKyNOOv2eJAMc21KA 9060
jax/_src/tree.py sha256=qf_oRe7McgufUBFD7LtV9KG8XL_fJB-kn7SZDUaSka4 3492
jax/_src/tree_util.py sha256=T9sJ-TWs4HRS7ZpTZYxTf3D3xxkGmv0LkBeEFzxqNlk 41254
jax/_src/typing.py sha256=QbrzhXnpotfG8jeLp1xPEL4nFea33-vD0TXylXDb9XI 2892
jax/_src/util.py sha256=EXgaKRmaRT1MSHfWDNbU5TMM7T9rBFqNfHEaqiNXLLY 19149
jax/_src/xla_bridge.py sha256=V9ItYKtrPwWwE1XyVpsGjyvLO3YdWOFbCXnFju6aP94 42572
jax/_src/clusters/__init__.py sha256=ZIzPN7-Awrf3mDafphxQXBXWwItcp_8axkztAA1sBYc 1095
jax/_src/clusters/cloud_tpu_cluster.py sha256=ILjkSzl0OI8cZyUzwCTAsXDG2LcguWfr1o4B4ZzKyuM 8148
jax/_src/clusters/cluster.py sha256=4zenRWGqAo0CHbhEq_X0lNZDvDm5WsWwNFzHYDmvtPQ 4901
jax/_src/clusters/ompi_cluster.py sha256=EUK4RbIetmu9LL_jzxr1cT3kRtTfgiRI__VcOLGKqmA 2294
jax/_src/clusters/slurm_cluster.py sha256=1AmqbARp7rNWe2v7UI_AD100v3qdg2o0IV66zuCf7M0 2297
jax/_src/cudnn/__init__.py sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU 581
jax/_src/cudnn/fused_attention_stablehlo.py sha256=JpSdIJC2zw-dq3Kte6531W-Esh74VlUz5iEKCFGxKn0 36670
jax/_src/debugger/__init__.py sha256=VPrkm_4GZ-dm4SN7L4xqXCwGSFiD5K4BkXGy0MvWnSw 885
jax/_src/debugger/cli_debugger.py sha256=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew 4780
jax/_src/debugger/colab_debugger.py sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA 7823
jax/_src/debugger/colab_lib.py sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U 4298
jax/_src/debugger/core.py sha256=0rEPhWjcCaiMey4sAuDivG1H5RFqqlfesAxlZZ1rlRU 8369
jax/_src/debugger/web_debugger.py sha256=WJzG5FubQxA9EiuyXzHUb_v2_2GqvovyQiy6EaBomY8 3282
jax/_src/extend/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/extend/random.py sha256=bAtdkmxevVfY9JfcwrWX6Ah8RXMoNelXg2I644ePHCM 1276
jax/_src/image/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
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/deprecation_module.py sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4 851
jax/_src/internal_test_util/export_back_compat_test_util.py sha256=VBDNP08YLWvuUdt1HBVx8LUcBi17jXScwuTQejB7wMk 15299
jax/_src/internal_test_util/lax_test_util.py sha256=LBych5KKJ1fYfxHTk9YyyiOuVGljYYyngz1Ckp__BNY 12148
jax/_src/internal_test_util/test_harnesses.py sha256=ZJBbtnWCQphK67bXX5dCus2XZMURqyAIxR-JoGJ6FtM 124694
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=XGIiZyYNXujjGE4M4T6feZ8yStPKiXaSSBiunyn5sJs 32701
jax/_src/interpreters/batching.py sha256=NJDTJxEy2T1BVJVXzyyyjMWcuua7lmTUou5Og2V6EjI 50430
jax/_src/interpreters/mlir.py sha256=M_naKrJPWaPj9iMOcVswS_kLv_o-_eWU81WdzlO5eqo 117364
jax/_src/interpreters/partial_eval.py sha256=87xtsxD5rbMZNGWnEwnXCn-9JK1tMvQS5il2TJVYG6I 125831
jax/_src/interpreters/pxla.py sha256=DwXz98ziAunCaZQ4-UEDlFl7bfjmBEWvSTaHcLGVJEM 138402
jax/_src/interpreters/xla.py sha256=AFLJHc6vzGVwcnD8Bm7ywuLeYraYJg8gEDXANMd_R2U 12203
jax/_src/lax/__init__.py sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI 690
jax/_src/lax/ann.py sha256=XEBN_-wTrqWPdCIM3m4gtXyznOCjVkrjRkTi6YXLE8M 17623
jax/_src/lax/convolution.py sha256=SNNtSYshLLvMAMwsrTfnDPGO-deZeqR-Y_Nu3rebJeM 44715
jax/_src/lax/eigh.py sha256=kHTPgQkUr3h575dvx06A9h8-vlzN4JbRkeZNwkjgAxo 22205
jax/_src/lax/fft.py sha256=izHxbi-S686onnDHQC3WRuu9PbKhDhYkZUtZPx6LxFI 10151
jax/_src/lax/lax.py sha256=Fkv-NIpJvInniuEguhUvSYCIShhiVw2MzKyAxZ96fXs 210995
jax/_src/lax/linalg.py sha256=pq380CYVFKEUJNLQxJTQYnPXP3qm_dkdRn841ucM0kQ 91611
jax/_src/lax/other.py sha256=wTmrVnNdLQajm04LyGa4ioIiJseZ9dT9bWbQmCSmO-k 10539
jax/_src/lax/parallel.py sha256=gUSrc33otbkZ8jt30h3BtkfFJDWJz5j98TSy7hadp3Q 75667
jax/_src/lax/qdwh.py sha256=f3QJwPMDDO_qC-VFaKrzSXUJ0njCSwSEjICsA5FckxU 8028
jax/_src/lax/slicing.py sha256=glCJwZAhftQCVnXybrH3KhxGogIqFVpVwW0glM771mg 111664
jax/_src/lax/special.py sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY 26940
jax/_src/lax/stack.py sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY 2560
jax/_src/lax/svd.py sha256=vSEwqDksgYifZlIxtqjDodJVul5b_C819162EZZ5zcE 11705
jax/_src/lax/utils.py sha256=KHbNG7zWt4w8vrSxSJv5iKubesvLdZlX9HVTXLyr5Iw 4968
jax/_src/lax/windowed_reductions.py sha256=StHNyfk40hMA9-SxVsI2_yC0w671r5tvQcQbua3A5Q8 40180
jax/_src/lax/control_flow/__init__.py sha256=3lXBsoR3ChIpfQWfWN7muM_aTgtQVENpf6m-nh_Ssvw 2103
jax/_src/lax/control_flow/common.py sha256=jE7vo0resS9tHdg9F9nleNB9Gwcv8X3VUzpMHNo0OxA 10976
jax/_src/lax/control_flow/conditionals.py sha256=x3vSQsquCWjZ4XFydP0kuanzgdN12wH5LdMsTfntr3c 42956
jax/_src/lax/control_flow/for_loop.py sha256=3-Cfq-xoXWruQDDtzH8vDgvtYNZvDCntU0fY3BH1ZRc 37122
jax/_src/lax/control_flow/loops.py sha256=mb0q02qk0otlJ3p9vme4_oiWM38JwgpgttXJtsq4m8I 109671
jax/_src/lax/control_flow/solves.py sha256=MtOEjg9pa0NTCfVkPaKlfEobIViNhOvKfvyv0CsR1g4 18874
jax/_src/lib/__init__.py sha256=HSyqGdFIAIFHVBk45EIzsqK-M5HHqza5-5ubmXqmipY 5476
jax/_src/lib/triton.py sha256=HbqoIneu52En_DHWH6Mph9rGtj324CRCXoT0SZA3u44 1336
jax/_src/lib/mlir/__init__.py sha256=vv85BXVE6lq8QZTUZtOIFPrwcoIWROkeiAqVa5a8SB0 885
jax/_src/lib/mlir/dialects/__init__.py sha256=tF3FV7JAD9WlLkd8TJz4L0ULoNlCtqMVJc7BZr_KgKs 1195
jax/_src/nn/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/nn/functions.py sha256=_pJUINxbGf0y7K5pjWIalL9Pg11eraNN6CcQfCD6LpI 19224
jax/_src/nn/initializers.py sha256=526DihqSYNUWM4i5EET9rvZl0btVP27vdZs8Tpl-Qu0 25570
jax/_src/numpy/__init__.py sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M 581
jax/_src/numpy/array_methods.py sha256=ISRBbYRa3HcXLjMD5egv5Ie7khsn5224QRBw-E57Gok 32582
jax/_src/numpy/fft.py sha256=rVEbkoRIJ_WiGPnrsl4bztstWiZ50acmrFPvl9pde58 11373
jax/_src/numpy/index_tricks.py sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g 9888
jax/_src/numpy/lax_numpy.py sha256=srvD6i-Tj2tmPqL6yNcmRt6efGbqMJkXHTHum001TVw 219682
jax/_src/numpy/linalg.py sha256=rtiU5yILGJD_xcQw0Kk_xTo-XEW99Wnt0gET4UQGCpg 31450
jax/_src/numpy/polynomial.py sha256=2uEH06Zqya_x15M2ROYEaV68fwmwCmXs-gKZe6mnXzw 12162
jax/_src/numpy/reductions.py sha256=VaDp9mMQzwGLI37b-mc3yiEWSAGtpQ790kEZ49TnGbM 41617
jax/_src/numpy/setops.py sha256=pdYBHaTrUHOhQtNIvH2VXy3kk2Vk39VmpM2BdUWwl0w 15407
jax/_src/numpy/ufunc_api.py sha256=RbGKNxN0MhSzkQPQOYbEeJS7PaaEwMtyDrS4MqbQPzg 15264
jax/_src/numpy/ufuncs.py sha256=A4kN0GkuisphqHQBk8A8nCdewKXGUqZrPyl2pxWcm84 28522
jax/_src/numpy/util.py sha256=55wWiEaqu2SHthyoIB6SYdIWCQeUoFmehrWiP9kYwfQ 19129
jax/_src/numpy/vectorize.py sha256=H-OiIbDQvKmjpOJlO9L7cpQOYn8cOJQod_7C6OHOrB8 13021
jax/_src/ops/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/ops/scatter.py sha256=kZ-h0mJzy57grnDe9PFZdv4TN6u6_OxRauIRtCVZ214 17593
jax/_src/ops/special.py sha256=nEM6g8Ha6-LY9Ai7gZaMAH3LkT6Xj362V17yKVoP8QA 3884
jax/_src/pallas/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/core.py sha256=fJYY46uKMqa__bwaXGOFh0J2VpvT8tVAArDvlQr-m4E 12353
jax/_src/pallas/pallas_call.py sha256=0GTbGeEH8P8QNxCTskXacW5EoR-3snd8QbnuGNlJwcw 25357
jax/_src/pallas/primitives.py sha256=SQ5tbo_O0qyHyzOWSYKlq8UrZfSWL9PjhKp8SisX934 16743
jax/_src/pallas/utils.py sha256=Z4RR0ZkyZWFiILDn48GIS45yui60vtDkiqPhca2JohE 5991
jax/_src/pallas/mosaic/__init__.py sha256=mlQsjEmalHvk5cLlYbm-VCHeEDbKRasWqtDrrttNN0s 2462
jax/_src/pallas/mosaic/core.py sha256=pIsvseu1Eu8kRKBe1ftlMVMWfvOjh9s-SPJyItYse6A 7643
jax/_src/pallas/mosaic/kernel_regeneration_util.py sha256=e3FjR_zYDDuUixKOk-W59WeDopm6Nh-n_VfLKGiKEm8 1995
jax/_src/pallas/mosaic/lowering.py sha256=ZD4tfVy8IfuVGG-7aOYJ5BktRfgWS8_PCtAEaTtAkNQ 75303
jax/_src/pallas/mosaic/pallas_call_registration.py sha256=Wwio6VdKTx63oLMtn1GD5AH_4SkvGVCqMKxk_Tb8Xd4 4250
jax/_src/pallas/mosaic/pipeline.py sha256=A7WI4ankZcs2wqaxUUDnOaihha5a3MDoliC4ixPXEFk 43393
jax/_src/pallas/mosaic/primitives.py sha256=J6qUdR4DQvQB-9VfCBqxZVoxm5SKIcXL8R5OOtZVPDU 18126
jax/_src/pallas/triton/__init__.py sha256=x4B61dVc6mxT_re9aORVC0QfLvoSo2g1lQfudxuIxmE 1120
jax/_src/pallas/triton/lowering.py sha256=hKuMtK6lptc5A7Ds4ZZ7E7bzFEQP4FIuW7-Skc_hOIg 79836
jax/_src/pallas/triton/pallas_call_registration.py sha256=XRcnTe4ylzz2AfdyNg-JoVDdu3zSsc6HY6dEtP_O-oI 9863
jax/_src/scipy/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/fft.py sha256=G9Zrf4TsbLQH6BZlKwcWFaDXiJcEiuL6eVG59avHXhk 6249
jax/_src/scipy/integrate.py sha256=6I9YyM7lwGwhpPVmDzy6wk2f9jz1iqVQVfiVNiNYj7U 1036
jax/_src/scipy/linalg.py sha256=a8twFXM19x3qZuxco2YU9-2l31_VR4Wr4cqEEf4i3wo 41777
jax/_src/scipy/ndimage.py sha256=viYzxaVZVP3pXG1yBKMSG5wssJUKYDerBxUcXfCT2ds 5362
jax/_src/scipy/signal.py sha256=IuhZdas7Uv-mk9u_WR3bW4aJQn_jYakhr0bwGtvEwBE 27552
jax/_src/scipy/special.py sha256=-Hho_ccefglXuNDKh9OTNfVmY0hl-RuDWFrKQZEoKoE 63694
jax/_src/scipy/cluster/__init__.py sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8 581
jax/_src/scipy/cluster/vq.py sha256=lmudnV7wOY8eN0CUT2a7GxA3mhwRGmjPa7cVF2PNB98 1779
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=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/sparse/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/sparse/linalg.py sha256=VCjQWXeUaf_9xqTCTIiGTVkMjaa9IvEwKgLrhinavEw 27326
jax/_src/scipy/spatial/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/scipy/spatial/transform.py sha256=VGTcZRDye7xJBLFPiIJM1DJdJTDiIPGytz7HFR0o4QU 15167
jax/_src/scipy/stats/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/stats/_core.py sha256=FLIRKSf8SY-lKXwscEFVn12ALW3iQtu4QDWH6HhqQMc 5725
jax/_src/scipy/stats/bernoulli.py sha256=9PntVJ6XZC9btsq51pzvgi1Ndpg7IGgAZzIglEOSyL8 2270
jax/_src/scipy/stats/beta.py sha256=YxIfH_-b5vXkqmOiC5n57J3kZMmkkf_aYEawfwTtHLw 2968
jax/_src/scipy/stats/betabinom.py sha256=vusiYROP2Vk2PNZs0E1BCiwEpFaIlRC1hZ02v8-Ga-E 2005
jax/_src/scipy/stats/binom.py sha256=T4wO99oUze6-nuv5Vw6_crdy1jury62cA8k29RYN9CM 1671
jax/_src/scipy/stats/cauchy.py sha256=ejSOHHOc0MJjsRFTJstpP4l5pTYWFWsRJdczaYlgbHw 3142
jax/_src/scipy/stats/chi2.py sha256=Q-vgl-gttIpxzlU59fSFIR-08fZkl4cT7P1MPK7cKzw 2952
jax/_src/scipy/stats/dirichlet.py sha256=v_KIk7RegcQYeNOD5memO7kLmj043NzmggMPVf73WSw 2148
jax/_src/scipy/stats/expon.py sha256=yVjLkyb0z9tQXtltE0Qk6WHjeQlW7IW0G78ypijU7vM 1326
jax/_src/scipy/stats/gamma.py sha256=gOi_AqGcD-d1nys898_UFEJF8dEaKzN2F6KzcPHYxjs 2567
jax/_src/scipy/stats/gennorm.py sha256=ESjS4pr1gpY-dKLfkBLzoBiuVmqx3xn7dZj5hHqvKCQ 1311
jax/_src/scipy/stats/geom.py sha256=PC6utDsp2J-AAm9ZEt5HOe-C94Eqe6DiNT_HAjtxexE 1394
jax/_src/scipy/stats/kde.py sha256=r-Ua7AKOF755cE9OPFvtodfXWZ0jo5U3DjpUoaKks3g 9876
jax/_src/scipy/stats/laplace.py sha256=uaazCUEB7qdWY48bxV2sP0msvWb9dpAxjkNnhqQWzEU 1803
jax/_src/scipy/stats/logistic.py sha256=3HIoYI-iN2UVGLc_UHRpaNw9a32VfPsoUObjCJWOjS0 2434
jax/_src/scipy/stats/multinomial.py sha256=Vgndccd5iNj1sV-2q32oHOzOeGo7LNLBq52ICRwq0qQ 1687
jax/_src/scipy/stats/multivariate_normal.py sha256=T86VsQGq7585Bte2nK9ATHCWKkkUuLk4aiuDqmw-ssU 2231
jax/_src/scipy/stats/nbinom.py sha256=GTXNvC69AUsqgnnT2ySJcdPe0Eq1bjNdrTF5BPP8rEA 1710
jax/_src/scipy/stats/norm.py sha256=CXkWEgZmVayncIpL2Vi5rLNFVxqlz215qedtEQm0oPI 2899
jax/_src/scipy/stats/pareto.py sha256=ahSAdtFCCf10Idfjsb0nvecNzgVvcCm8X5JxJnRAEOk 1509
jax/_src/scipy/stats/poisson.py sha256=Xikr-qVqCQvD8BM12XQN5QquFgI00gsriu7oCfQnuUM 1700
jax/_src/scipy/stats/t.py sha256=vx7uvNW6xzMTCOHl6423T8ho6E4oMyRNdqIx-ODAUiE 1834
jax/_src/scipy/stats/truncnorm.py sha256=UiA72yshSWEQFQJkfxgUO_jHQt4P7A7ZT6z56OxzKRs 3993
jax/_src/scipy/stats/uniform.py sha256=ijGxgJabQi95MyiSex0aaBOJnBTUAnJ4KF-lhJcBl3w 2122
jax/_src/scipy/stats/vonmises.py sha256=iCC8-yFGHGH6ky4XD_2ISRt-cjRIh2acHKYBwIPFzCQ 1284
jax/_src/scipy/stats/wrapcauchy.py sha256=-k6Wt4lROA5sK9Juyc78a8YO3imF0HRx0iYLW6UVHzw 1422
jax/_src/state/__init__.py sha256=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU 866
jax/_src/state/discharge.py sha256=wS-uXlK9iOM-WixcObmgO6f7gjqKgce_edxteHBXw7o 36963
jax/_src/state/indexing.py sha256=Hykl5Z-UNtQQ5LduX_OJM3ggZha7z3nH3wjh2FxWW80 8443
jax/_src/state/primitives.py sha256=v6IOxjWutFixvreCKU3PboIeY8NlIsL5tAKjW2QXm1g 23555
jax/_src/state/types.py sha256=AA-MoZixDqnHWh6upqlJwtfaGlKSuJfwi8tVf3SmBbQ 6685
jax/_src/state/utils.py sha256=AwdUqrfZto_M6x2L5L4KCoBARMsjhYwfb3jvjmawjsU 2411
jax/_src/third_party/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/_src/third_party/numpy/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
jax/_src/third_party/numpy/linalg.py sha256=HDpnzipYvKzz7-xxu_2Ft56Ty8czVtpxhmNB2mhIiVU 6231
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=A_SKMrHHpf6AnbS1YgGxg3RPbNQo9FWgOGLLkslwHQo 5881
jax/_src/third_party/scipy/linalg.py sha256=PyQc2ABPlYIjbQF8pH8729L7DIHNAiN6JyjyjYOjZao 2698
jax/_src/third_party/scipy/signal_helper.py sha256=VqLSblEWCItAViXc6lIXRbF-FT83l9305aHIZP9B788 3081
jax/example_libraries/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/example_libraries/optimizers.py sha256=224Ba6KPYLs3zlF3g7dlubbQkIkvoko-WGqDx4svBzE 20559
jax/example_libraries/stax.py sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8 13862
jax/experimental/__init__.py sha256=PIs2CgK8WYTCTx1EODRmbnaQuPWaJb-NDFYy9diby_4 877
jax/experimental/attrs.py sha256=m_iOiq5bKo6emXg7CfpdWAvkYq56Gedhmoe4u4osWgE 9155
jax/experimental/checkify.py sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA 1213
jax/experimental/custom_partitioning.py sha256=VT4g15KfF4BAv1pxOI61_jKdI3AHHkhQCnjQcQxWFuI 22825
jax/experimental/host_callback.py sha256=mKudOnX4IhhbJYL10yHbCSamsArUkpGa3fS_86oE-Qc 81388
jax/experimental/jet.py sha256=k_ZTCD4YyQ7g7bieTLP2zNnKZA5jHopI0eqKYGXwsg8 26689
jax/experimental/layout.py sha256=tsBD272jSV5nas1rK-h27UQpAEVHRs825quM3hHLMrU 672
jax/experimental/maps.py sha256=NmRFDPQdGIFLD9YnSOr9Ui7_kPZV3JaV3MXGcVr_2vo 1660
jax/experimental/mesh_utils.py sha256=n54kJr0Q-Vtwc2w9Yi42bqFYpyz611K-XcqdZS2Bh9M 27505
jax/experimental/multihost_utils.py sha256=xbm-MycVR98sd95Tcstq-UYewQ8JRpvSwxagmrhLokQ 18355
jax/experimental/ode.py sha256=rYtcxYMLz5vR5aiPYiRcQKsHqLFF4JR4taypMeJr78M 10659
jax/experimental/pjit.py sha256=Yq-bII0QH18i7VYcphiiQmvx3xRcr8Ktsr1ijPhVepg 808
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_alike.py sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M 646
jax/experimental/shard_map.py sha256=noCZW0x5wCwnTz2cD5T-FTFHGtBsLlKyGxkFRln47Ig 81891
jax/experimental/topologies.py sha256=DxNh3ErC2h9eUrG8UzN007aE-r_EiZtU7R2NeyWkbus 2140
jax/experimental/x64_context.py sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30 1710
jax/experimental/array_api/__init__.py sha256=XZNTO6JVB_j5kZnwFm2ouu7SxbgRez4IYwgWr7xSuGA 5364
jax/experimental/array_api/_array_methods.py sha256=6rP6HnjVF1CBdSpwf0cEkj1wLL4rbUTKkiuIBaWueb4 1759
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=v3ZQsTh_9qLqdA6zKMGpfbYhRn_OVCueGM3YwqNXtCs 5391
jax/experimental/array_api/_dtypes.py sha256=4itd8mk4z4SpohYVHfELDpGQF87fu2ScT61nBD3tjG4 968
jax/experimental/array_api/_elementwise_functions.py sha256=hPsPNdyFwEg8UGjz8xJnLqDlH_vrLT2_44PIf9HWFqw 14835
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=mdCDSzLE7XMpn73zOe1uq1yyQ9JKrBo72iygziG-ZQA 4775
jax/experimental/array_api/_manipulation_functions.py sha256=egQnTwdYoV8c65D0rh94zUG4F6x_zggjsNiCGUO6j_I 2826
jax/experimental/array_api/_searching_functions.py sha256=VhWWD3rH1lQi4EfoGzKqC74bnNmCB4pmByk0WqTG3ek 1483
jax/experimental/array_api/_set_functions.py sha256=Kv-4PcMtsszLMEgG_qK84kt-2qJGai0KSa2ArPr22W4 1375
jax/experimental/array_api/_sorting_functions.py sha256=XMYMwfgPC8AMC3NIlhuBGe3hGx22WqTmSn65CIegKzA 1118
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=OEDELaMerquTrn2FSXDImzsrawzBUdvWXG-da3MkixM 1153
jax/experimental/array_serialization/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/array_serialization/serialization.py sha256=r5UnmZ0o1X8F2FCrXIHzCG0KiKZqK71IKIP8sJyRmA0 23136
jax/experimental/array_serialization/serialization_test.py sha256=INQMiRaWx7tKYwEwonhReD-BEIYa22XEgcKYjl551Sc 15781
jax/experimental/compilation_cache/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/compilation_cache/compilation_cache.py sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As 820
jax/experimental/export/__init__.py sha256=dk1YZG7FSmrB47imONOWDKSEsX26XyyjI5WiHVULU8k 1268
jax/experimental/export/_export.py sha256=suWuyMuhx6-IHpcOtgtkjkWhXopiUKZmZpPBOBqYDpA 58950
jax/experimental/export/_serialization.py sha256=f3Bu0wxMA0DvJ8Fw1pMyd7W_xi4k5P_Npv9k3HXddR4 17212
jax/experimental/export/_shape_poly.py sha256=gElBg9TnYmy_gpxDGsLAwEd3Wa4ZFqa7Dx9HbL_Eo14 85388
jax/experimental/export/_shape_poly_decision.py sha256=x9AADPHzkyQSjJEtxbIX0rOsWBqT2T2_TJ1oXsgTmN0 20711
jax/experimental/export/serialization_generated.py sha256=wc5JKltgI-SJx6g361gxpfY4OK2DvsaVARhrm5neJzU 26195
jax/experimental/jax2tf/__init__.py sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw 944
jax/experimental/jax2tf/call_tf.py sha256=lZ1TLh0iFSYWRMCcmKkfym_DLLGCUFS8ggqvA6Vy1-s 27665
jax/experimental/jax2tf/impl_no_xla.py sha256=SIvqeN30ld0PqoGduuUC3JwMxvCPvOsqUZnWxF_M9W8 54476
jax/experimental/jax2tf/jax2tf.py sha256=j3QphITfS649XeDCs_X4JJXTV6p4WEzd4eceKxNxADI 143199
jax/experimental/jax2tf/examples/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/jax2tf/examples/keras_reuse_main.py sha256=ZwRgp2kz871Y1IBtR_m8lVQB6wZY6IwdSSD9YLVo0is 2900
jax/experimental/jax2tf/examples/keras_reuse_main_test.py sha256=1FBpLLwCpm9v4udaL0tEN3umiqF4a1lXSr8k483R9Eg 1616
jax/experimental/jax2tf/examples/mnist_lib.py sha256=LftkJ-7__c0JpPxUWdx9uv3QiTGbOKRXzLhc1ixkev8 11520
jax/experimental/jax2tf/examples/saved_model_lib.py sha256=UdSLpLnb2zABl5MWLZ7JRfGinscpWMSwXHb6a4rtFxA 7096
jax/experimental/jax2tf/examples/saved_model_main.py sha256=f4BcJt5KvsNvQrZsKhV4_J6QrxHkppRgu792dy9E2S0 8051
jax/experimental/jax2tf/examples/saved_model_main_test.py sha256=Ij1lZWIgrePGW8aElclPa2F9-ZW_msbAsJSTMQL799g 2467
jax/experimental/jax2tf/examples/serving/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/jax2tf/examples/serving/model_server_request.py sha256=IyW59CMUv4y9enSKT95mzgNQEqF8_65ZUAO_p8x41r8 4952
jax/experimental/jax2tf/tests/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/experimental/jax2tf/tests/back_compat_tf_test.py sha256=9FGx-2zxn89IzuQ6OYhaqA0zII0OQ6L6jktQvHzSqY0 5412
jax/experimental/jax2tf/tests/call_tf_test.py sha256=3bt6V_bcE223ShF5pezs6aoMpVvCf5ouZ1XtBuYSaiU 64801
jax/experimental/jax2tf/tests/control_flow_ops_test.py sha256=cM5bEK9TiF_V7FDMA0Ix8NUZhuNAgdb8K6qBYFrb9fc 10127
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=EdkD4mRtzfAcIqpnrAagOR9YvzBcFJlMAGyNkqCPqp4 59208
jax/experimental/jax2tf/tests/jax2tf_test.py sha256=MdIt1fAu5R6WxjtRO7hc2Lmw4Lre5dzj9HVJeTUlzPw 68314
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py sha256=F_tkcePs1M1XDh83wlgFEu8b8iPZMJwkvPu1yWYI3cs 6710
jax/experimental/jax2tf/tests/model_harness.py sha256=1lzDFrtZV9XmDkkIxoQO0VtNPlmzptr1RUuf15mBjQ4 13951
jax/experimental/jax2tf/tests/models_test_main.py sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac 8928
jax/experimental/jax2tf/tests/primitives_test.py sha256=wP442L9HwVRIG2hBbUy1N1GNLfkPxCS9wIkGS05o91M 14099
jax/experimental/jax2tf/tests/savedmodel_test.py sha256=ZnhxROx6NRlsPuq95fTt39yNHEQGVEGWgMuKdUUyS8M 12815
jax/experimental/jax2tf/tests/shape_poly_test.py sha256=YiGeI2270Z2NM3alt_e88SeptRd7O5Jyt500GgqzTxE 134345
jax/experimental/jax2tf/tests/sharding_test.py sha256=a-OWyXIgrushNtZqrEyoSXCWj6h1X-1JtN9Yvq4k1os 31884
jax/experimental/jax2tf/tests/tf_test_util.py sha256=7L-xikndbablYLUybeHvRQptIL4vsZ-xd7Yoyh_XvAQ 20034
jax/experimental/key_reuse/__init__.py sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ 1688
jax/experimental/key_reuse/_core.py sha256=ZeHQSVN-jnvarc-Onvs8Qi_R_123kbAc4ZddL5jQ798 22197
jax/experimental/mosaic/__init__.py sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0 770
jax/experimental/mosaic/dialects.py sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y 762
jax/experimental/pallas/__init__.py sha256=gGfMuwK6zh0Dgg5QMYa3OJM0PEgNKG7rm3Drncb9oMA 2094
jax/experimental/pallas/gpu.py sha256=JfWURS5-kosWDZTENWHOKolig0-Wk2iD0lSqS_kvmuU 897
jax/experimental/pallas/tpu.py sha256=DPAT3IP_nAFNBDKDmu-1MQcyIZ0m5sDwwWuf6YR1pRE 2187
jax/experimental/pallas/ops/__init__.py sha256=AMR7zXuBocpVTEuKBVKcrP5RJsgTCmzbaxz-Tbtj234 965
jax/experimental/pallas/ops/attention.py sha256=hzjxm73o_bPA6IQV6bHkblFwYLO_ov8GXZoot57Zl6k 19059
jax/experimental/pallas/ops/layer_norm.py sha256=69JJCFFul_osv8y5p5RDyXGObLetVH1AUVUnDwY-hNU 11099
jax/experimental/pallas/ops/rms_norm.py sha256=qaquKh22ka6HMiMZbq2h_vZhzRyZUHjZ4kp92ePohX4 9992
jax/experimental/pallas/ops/softmax.py sha256=M8TJA66HR9OhvdQrA0SzrTrZNyU5n-kSkYuogkcWEIg 2702
jax/experimental/pallas/ops/tpu/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/experimental/pallas/ops/tpu/all_gather.py sha256=7QywIkmmDXtqcCCBYK2YXvvMAbrI15AM9LKDX1jcTUQ 5536
jax/experimental/pallas/ops/tpu/example_kernel.py sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM 802
jax/experimental/pallas/ops/tpu/flash_attention.py sha256=nX5EZ_x3VmIvX-hZ6rgUa88Nq2_APD9ZRFqc109hHNE 49725
jax/experimental/pallas/ops/tpu/megablox/__init__.py sha256=Qutkom55JBjOTRrTwsmPrjkZPUx3om4iqXxp5IUZ5c8 643
jax/experimental/pallas/ops/tpu/megablox/common.py sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38 1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py sha256=3LIXN9rn6UGN2Dn2oEgiI26c8ydgqc6NJirQWOBaZO0 27242
jax/experimental/pallas/ops/tpu/megablox/ops.py sha256=UMptQqdj5o3jgSM32_1zupFNHgD4ft7MQ1zfPbb5xZI 2944
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py sha256=bCNpNWR-hJeynDYamiq0BfIsRtvvBoGu2SopmYG48UY 681
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py sha256=mTLvRwRaF9WOJ29J8bEe1XGKELsqihZJFEF-kWmw3sc 16215
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py sha256=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8 2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py sha256=-5YjykKjvFwHvuk2NKd5L1lEwH5aaKbr3RfjHvNlQwU 73093
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask.py sha256=HddoyCimPGN66heae28qj7JnUOo5d2l-cK3bSlslhro 14542
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_mask_info.py sha256=lLtXMkHOllIL2OVH0HewOo-ySqAzTsqOkHlDy71k2LA 33390
jax/experimental/sparse/__init__.py sha256=hRWildIHCGDjwsHx8I8YzwJk6Q5cbkPSQ2NMPd1EPUM 11186
jax/experimental/sparse/_base.py sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g 3239
jax/experimental/sparse/_lowerings.py sha256=NW3WFfrvsf5O9sxB4vN6fMj6b7F0NxtQEuj3L1oO-NA 7661
jax/experimental/sparse/ad.py sha256=TFSakBd97x3rP0i9KemXKlcIl8vfijmpxdvOggiT6mU 7588
jax/experimental/sparse/api.py sha256=5MsOHBMgSnDn8X6024oDJFBUxY1BuU9ldrXy89rHQho 6631
jax/experimental/sparse/bcoo.py sha256=pWVUYzg9AVseWedJhavgT4k2DDN_N3mWhY7m2iQv9_Q 128304
jax/experimental/sparse/bcsr.py sha256=OUfQcM1pEzJLdlh_TMOz6tzcsdiYLmQzXFdRV9bwoG8 38224
jax/experimental/sparse/coo.py sha256=MKH3YpJvSroCUItkDlUwHZxkJnyG1AXPlLsO47_ooGY 24122
jax/experimental/sparse/csr.py sha256=5XdYEe3aA2W00_TK8pjZ30J0lnWD0dfwIk3YO0_epSk 24414
jax/experimental/sparse/linalg.py sha256=fh3KocDf6MVEx_n1gVt3l15mAMfEJrZefFFtptoUAoM 23291
jax/experimental/sparse/random.py sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA 3882
jax/experimental/sparse/test_util.py sha256=41Hvh5w84LiGxrZ1TPZWTQs1yiTSTCrhv5LP0E-89Kc 9597
jax/experimental/sparse/transform.py sha256=53Js6AbAApsDnItGALSqkw8MORqT5GLDEjImiTQkqCY 36957
jax/experimental/sparse/util.py sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y 4381
jax/extend/__init__.py sha256=cDhCUK2LM4fNu0JYq8amg4G9dHflTjkrSyHSgmSEyMY 1249
jax/extend/backend.py sha256=zDbInIE5czEJduevvmZjKRB_9Avtnvdkf7fHFYsQPrg 776
jax/extend/core.py sha256=gKqI1oYsOW0MicQ81E5Bzl0b_5u58TM6tgDNaSzBvPs 781
jax/extend/linear_util.py sha256=HlQLycGFvu0ZbYw0qU7A5l_mRbf6w75rd4VA7s12yb8 986
jax/extend/random.py sha256=0rUXE9dZkXLXtw_AVdnUVLiolBvtfEyQCJ2V_MXNWgU 1083
jax/extend/source_info_util.py sha256=eu0WFC2M4NgGFlbT9MRaBRUT5dO-Gvqe7AGgQyfUrlI 1215
jax/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/mhlo.py sha256=KkubU5XifH1vf8E9AY_aDM5nbDDafsnRfraerqfP4kA 844
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/image/__init__.py sha256=pEcSFsopr4U6kry9Gy82id9Whs95vn0XfBIy7haJm6c 1027
jax/interpreters/__init__.py sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko 690
jax/interpreters/ad.py sha256=BjuN2XLu4xEetW4JuqsC_FC8oiYRjHDqa4-p6kcv7AQ 3302
jax/interpreters/batching.py sha256=jjy5WCy9pS2mUzdd0fjKViSEPO6L0e5zL5gdkgFR-eY 2582
jax/interpreters/mlir.py sha256=dQAg30IEPU5_3NypHtc9IoTySAM9x_y5aSDivt-99zI 2800
jax/interpreters/partial_eval.py sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E 4537
jax/interpreters/pxla.py sha256=BBH-kEticPc4e06G9kkNL_60BKlhkySRUNl3GPvZHoY 1850
jax/interpreters/xla.py sha256=w0snrW9QYScupJNeFkjKM6s0kvLMMavMkBeWDRYKb-U 3454
jax/lax/__init__.py sha256=snyYN8ZtNsZnY_ZEC4zKNLGslOkbMD-kVt9LxOlSLCE 10943
jax/lax/linalg.py sha256=w66gTsb4Z1X5_PjFUQbSP2FJgN0nNEZ9wMaBS9li6xY 1016
jax/lib/__init__.py sha256=VTtd0mOX-XzQ4OxMJAHpNP4VFFyD0ytw_TG3whRJeTA 767
jax/lib/xla_bridge.py sha256=j7hLTCcHMO9RJw5FTsHsK78RezklPm8pDrQNt9tlHPw 840
jax/nn/__init__.py sha256=uf005wuZkgtF0Tv-1Ex7er0JOHF4wYs655UnyOrcbj4 1896
jax/nn/initializers.py sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs 1469
jax/numpy/__init__.py sha256=bkomDQW_gDqER6PMsmJWrhFwPda8MhBifik2Ysfhz3c 11580
jax/numpy/__init__.pyi sha256=eSqGVP0ELMNsGVt2OqNde0ldXvvQ4VN9vlKv2BT6kHY 38038
jax/numpy/fft.py sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk 1084
jax/numpy/linalg.py sha256=olFy0Mtm-dBsv_HhWgPki_J6jBACg0lzdFSO5M37Xc0 1451
jax/ops/__init__.py sha256=3kFyxifjh_OkJqQDDYjSkQw6EH173Rf-fb7s5aPP1iQ 870
jax/scipy/__init__.py sha256=IsnrQSBykAPMH3rSy7XltTWbqsGYVIySjaSMRk66i1c 1474
jax/scipy/fft.py sha256=0EABRVTc4R4OlsBbYFtTV6_dSkVmUdzSnuAZyPW4_nE 809
jax/scipy/integrate.py sha256=STEn2u4Kw5N-L36DwzH0AoSSvT34knl11TtiZT9KvYQ 777
jax/scipy/linalg.py sha256=yqsNTdJ19X6BxTrZfi8RFJZ_vSjebVkIZnuuKFPCshY 1350
jax/scipy/ndimage.py sha256=JUcTRgkwTxZ6GwR3yMLFU31szaYuUPc83JGfW2yWxOQ 788
jax/scipy/signal.py sha256=ILku2cOpXj0XWAGwBVTv7cb-n2BI_O8gl1R0juJZGxo 975
jax/scipy/special.py sha256=jdGeirmuJrppj9EFta_eeXa_YEPK8Ma5OGNZMNVlY6E 1568
jax/scipy/cluster/__init__.py sha256=vQWrNtPmBCHRT1-G_u6MN9byPWzUfwNawdc9HzSYGcI 768
jax/scipy/cluster/vq.py sha256=LrojEdKqZlhFlSZCYbuCWck7Jn2MeF9aqFf6ip_Y2DM 776
jax/scipy/interpolate/__init__.py sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM 760
jax/scipy/optimize/__init__.py sha256=sLE5C83qmDBkS7qF0IzbeStYQwrAJWAze9-pmTVPFNg 822
jax/scipy/sparse/__init__.py sha256=HCUon--lvEN7ep4bU1gcntvclsQS3K3C2okuFHJpWz4 757
jax/scipy/sparse/linalg.py sha256=sXphTZ_Gba0IfRZ6vuGkvwcslNi0nOwvt-yQ3XrtBVc 810
jax/scipy/spatial/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/scipy/spatial/transform.py sha256=fKQnQ0IXpn5ri0pDjhEWQxc7Oem02FWhD2wb1Om4bro 802
jax/scipy/stats/__init__.py sha256=7D_LWgxpwfFnL46KaPYEc4IVTmDj7K5ynv6fRuP_VUU 1995
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=CfbYJ8U5fEmNa15Rgeqq-X3nEVY1P3GtVDKWj8a5gy4 818
jax/scipy/stats/vonmises.py sha256=nQvzYinMztVFMPZm0fd14ZmHxxoi1CZC_fvRgmBwAMo 791
jax/scipy/stats/wrapcauchy.py sha256=aWOW5KZnRAFa2rgHxwTybH1c-gbzYE0KhGVdy6BNDbY 793
jax/tools/__init__.py sha256=BTyMF0yh0M75AsNwF_0NSd5gHnIVpxhsLznsWkxu3XQ 581
jax/tools/build_utils.py sha256=31JhjUYidLlxyogfXJ3KvjiKSpKg9nb_XF8qLFu8Rrc 3410
jax/tools/colab_tpu.py sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo 890
jax/tools/jax_to_ir.py sha256=ksDTFbKTomXcau9qapiOpD3iw-Jg5efD0lNykcQNtvU 8684
jax/tools/pgo_nsys_converter.py sha256=udKwSZG-RDhlvA_Z0vGkW92nR1BLXGVBdxTU0xLJhVE 2716
jax-0.4.26.dist-info/AUTHORS sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY 313
jax-0.4.26.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax-0.4.26.dist-info/METADATA sha256=iAhoIbsqnaLrNzO0mCwOzxcRvR_NTRMlHyyBQ1uKb28 23551
jax-0.4.26.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
jax-0.4.26.dist-info/top_level.txt sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE 4
jax-0.4.26.dist-info/RECORD — —

top_level.txt

jax