jax

View on PyPIReverse Dependencies (925)

0.4.30 jax-0.4.30-py3-none-any.whl

Wheel Details

Project: jax
Version: 0.4.30
Filename: jax-0.4.30-py3-none-any.whl
Download: [link]
Size: 2009197
MD5: 61b1172afdf1f78ea4a55bedd8b038e8
SHA256: 289b30ae03b52f7f4baf6ef082a9f4e3e29c1080e22d13512c5ecf02d5f1a55b
Uploaded: 2024-06-18 14:47:10 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: jax
Version: 0.4.30
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: jaxlib (<=0.4.30,>=0.4.27)
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: jaxlib (==0.4.29); extra == "ci"
Requires-Dist: jaxlib (==0.4.30); extra == "cuda"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.4.30,>=0.4.30); extra == "cuda"
Requires-Dist: jaxlib (==0.4.30); extra == "cuda12"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.4.30,>=0.4.30); extra == "cuda12"
Requires-Dist: jaxlib (==0.4.30); extra == "cuda12-local"
Requires-Dist: jax-cuda12-plugin (==0.4.30); extra == "cuda12-local"
Requires-Dist: jaxlib (==0.4.30); extra == "cuda12-pip"
Requires-Dist: jax-cuda12-plugin[with_cuda] (<=0.4.30,>=0.4.30); extra == "cuda12-pip"
Requires-Dist: jaxlib (==0.4.27); extra == "minimum-jaxlib"
Requires-Dist: jaxlib (<=0.4.30,>=0.4.30); extra == "tpu"
Requires-Dist: libtpu-nightly (==0.1.dev20240617); extra == "tpu"
Requires-Dist: requests; extra == "tpu"
Provides-Extra: ci
Provides-Extra: cpu
Provides-Extra: cuda
Provides-Extra: cuda12
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: 20604 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=-f6FsMEIaxsW4KchvWzPnZUaRhKCSvvm-Wwldr-cKog 10419
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/core.py sha256=c4PuNexpxVfjQwPbYchzl3s3XAGSiMKmiV203SwwScY 8341
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/export.py sha256=3GnVQzmZviW7c2Rmih_VKsH7F0Pf-wv7RuyalU7FKvo 1365
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=rdbVjKk_jrWIMSCY_5iN9DmKzrB2ao8lXynYwoREOVI 1743
jax/stages.py sha256=vjr2-Kn8mia1PS9zorkDVv-C_0E1C3LiDm63K_RLnTY 1314
jax/test_util.py sha256=Ld3VuPa_Q4huIBWhk-ZImd3IaB-5QKZkWu6QlDYApaU 835
jax/tree.py sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA 956
jax/tree_util.py sha256=CQbhHUdGuidEAZlZKIyCD83F7jebmkzPVit0lclT-4c 3075
jax/typing.py sha256=AlROtZQPE5JQrco8cMLaCNHMVC4qS0V3f982Ri5aFZk 3331
jax/util.py sha256=hcswb5c2PcbRy4oRxlPXUb1NZKy3lhwUaUgIFU5rxHs 1135
jax/version.py sha256=akXF_t9l2_nbQOEA9RWDfZeqGsInRGdoY55yKHsd9NA 5877
jax/_src/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/abstract_arrays.py sha256=EYmgc9uSeRdCPQl7UHH1BVso-3JPx8GJzlnQ3v7CA4k 2664
jax/_src/ad_checkpoint.py sha256=ketM9ytG1ZfOU026SVzsA3241k2xnxhK_J2AHz8d6yg 36560
jax/_src/ad_util.py sha256=zaCPpqzwTIajGRr9CykigMgmG-IsOdgcpvUv526MZVY 3773
jax/_src/api.py sha256=fIC6i5DB_kvpW6mOmVGavbhYIL_X9HiwG_20TJHw7_o 128913
jax/_src/api_util.py sha256=z2imm1SEimGzNKTiCxR96yVFR3PyARdf6Ht9wCXksog 27547
jax/_src/array.py sha256=6zX6NlUBCzmdn00wOyG0_8CF3RgsaqWrIruDpy7qj6o 46360
jax/_src/basearray.py sha256=z_wpG05FP8BPEwy2GYuP_RSL_Z66H4kh5h7IgyqzUhc 4563
jax/_src/basearray.pyi sha256=oIxfLiTe2wCeK-bxwxBACE6ePH-wXTBf9FPtn2zzt3Y 10724
jax/_src/cache_key.py sha256=0NJqK01RmW0YW-ecBknb0ED7WjvfLb0bjjfrGbDBKlU 10610
jax/_src/callback.py sha256=mwhNMKfGhJij1jM19IknKqpcw-6Jy0ZxBwrF76nb3e0 18652
jax/_src/checkify.py sha256=EVELpIfFOfOtQ40KhIKSjtMCW5Bp50kuRyGlgpv28tg 53086
jax/_src/cloud_tpu_init.py sha256=7sVEY5Cg0IF5Lwar8bM8DN61K-vLxS1UzJ0RBdXC9QA 3272
jax/_src/compilation_cache.py sha256=0OQmEp8nsIQIUoza1lsC4h7IZM5yDLyhcGNLwbilDNc 9990
jax/_src/compilation_cache_interface.py sha256=l5339F2eZh6ZLGEUd4nklWZKfMXAGbuJN_iKqivfK-c 863
jax/_src/compiler.py sha256=lq0TiXNlH7mquUmeRfSQIACaRrKZe3hw3QOy2JVvmJ0 25596
jax/_src/compute_on.py sha256=Uj3md0M7BRXm0pKOm_trk0384111N59LjOFZbJe2hqc 1847
jax/_src/config.py sha256=y3p1fX621ZBZNWmY_XAbGQcb0dre8eZmvUqi4tOy3KY 65653
jax/_src/core.py sha256=meLKLA6ECr5NVLXJ9WtVivOqyx3XS5H3kPcynWp4DSw 122714
jax/_src/custom_api_util.py sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw 876
jax/_src/custom_batching.py sha256=E3h4oixXmgCT9mP2wVhEH1jeuQfde9DEKhx9MhNGH24 10474
jax/_src/custom_derivatives.py sha256=YVjWcaMtMtl-628bBMFOm92Y13gUrwzIdNsMOPL5jhU 60736
jax/_src/custom_transpose.py sha256=MLAVQN830Au4jAD_LwkR4F8ilE7o_epHKOPt0gNSVKE 8256
jax/_src/debugging.py sha256=8xUz4IkvV4mvPtGxZxDnASbwNzDZnWYSDy39ENsdVzM 24960
jax/_src/deprecations.py sha256=4OBOhsNEQtAPfFKPPUtOnchS-Rcu9w1ckT1gnsTDAI4 3846
jax/_src/dispatch.py sha256=wNyt0Gd-qHcBreit3mQTfaiTbkf5p1R25TAKJxdZY_c 21078
jax/_src/distributed.py sha256=DZdxkNk40j3dnXhvfX2lpb3j_O0OqCHwFmRikWPi5wY 8921
jax/_src/dlpack.py sha256=VBrJCpLyaftAqahoe_r0BiVlq2ZGB6mx2Q8YQIuirBA 11160
jax/_src/dtypes.py sha256=bgpKobkq4ciTZQ4qMQaxcereUT2Vmvu3rcUy0ThQEiY 29631
jax/_src/earray.py sha256=__Weq82XKinbBn67NoJ-f89gGsF_r4fLutUPvlLSeG0 4141
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=aAwg957xAlCGxQY0pt7tx0NyaGwZEvIVgMOO9T5ZLQs 2085
jax/_src/hardware_utils.py sha256=s0UuPPWP6eB7oqQ8o02dL3RqzQTYTsMKcNCukGoO5zI 1714
jax/_src/jaxpr_util.py sha256=pI6hAmAsMzrkw_Y5zosHN--ZKNbsfjfnVff0zPU4NGo 6412
jax/_src/lax_reference.py sha256=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM 18163
jax/_src/layout.py sha256=gY35O7f8LuO0IqZundMAPZOR_WkFge4E1cAhn7pEGXw 3377
jax/_src/lazy_loader.py sha256=pXw4tMX_AEQhDzSG-5BOiTljnpV1xVtGl0qAwmD9_Zo 1333
jax/_src/linear_util.py sha256=Q-3UMAk6Ykp32Ea-0vvIRTyrV01sO4QNHvpY79ml-as 13807
jax/_src/logging_config.py sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY 2040
jax/_src/lru_cache.py sha256=eFbuZ4e_FaQOk5gTQpysFVIHxj6dsECFOVZpuXWsmtc 5961
jax/_src/maps.py sha256=9ZDl2B6j9mJ1wT6S65DibKtkWrHunnKrki2KciBwW64 86751
jax/_src/mesh.py sha256=GT96mmxIr-ZeEuW31mLv5wd_PBu6EadkcBlrgCFPDb0 11334
jax/_src/monitoring.py sha256=ro92p2vr1PU4f0b7Ghk_kkJQBLoxmO9MQ1n3APyBfqk 3888
jax/_src/op_shardings.py sha256=ZYx-ASvqGPXifwEBC1utdbAJA-3ZCgWM037y6l4Wp4I 3865
jax/_src/partition_spec.py sha256=Rlw94T11ETt8EGpQgagi4QJ3sBCuQ4uvg7FLTMl8SzY 1689
jax/_src/path.py sha256=Zd6Q4jJEK8m4MOVUsiT2ALoUDHvoZjP37SZFYJzacVI 1121
jax/_src/pickle_util.py sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ 2312
jax/_src/pjit.py sha256=uUWvLIkbuk4T6nIaDkWGi8Jc_OyeUJUmHJCey5ugtVw 115244
jax/_src/pretty_printer.py sha256=Lyy44ODraJilE7vr2INV1Fayq3nfWcy3xzNHQFFoSGc 15505
jax/_src/prng.py sha256=cQ_5bCbaa9WC_JgpOErdj7XCDBcWFowiwDTr-SNw79U 44631
jax/_src/profiler.py sha256=sMQC7tVIfOxx6zzv-FyBIrUpZNrN_8mrox58IUJGzu0 16260
jax/_src/public_test_util.py sha256=Z4p1XzuocBDSXwI2HBFkssqPrIGpO3R3VETa5gyi2bY 10451
jax/_src/random.py sha256=QXvN4P1TOPC1BZU8BMxuYk3RFc-gWsyNb6PzquezU10 99736
jax/_src/shard_alike.py sha256=sc7ldeOevCvlBX-XRkO00a8J0EctrBYoWitT9ocbE_g 3827
jax/_src/sharding.py sha256=KthHBIKZfd6GUmCTcJyGOdGJobbSZsrjbKd8Q46RgfI 7787
jax/_src/sharding_impls.py sha256=nMYuUTxB_D0Lc5fZuoJwwYZzOWOj7ejvc0QITnMkDGA 60856
jax/_src/sharding_specs.py sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI 8617
jax/_src/source_info_util.py sha256=Jcpqzw3pN8n-pzkSx3a59MnhfnrxY85jB3EsRJ-4zJA 9896
jax/_src/sourcemap.py sha256=Ho5H15VaHkIcPqVd3TXAz3_0VkZmH4yxXpxsdHdwj4Y 6613
jax/_src/stages.py sha256=O2zl4hZx6TeuZlwlK34RQ6iYO_c0TLUxYSuYpDgoKaU 29159
jax/_src/test_util.py sha256=_wkhq4aCBLb7HxyPZib-URVQuTGnv1pWwYNfu6qbKmg 68646
jax/_src/tpu_custom_call.py sha256=eH44D48-MQE6j2vr_rcV7iNLWUoQNLZNz6Sz-EhRgi0 17905
jax/_src/traceback_util.py sha256=DYMptV1Dw4pg4jqrCu6sfjQvP7K51npwqOVCaubhNC8 9076
jax/_src/tree.py sha256=O9YrlPCuiEe25gXRlePAI1wqvT9haj4wmHwkmVRdEFU 9272
jax/_src/tree_util.py sha256=XWu1g-f0fSSyNwn7qrp1g03918os0pZTYty_Bd-UUUo 46219
jax/_src/typing.py sha256=ZtsYz2BR92KzH0688ZHjBPv6STZmNHFQOl2Cx4xVNO8 3246
jax/_src/util.py sha256=1lbHxRPrrSr9XVzqvwy6iygPS6o2-luM4CrfroIjy3w 19943
jax/_src/xla_bridge.py sha256=_e65HPNJSiOl6ktnrEKDrg6tXXT8ClkiUCxWYZ9csQs 44322
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=hQKLt8oWXuJC8nvnOAWtkVCg4T5Mn0F4ejn2BkdZfeE 38618
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=vQP8ysVp4tDbmTJvS4SZ2VrRfPb69lWSXFUH32FNJ4I 8368
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=uzjmq7bdPfDUxYW0h8b3-YkLZiXiCQO24iBxdWxKa0U 61436
jax/_src/export/serialization.py sha256=BdsXjonB6rMGLTFPm_kGhFzh6HzS2CDQkvALiq9k3zA 16973
jax/_src/export/serialization_generated.py sha256=rkHmXVrDM4PpQYDVlHEYwHGvJ9MV_UsKmzttrW-IwCQ 26127
jax/_src/export/shape_poly.py sha256=-xpfgIBWz5TjZbNylkJloFg9Rhk_1Ykt1iNq4UFeiNk 84895
jax/_src/export/shape_poly_decision.py sha256=RKP2LFk6SD3JaAfroAldWp5IywBb1gMUSSdLSSjSR8A 20690
jax/_src/extend/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/extend/ffi.py sha256=DvNbhE7Hjr8e7BVEgMklgMZvC4fYNkPfBSFpO3mN1Co 5333
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=b7m2TDm2mg8f94sR9iusWT1mCfffWfTYDg0yD6u5LXs 15402
jax/_src/internal_test_util/lax_test_util.py sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo 12140
jax/_src/internal_test_util/test_harnesses.py sha256=glgK2t04KdkxQgoHvI6ndv89y9NO6FFMCGo7GEbFtbc 124371
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=qlEJyZyRbQ_glqDDHbiOm20vlp5iOkxu0_p8FiJdui8 32729
jax/_src/interpreters/batching.py sha256=MvTfWRqfRKjpyuiN85TtfZJPOSp2fzypD9CM2gSO4Dw 50401
jax/_src/interpreters/mlir.py sha256=Blex1DENYHi3lCwRD3YsIOc4lFMv8X9UBXbY2weZnlY 117086
jax/_src/interpreters/partial_eval.py sha256=IZVb-YDp-Mkhf8aqEg-_x_kwOYzOe6YKH32Rhy4EDIE 126792
jax/_src/interpreters/pxla.py sha256=rlCXHOsHw4ubJ_XuJFeZSfMkmBvg7QWIcnaZOozwchI 136389
jax/_src/interpreters/xla.py sha256=6f4ocriwE3wnokddC842rqz6E4UI3MOa6Cs_ha8rq2E 12187
jax/_src/lax/__init__.py sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI 690
jax/_src/lax/ann.py sha256=IEPVOc_BisexDj7lyGFLGw7pIzMyG6dLJ03OI3qEkaU 17453
jax/_src/lax/convolution.py sha256=5dD_ZJjYZh10PXnrO_4LAyC8FjBtD1YKSBA7OlGXV_U 44602
jax/_src/lax/eigh.py sha256=GhrwPlBnfUlCa8e6XNq40kERV9-knvD4j5rnntxdvdQ 22235
jax/_src/lax/fft.py sha256=xrCTzjke5flYvm4anB8URXLiTwHRJDete2s0Eh6qOdQ 7138
jax/_src/lax/lax.py sha256=az9a9UtWwZDmumnhCUv_VUbq4OoXWcw30EhZegOcMKI 215115
jax/_src/lax/linalg.py sha256=Nx93TeeEdHre1bLPvbTuK582560ZadsQk6nm18tOzyg 92885
jax/_src/lax/other.py sha256=wTmrVnNdLQajm04LyGa4ioIiJseZ9dT9bWbQmCSmO-k 10539
jax/_src/lax/parallel.py sha256=6iBePK8-vPsU333EN801WBoaRrPA8UEUuQ3A-Dvbg8g 75616
jax/_src/lax/qdwh.py sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo 9391
jax/_src/lax/slicing.py sha256=wEYKuozHNqUWIcQxPsvJ6VwiXuPdEZWP9IAjiwVrAbg 113194
jax/_src/lax/special.py sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY 26940
jax/_src/lax/stack.py sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY 2560
jax/_src/lax/svd.py sha256=iVY3iFwt0ug1ow_SkFIJd0lNYdn9eXrCKhNbZ0Dh4Bc 8794
jax/_src/lax/utils.py sha256=KHbNG7zWt4w8vrSxSJv5iKubesvLdZlX9HVTXLyr5Iw 4968
jax/_src/lax/windowed_reductions.py sha256=r58DmIzQLm2SDi3EBNI-ErFC4CFLNIIFk4zU_4OsKjI 42356
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=ND1BaJ2ZOoQ0yq4yAWJkF_q841XqN4M1eTB3ZW0obQE 42975
jax/_src/lax/control_flow/for_loop.py sha256=a4oT6OaLYP9AaXw0HfS_0iuOLan4uGsHaodrni_bgHg 37120
jax/_src/lax/control_flow/loops.py sha256=mp3-rt8iCnlDkCuYYmU3nGXpzC0mh7KTNRUFymZc3Nw 109110
jax/_src/lax/control_flow/solves.py sha256=JCyYNtgi_hoQ8Qy1tdyr-UbZ4Y5JhZzwUXwFc6fz4s4 18876
jax/_src/lib/__init__.py sha256=CokuBqYFqbcFbFqLuQQLbYcptj8u5ZwZT8nNk9WxdpE 5592
jax/_src/lib/mosaic_gpu.py sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4 898
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=nzUOcP0arOJbFZ7ho9qHCyiMDHhZpcmxUvC_oVmEwdY 1464
jax/_src/nn/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/nn/functions.py sha256=lsOPEngIxqjaEE8sIloanj-liZJKXgL4o4MFkBnXD50 21153
jax/_src/nn/initializers.py sha256=526DihqSYNUWM4i5EET9rvZl0btVP27vdZs8Tpl-Qu0 25570
jax/_src/numpy/__init__.py sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M 581
jax/_src/numpy/array_methods.py sha256=xpuooSQKi8x5gCuzx0mCK7OHW1CkmALhIxJ4T3moaMg 32851
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=A0TxBxoW2NmUObQXkC6h39Rz9q9B787-qB9TqWa107Q 293223
jax/_src/numpy/linalg.py sha256=jkLG-xHs-YFtsXsTON8KNUsfZh--ht6dRWJxEHbZhyw 77195
jax/_src/numpy/polynomial.py sha256=2uEH06Zqya_x15M2ROYEaV68fwmwCmXs-gKZe6mnXzw 12162
jax/_src/numpy/reductions.py sha256=u-1Jzm0WB3Aw5bHhTMmB_gdyozyDx-a0yEFUx1t_faI 44379
jax/_src/numpy/setops.py sha256=sNXW1VA-4GI4hudzC-wcoiP93W9_qWO1FbZ28lFoi0c 39503
jax/_src/numpy/ufunc_api.py sha256=m6awRsc-YXjgc2GmmcLnDpG-K6XlapsRbUUtRIKPDnc 15257
jax/_src/numpy/ufuncs.py sha256=Y5wcahSe22sfbsUN90hnxyp9jxLcJjFizE4MbWisgkI 31797
jax/_src/numpy/util.py sha256=55wWiEaqu2SHthyoIB6SYdIWCQeUoFmehrWiP9kYwfQ 19129
jax/_src/numpy/vectorize.py sha256=X5XsHruajjT0OJlHXUfPSiVuo8f59unsKGQFXKcXNao 13005
jax/_src/ops/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/ops/scatter.py sha256=Y6pVR2ltOION00DlkOKpz7sNtVDssFuH-y51ZqTLEhQ 17605
jax/_src/ops/special.py sha256=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA 4151
jax/_src/pallas/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/core.py sha256=in1Wa3h5gJi1kbWWk4IbLWHatOM0wLk7c_u8_Zx5HWg 13909
jax/_src/pallas/pallas_call.py sha256=dofBGC7oLIP9H8DxFcagu7qcW_-gBYwdiMftCZX0EJM 41950
jax/_src/pallas/primitives.py sha256=4e4cKinv5KLvM78IDqfe4mgrPEnxJ1EIljLlutLHmhA 22859
jax/_src/pallas/utils.py sha256=aD-qsJhsPq1IYpEFH7pHWld18Tvpy08atzyrPWPFKOo 6047
jax/_src/pallas/mosaic/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/mosaic/core.py sha256=JZJZm6EWoJpyRkY_s1RaTAHetl149D14YHLZrrDHw_c 8119
jax/_src/pallas/mosaic/lowering.py sha256=MlK-KDfwbHExDcaoOrwpwsOHvp1o_VaBWspJrsaBylQ 88147
jax/_src/pallas/mosaic/pallas_call_registration.py sha256=RwM7Vt-BpAEzsALqdjT9Qn75jjX1IdKz81_Z0N83zVY 4581
jax/_src/pallas/mosaic/pipeline.py sha256=yOlq88s3b5vwpsCJsmFMVuZODIP1hr3QnquKuM1DKIs 32091
jax/_src/pallas/mosaic/primitives.py sha256=Yu_2chcqDm5Jt0VfCJSqf0dvB4tgQJcYaGgCJbg5VfQ 25039
jax/_src/pallas/mosaic/random.py sha256=dUizvqKbOILsg1EN2e95NwCfUVQd5mccb0tac3XTELY 5996
jax/_src/pallas/mosaic_gpu/__init__.py sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU 581
jax/_src/pallas/mosaic_gpu/lowering.py sha256=9xK3jIIed8YnKcNX0uO1Wpjlk7VVsussf0gFaGWimgU 13053
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py sha256=JlXOTRrYy8YAlL_BNcDv44aF0tqs2PZ0WOA3EDGqpeA 2630
jax/_src/pallas/triton/__init__.py sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc 581
jax/_src/pallas/triton/lowering.py sha256=kllyQAq3euNaTUiuxFyAZZawUE649q2A3QIUd92AeBA 80079
jax/_src/pallas/triton/pallas_call_registration.py sha256=hm2iy3nlNoeMfFccg9C3BhCE7uzp3JP1QBlQJ1diF7E 4203
jax/_src/pallas/triton/primitives.py sha256=9ht9HAeiusMIVkUewn3wbdVeXSYJYrjMYUXVQ7r9GUg 3663
jax/_src/scipy/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/fft.py sha256=x9AMSAKrvh2SprF9a73y3AtiT7OlIxoo_vAXhmbOjJc 14399
jax/_src/scipy/integrate.py sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0 2346
jax/_src/scipy/linalg.py sha256=sK5zN9ZsN8p8oQFRmTXyvNDB4dygM1Hb5ImZ19ZrrC8 74994
jax/_src/scipy/ndimage.py sha256=EYNMo8KWPjBT6xvr6pA8HOYj22rGNjhAbeqEX8BLPvk 7110
jax/_src/scipy/signal.py sha256=xa6AbLozyka8BnqolVOWYykBB6gcUaE4P1d0MIR35Cw 47529
jax/_src/scipy/special.py sha256=eEmdpHEwnz9Yas12W-PdhWneR-IXMgHrufFXbNLE2BE 78262
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=0FoXis-v5H2Ja3fgULYIdcGinHMCcNfPJyFVt6ol3go 7742
jax/_src/scipy/optimize/bfgs.py sha256=fJ5GdaoI79pGHpFlX0USGF-1ooGa_-bKjMVvAaqqu1U 5582
jax/_src/scipy/optimize/line_search.py sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc 13235
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=TZTud9BHHcEztrE62Kf203FT4FULwOlJ0gXDaODycgU 16904
jax/_src/scipy/stats/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/_src/scipy/stats/_core.py sha256=knsMizDgnXvFDr3Gb8Hhua_-S4XTjdXx6WfZGEo1ofY 7205
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=_B3d3G87qXGtUn6lhllxGfTNoOH8X9oNqgMVIKwqQh0 3299
jax/_src/scipy/stats/binom.py sha256=Imb5lRNyIIQKvOr5D7YZNvllEf8Uo7VDhseeY6bvbDY 2562
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=fzZUpOFkFrxqjbx3fzRnQNd9zp_6RhQ-m-my2MCjsds 6957
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=u43L8WlSayMB-34dSHcdF-MUdIVxcQXTh6AnL-qoIYo 8499
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=lC2yPw3KD8eXCSl2SDzAFlojhUfS2skbOSRFrvQ-YFU 866
jax/_src/state/discharge.py sha256=mVW1wCBw_yiZ_8S1n6Z7lj4QLREqSl4K1S_t-r4ChHs 37147
jax/_src/state/indexing.py sha256=WniY4Oer3HMSJQ6cAwi3aOh-5SCqyeqJ31AC9N3qxog 9183
jax/_src/state/primitives.py sha256=Yk75-IgrRRoFSqj-8UIrdfOH-LSxs92aTY8LjIo2Yzs 23477
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/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=CuLwpckMxPiN61kLYDyxeB0FGIHO0gZVcT43tIGABvw 6306
jax/_src/third_party/scipy/linalg.py sha256=VV1zywtvAn3ieZ4SE0WtmyDDlfkD59ATHyD2nZkCLvk 3776
jax/_src/third_party/scipy/signal_helper.py sha256=Ygxut_RD0Tixc4Gt0AcW32QhNTmV5EBxLYn6RjTUipU 3495
jax/example_libraries/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/example_libraries/optimizers.py sha256=nbNnKfHnQ3ucYLFrs7yi4x8A_FL7psVCoKN4pEdEdxc 20536
jax/example_libraries/stax.py sha256=jNCoL5RlFv9JnLssUHthGRaNcA9xY57BKPxF76Nz7u8 13862
jax/experimental/__init__.py sha256=OG7eTYk40QQN4XxZ7E7mFnz7pI2kzo7FH4tU5bteG60 930
jax/experimental/attrs.py sha256=V5USiVvEd4tfZ7DSNQex16ya9BfozvazNiYFxJlVUT0 9463
jax/experimental/checkify.py sha256=OfaBF3DRHQpmIHEex-3nFtu3D4hHGLK7GIpSVZbewYA 1213
jax/experimental/compute_on.py sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE 644
jax/experimental/custom_partitioning.py sha256=KgBK38cD3X44G1b1PCNEXh-pkWfBSHxtfjtp1q5Tsr8 22384
jax/experimental/host_callback.py sha256=djRPtc0Wv9cA3VBQqH9oYvRAcYWwUaOPu2RxF_i7EvA 85063
jax/experimental/jet.py sha256=FjrIRKB48rgy6rK1apiWXvPT8gGibs7ZC2gkzEz-CeY 26828
jax/experimental/layout.py sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY 675
jax/experimental/maps.py sha256=Z22ZNuBSQGTwtvKyith2ZhlML55fvcyMTZKAtwwLTVc 1657
jax/experimental/mesh_utils.py sha256=dXmNBvtZuIvW-wtypxv9CpMZU1zfMnUKpthEtYS0F3U 29152
jax/experimental/multihost_utils.py sha256=1WaaACPe6JAmNGeuSnAjCkdB_QzhUxjuFlXZABYNJwU 18515
jax/experimental/ode.py sha256=5MJfwT04Jgk0_J72aGLuMVPJ9J_B5gEtOMs8g_F6qnI 10651
jax/experimental/pjit.py sha256=Yq-bII0QH18i7VYcphiiQmvx3xRcr8Ktsr1ijPhVepg 808
jax/experimental/profiler.py sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus 1424
jax/experimental/rnn.py sha256=JfyOKcTOYm-78pDjuVhXb4qIJs4LF1f6cOExHQy19bY 18666
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=v83b9CHJhVG_ErVDiJ6KjAeI9sJJqRyWIJB1ZMxC8Y0 87003
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=QZb8tEnU0uyjvu87sCyk31gOcP78sf8SFDWi5sOlPyI 5345
jax/experimental/array_api/_array_methods.py sha256=ShBvQnAkkkTTlb-ObrnJWxtA7wR-Q0_UIPbaFig4Syo 1649
jax/experimental/array_api/_creation_functions.py sha256=ZzDon7kbUgeIXkRGypVeaI-PQDiGJnEyKpl2uIywUpk 1361
jax/experimental/array_api/_data_type_functions.py sha256=MDFWU2Gov0Ac_n7tOaPGqVKMjixpLRdzXHiHlWrGU44 2380
jax/experimental/array_api/_elementwise_functions.py sha256=HnC1Jp0hMI3sk-UXw550awquLAZV1pB9xhHT6YTlQbs 3064
jax/experimental/array_api/_fft_functions.py sha256=UsD6PchkbDTnefrp9sYowcOCJ_in3cq9Sofbsja4GDo 1044
jax/experimental/array_api/_linear_algebra_functions.py sha256=ab_1c1zUtGQ5DjTufkTQNm5NeL4RZ0rIO-tkt060oyc 1032
jax/experimental/array_api/_manipulation_functions.py sha256=q5EYezNtB43XwzzyuZQ4wBufh25uSzULaEBmF5eNoZo 885
jax/experimental/array_api/_statistical_functions.py sha256=1Lnwem9BFR42x13qf2cDs25bh1PgOeKodt8bi1hqRn4 995
jax/experimental/array_api/_utility_functions.py sha256=sq0nyOK-k7HQeZv8QO_OATXplWLcgvLORKC5pPTLqT8 2697
jax/experimental/array_api/_version.py sha256=5WxaUsOOnZkdWHKRESGLI-rt7A8tkXP2rJWnI56VKoE 616
jax/experimental/array_api/fft.py sha256=vjlTeLNMh6pZn6CjXFeP9itrRqNJvOel6i9CA1_fHFk 937
jax/experimental/array_api/linalg.py sha256=aW6Q6vT4RgF9bUJ9O2YIlHf5fIYfo2YvTDasanf2eZw 1214
jax/experimental/array_serialization/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/array_serialization/serialization.py sha256=sgHQgqV3nDXianEj_A4iwzqC7RWdIhWYp7gkJ5X75Ak 24206
jax/experimental/array_serialization/serialization_test.py sha256=QPNr1RuUEk0cQk_k3Frs86XLKkJUeUqH1ihu_v-7gN4 18323
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=voUU40tcps7sW-o3oVUbT1yaf_jtbsSRdBDFjUFFQGs 3344
jax/experimental/jax2tf/__init__.py sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw 944
jax/experimental/jax2tf/call_tf.py sha256=cSA0dUhe_kEiyB5GSoPhsxGDE-QtKpqpEyXNcuw7ZO4 27371
jax/experimental/jax2tf/impl_no_xla.py sha256=Pp0hTZRHpuWqSezBe6cYSy8_0smsrrccPnlpa8j0mwo 54452
jax/experimental/jax2tf/jax2tf.py sha256=0DRBMqDHylyMrQVa8dWAqnL40LGYZssjUWv9JKfq6q4 143094
jax/experimental/jax2tf/examples/__init__.py sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw 581
jax/experimental/jax2tf/examples/keras_reuse_main.py sha256=SQEnak_ty2Ggj6jSpLuIygFC5WXI2RdrpZXSY1ooiAw 2852
jax/experimental/jax2tf/examples/keras_reuse_main_test.py sha256=P618Ht7Xrt4p4T5Oz83i9KIGC8Ziiv_-4iQVxiBz4kc 1608
jax/experimental/jax2tf/examples/mnist_lib.py sha256=RedlIZ9_S4Unk_vbz9dOf_8nL9VykgD9qrwTkRTpYXw 11434
jax/experimental/jax2tf/examples/saved_model_lib.py sha256=uSg9xqRne3v9XPq2p5mxoDaGYva26bUhKAzosjnl0-Y 7048
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=n6D9kZ_6fsrNjyfoQDn1vUy7A7DtBPiQjZl70mrF5rs 4900
jax/experimental/jax2tf/tests/__init__.py sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg 581
jax/experimental/jax2tf/tests/back_compat_tf_test.py sha256=jdRGLBGS5TQ2lgm157q27aV_uCXs1Pq9pIh6d6M5ksw 5388
jax/experimental/jax2tf/tests/call_tf_test.py sha256=evLuyxp4SfH1uk4wlGUZWFL4Z3HooLHLV-EZruTeFjQ 64784
jax/experimental/jax2tf/tests/control_flow_ops_test.py sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg 10108
jax/experimental/jax2tf/tests/converters.py sha256=eqmQzNLSxZ3fhRIjJ5VpuRUH84VfnIHqju_na1OFFb8 3528
jax/experimental/jax2tf/tests/cross_compilation_check.py sha256=UVVnIwHOdzvbMyMxCekTLW2nZ8a7731kO4zjT1qdCBA 7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py sha256=hdCDsL4hV9KzwXTBvprphBh13aUAV8YuzVok626LnPI 59626
jax/experimental/jax2tf/tests/jax2tf_test.py sha256=mFi0CPhVfjAIjjw4JcUuy18dQoKB0CWt27XicYV5BIw 68790
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=Bw7Mx5u0LYFdNioTXfOAxnX_To_mkyWdK8KZ73B6J3g 14229
jax/experimental/jax2tf/tests/savedmodel_test.py sha256=aRv0nMM6Z3GNMW0Ubzm_sg0lBEt5Ct5JKukodYu8578 12772
jax/experimental/jax2tf/tests/shape_poly_test.py sha256=8yzf-cBaR2QjJFrcp1fL5Bfio3uMTjD4WuDEBTCD9dw 134169
jax/experimental/jax2tf/tests/sharding_test.py sha256=VejnbSL3PUQwbrZ3ZU1zo1F7QK6sCb0c2vU-cXmWWgc 25604
jax/experimental/jax2tf/tests/tf_test_util.py sha256=cJMm3VWpZNwYW0p-b01z-eFR147lnKaQhY4kHGsAasg 19972
jax/experimental/key_reuse/__init__.py sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ 1688
jax/experimental/key_reuse/_core.py sha256=DQ4pjWN7KWp2wmUJ-XGESoJsU4IgGRpin5zAMT6zqu0 22789
jax/experimental/mosaic/__init__.py sha256=Be-z9rRmax_7Ps7HzLkWBgHOMkU_T9JGvbyWkI-njt0 770
jax/experimental/mosaic/dialects.py sha256=seTU0P-foAZipjdeIjopV59L9G1uTV6VkxFX4qFaP7Y 762
jax/experimental/mosaic/gpu/__init__.py sha256=OgQRALMmE6sfmmrnFCfTb5cRMBBeaUulEmFfscez-NE 24800
jax/experimental/mosaic/gpu/dsl.py sha256=UZQ2woOx5EfmT_vpXtqB4kHFqfq-x-Ufow_T26jItlw 1260
jax/experimental/mosaic/gpu/fragmented_array.py sha256=C25_tEw3msJvWzht334vM1umMzPuRPK0uKr_TH4rkhc 24673
jax/experimental/mosaic/gpu/profiler.py sha256=RU40Yh0Z4n__K6raDJbBBwOhXYr-hbxr_2yQPYZ-JLQ 10150
jax/experimental/mosaic/gpu/utils.py sha256=XmhbBJu_-3Wp-RB-HN1y8nhvlNUAHlU37c34UXXIbO0 22847
jax/experimental/mosaic/gpu/wgmma.py sha256=rNiSegOiFuSnVv1eq0pRjvnD9AevH8-In4r0j3Tveng 17758
jax/experimental/pallas/__init__.py sha256=CCCYdl-BbabdAGIvqcHRhuA0bXdLQDELJCRayzwawVQ 2144
jax/experimental/pallas/gpu.py sha256=sgvpHfp8om5F87TEh7lyxzM_nuMFkfnrsEWmB9ndVU8 745
jax/experimental/pallas/tpu.py sha256=LORTj9qIKP1j_KyNpaxU0TZUG2LanpJ1umdeV5NpsT8 2591
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=hzjxm73o_bPA6IQV6bHkblFwYLO_ov8GXZoot57Zl6k 19059
jax/experimental/pallas/ops/gpu/decode_attention.py sha256=YyGtueXxXhrjq4nIntNa-K0FmiG6z1YeGvD5qp7DD2Y 10365
jax/experimental/pallas/ops/gpu/layer_norm.py sha256=69JJCFFul_osv8y5p5RDyXGObLetVH1AUVUnDwY-hNU 11099
jax/experimental/pallas/ops/gpu/rms_norm.py sha256=qaquKh22ka6HMiMZbq2h_vZhzRyZUHjZ4kp92ePohX4 9992
jax/experimental/pallas/ops/gpu/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=x-piJ5UMQ3XhJh6f91VS_EZu4vi6jDbV83AxZmbH478 5575
jax/experimental/pallas/ops/tpu/example_kernel.py sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM 802
jax/experimental/pallas/ops/tpu/flash_attention.py sha256=eQxtn3Xj7LsvOknxOsQe2VHHvtHA12vr7Q-JnCbUfqI 49695
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=FiyYHnafRgeMsbl_HUTb9fxM-sV3cL0M05AVeOqTPFg 27332
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=h5xu9tMelefc_4tIQOyVK6djdRQdg9JyhhGPinznNgI 21239
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=TpPvw5zBBqvleRSZQear7qZrvlZKI_PFwFyc45IR5q8 2389
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py sha256=agBz5_5vNJdsGu_NTLaY0gLgX5JzNlW7-tjZuqIn3bM 72275
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=hl5gKZQOgc31Rk6Mkdgr6YYjILwQGPUajPG2r04IT8U 33382
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=67FL1XxfncZN7lROee0VhQHyUih8HySgGdEfZ2xalkw 128216
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/nm.py sha256=4_n7yh5Koyhq3qoVBP5xHWiLmglcWjSHymiZ8QJkvPg 8563
jax/experimental/sparse/random.py sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA 3882
jax/experimental/sparse/test_util.py sha256=x6vD7pf9qNEqfOBj1a8hf8my7mW_Mu7RR5EK0TmBWF0 9489
jax/experimental/sparse/transform.py sha256=yt9hFd_ua42ZBaYSflRuXcDgqJd45US5_Pu0_Awssnw 37258
jax/experimental/sparse/util.py sha256=dtW6OMw26u1MfXqRaiAazUW7Zxe_BcRjbCvuK-4Sy3Y 4381
jax/extend/__init__.py sha256=rzXCgE1QFpnc0HPRj52vJOxD9_a5LzALYh_bQrYCHIU 1265
jax/extend/backend.py sha256=zDbInIE5czEJduevvmZjKRB_9Avtnvdkf7fHFYsQPrg 776
jax/extend/ffi.py sha256=9_aOx9X9_GtkbyA2pdj2Gi57gehxH0UmReQbmn64VpE 841
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/core/__init__.py sha256=XqUEruxcH7n0c4POF88VQ9vM-Zd24HjvDQ-J3-Nl1R8 1036
jax/extend/core/primitives.py sha256=qCXH7_O1MtW1_pnwJq4PlhuGWmbYE8Ze100fCH95O20 6199
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=jTiAajxMg80DdSI-H7jcJiOVGW4KztEWVnpX8Ign0U0 2756
jax/interpreters/partial_eval.py sha256=siin9wLPOMxWcgvSdbIoBgV7bRJoJvxmBOH1ZaKqh3E 4537
jax/interpreters/pxla.py sha256=B-WvovzyId65My-eOizYtBI5mFvn8GXWIdhqowDRDrk 1824
jax/interpreters/xla.py sha256=pMqYjMs2CafjXY7MEjDK7FWgzQKsBiskziQYjWkExF4 2540
jax/lax/__init__.py sha256=qG0S_o6u-426yVeERcIyd8BYa3hujGoTy9B_d99bDDk 10844
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=F90T5RNBQV4C0t5oy2Yf5XJDublLCKQP3Sz-K8N2Ww4 1880
jax/nn/initializers.py sha256=P6Ck5_RjLUUB-pwiCtCbg5BozocSxtz1UDgdfVK6Ghs 1469
jax/numpy/__init__.py sha256=6mjMj_21P-quLn0_4gnSI2T8hx0ztBoXBKfJEr4OjF8 11642
jax/numpy/__init__.pyi sha256=7-WGBt2MwFNWFlw2iV-whNjAx1ayDdNt6FlI-9mI7HE 39997
jax/numpy/fft.py sha256=uPIPoBVZmUyE1BOCtJeYVSTPwKUWDhbNV7GpzeenTTk 1084
jax/numpy/linalg.py sha256=1v976Pjx1-03riLwe5hwMurNI7e247hitNB4vIrzXM4 1419
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=BoVGqZF8EAkYm5x2GtZHup0VOLNC-DH-aryO4Xv6gRM 1592
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=A98sPJsk_W7OEgD7-ujh4nx_vRGqKUYkJ4hZ1MQIzTI 3867
jax/tools/colab_tpu.py sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo 890
jax/tools/jax_to_ir.py sha256=N0UNIfyRMT0Ahu-Z3grrJauS7wEt4wZGaS2_Orgma4Q 8692
jax/tools/pgo_nsys_converter.py sha256=udKwSZG-RDhlvA_Z0vGkW92nR1BLXGVBdxTU0xLJhVE 2716
jax-0.4.30.dist-info/AUTHORS sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY 313
jax-0.4.30.dist-info/LICENSE sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA 11358
jax-0.4.30.dist-info/METADATA sha256=l0dcxtFvS_oisDhJKGImfU4gCcJzk_lHtgcTvIy9Gkw 22527
jax-0.4.30.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
jax-0.4.30.dist-info/top_level.txt sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE 4
jax-0.4.30.dist-info/RECORD

top_level.txt

jax