jax

View on PyPI — Reverse Dependencies (1071)

0.4.35

jax-0.4.35-py3-none-any.whl

Wheel Details

Project:	jax
Version:	0.4.35
Filename:	jax-0.4.35-py3-none-any.whl
Download:	[link]
Size:	2158621
MD5:	e5d965963d3407af6a31d3bc74f39b30
SHA256:	fa99e909a31424abfec750019a6dd36f6acc18a6e7d40e2c0086b932cc351325
Uploaded:	2024-10-22 20:56:34 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

WHEEL

RECORD

Path	Digest	Size
jax/__init__.py	sha256=iv6CFLh754AD2umlRuvaUUpnyjPQ7irERrnmQDZOFYk	10516
jax/ad_checkpoint.py	sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U	932
jax/api_util.py	sha256=Jd8iVlf7i8crDpyvCg7OTaumZjtGvTIO9pUHXlXmQsU	921
jax/cloud_tpu_init.py	sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs	651
jax/collect_profile.py	sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk	4730
jax/core.py	sha256=A3FylYCnOvePfdOAgpKNg0el6vL6YmracB4npqGIffo	8133
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/flatten_util.py	sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY	645
jax/monitoring.py	sha256=9JRDF2xgMh8NcGYdk79ytPhRGQ_CCQTJHX7g72jjxWs	1191
jax/profiler.py	sha256=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c	1119
jax/py.typed	sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU	0
jax/random.py	sha256=5ONY8K1aaL7gymnR-KLhE_bl6i6rLjgu1KgvpWOqjws	9858
jax/sharding.py	sha256=2Id3LLKs6Ms3Ou8xzUZ1ZbWqkfSMv4S_bAqxfDOxz60	1624
jax/stages.py	sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218	1314
jax/test_util.py	sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs	835
jax/tree.py	sha256=PTjPWygVYFjdLA7CLayik2QMP8K6EIbNsd4AuyhEpBA	956
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=QIaW0WIq1ApR5XQKggQ7AtlJy4G5142_1bfcBZ-_ReE	5877
jax/_src/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/abstract_arrays.py	sha256=pTl0x1Q7nQCrKEk4dmoMINW6hQuJ-Bezwp5Dd6pw1lI	2577
jax/_src/ad_checkpoint.py	sha256=q4uIZvGwLRa8Xf3yOsyWPgiUm2oRSZX3iFEMMcxn45k	38846
jax/_src/ad_util.py	sha256=asumAmqWn7RJXOxzo8WeHMDwIk9znu2-AV9v7W7SYcY	4212
jax/_src/api.py	sha256=0TkoBIvV8ewpOHawEI_FJyvWFiQigujd3ySXE8MldSI	120927
jax/_src/api_util.py	sha256=ezjkpBmwteUPF47FNqATvRXAjbg6OVV4hAlPcivDlfM	28107
jax/_src/array.py	sha256=C1CXoaKKunwnWz6_uYSCE66Y6i-zIzt0xCLhCwVRfGQ	46504
jax/_src/basearray.py	sha256=ctNFZxsznhdivpAEsnHNfVuSEKaZLvR6RDlH6b0cBEM	6485
jax/_src/basearray.pyi	sha256=F8VxJtn9jB6iGzI-2B_kOedk1Pullmmyl21azn_k-CE	13022
jax/_src/blocked_sampler.py	sha256=_Yow0h3WsUkChkB-ySxQ2EMEYnmcIs2lsD5EtpUmUQw	5686
jax/_src/cache_key.py	sha256=Wqk9HBjiIgilCXVoe7NEaWKRD8y2bniO6eqIKqwCS_Q	11359
jax/_src/callback.py	sha256=TUxh39OtrKGxz17s3XIxPOXN6la0dnOzSgZb9-m7vRg	22974
jax/_src/checkify.py	sha256=fIrpdc0vmmH6iQcAhiF9d5UxY0M8c6KVmvrlJWsVBT0	56114
jax/_src/cloud_tpu_init.py	sha256=M1_pAgBKkTFxnBk4-_WT-8mZmuHBAkbLjmr0GobeZKc	3425
jax/_src/compilation_cache.py	sha256=rhBv2tXVyuK9BjZmdwzBIzQ7DwwINkxzKcvTekoBFKs	11129
jax/_src/compilation_cache_interface.py	sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo	887
jax/_src/compiler.py	sha256=Bx8G8jN1NtfwB8KkipzXCCZLUh3dGd-OApIYo5q3oyU	27703
jax/_src/compute_on.py	sha256=JaX9OWWs3KguOeMyNgLvaakYjucngSi4EvG720RUwhs	2091
jax/_src/config.py	sha256=ME25ZWkS8DPkXZpjIgvDWv8S_DC976NKA3BqZJ6WqTA	70167
jax/_src/core.py	sha256=lDB2rfQKirxuRM991cJUVj8xhUMkrSIIe2MZfTodS5o	123195
jax/_src/custom_api_util.py	sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw	876
jax/_src/custom_batching.py	sha256=PPhjqnm1hJQKmoMr-OYwtt7BlJIUOaDSZ-lb249BqAw	10746
jax/_src/custom_derivatives.py	sha256=Hi9WTt_zYza8xQsp9jLCZ8ZbBQZL_QslF0SDqF5pdco	72650
jax/_src/custom_partitioning.py	sha256=WZ8KJN7FXza8yFyFCHRpYwls3GEhg-o9XNaP0716Yjo	22635
jax/_src/custom_transpose.py	sha256=QD6CdGixm163Vy0yUWtXFQ-jJ67Jq8oQFtdbKAoGAgQ	8283
jax/_src/debugging.py	sha256=_XY-rTEjo_-dARgWAj6R_iE4WPM6wlzYlfEyOwQDpZc	25904
jax/_src/deprecations.py	sha256=WdTALbR_IeHHeuvsWgMXhKGiT3NgO_G6SFPveAI-p44	4715
jax/_src/dispatch.py	sha256=4z7lZzVUCX5KgqHBwsAmsilesXfyY5YUxIr5zh9zcuA	24596
jax/_src/distributed.py	sha256=dEO9cQZD8GibtXesCPfw9Q3USyxI5nKjnjWwLAMoKNo	10855
jax/_src/dlpack.py	sha256=pzQdxNalm0JLOOZJyHaLYSAT5DfP_UTPHFl5lBSX9SU	11544
jax/_src/dtypes.py	sha256=eGFE5gs2fpc_rlaI9EJKYSOxhhhUZJQthArUwpSU0xM	31764
jax/_src/earray.py	sha256=K0j3fyvm2ooXvaTTU04zn2KAwLs3RwI4IrbkkAHG4Rk	4468
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=2yIUoXYA4XLYehScnEoerQIu6geOQskNqQV-_6jerG0	24766
jax/_src/flatten_util.py	sha256=T8p7oExtun3ddhh8fn4XL5_TzZEh6ChM_qiZhmcPaOk	3554
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=SyNb2xHEY_1sa7KEnLkQ9PxFHwKKh3853dlLBksGpkM	18163
jax/_src/layout.py	sha256=HRqNA8RLCEmpxZxnz9Rj5l8G3n4ekelJPA0mDe_bLAw	5367
jax/_src/lazy_loader.py	sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw	1842
jax/_src/linear_util.py	sha256=v3P-SRcFVJ9FQDUOT6jRFP7Q7i3HM78BqhZ51E5u76Q	13858
jax/_src/logging_config.py	sha256=usQjXadUzBGKOSNtwPsXDPcZeQDaoT06Gzn7ON46yIY	2040
jax/_src/lru_cache.py	sha256=4-A3WLeb-UhUxH7D5VrDUdjUl_BxahLXs8TvS4upO7s	6661
jax/_src/mesh.py	sha256=bjDvmlOPZXE0nbYpBCU0uj_byd_UvYUEmO1QMH6irxs	13844
jax/_src/mesh_utils.py	sha256=kCNOdV5tg_ztUva4FMi-9mhs4sii4kaAW36uvaTs_i0	31074
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=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A	1452
jax/_src/pickle_util.py	sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ	2312
jax/_src/pjit.py	sha256=YgbR8TML8MiTYa6v5WErl4H8WkdSaPqqbtyk3qEduzg	116700
jax/_src/pretty_printer.py	sha256=mnGHSMkb8EaXh_1_9dsKOGh2XI_9dSaaAUvFJP-XdMM	15500
jax/_src/prng.py	sha256=jUFmbi4qXO7fxISOw4zt9Io4zKstoATt88YJV0V1FHA	43815
jax/_src/profiler.py	sha256=278hQdUi8B7FVXo-YKhY9jf1qXSaBNem50ZAeB6F4DI	16195
jax/_src/public_test_util.py	sha256=8HSKZbFsZXbSsb5d_PBZDcMyEbTh9-DK-CtULw8MdfQ	10573
jax/_src/random.py	sha256=L118dTz6kw1LnRB0W6An3ko9Qf0XnmrqLK8RO7dFCms	100010
jax/_src/shard_alike.py	sha256=BBiCvSws_KhVzh8TruuAUe7hx78NR5ueZKR-Kvl95gE	3827
jax/_src/sharding.py	sha256=eCz8NGzyBerihub7QQ7XqRSws-Wn3o7kegnf-J3Vzog	8087
jax/_src/sharding_impls.py	sha256=3BpTLcMniQJBY7fAucIkWUktZn_OkdOtj46bmMd87c0	65364
jax/_src/sharding_specs.py	sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI	8617
jax/_src/source_info_util.py	sha256=EMqYD1WALWmyEZ1BQu1lpmlHQdVZkWH2_kjNHcqnLTE	10140
jax/_src/sourcemap.py	sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y	6640
jax/_src/stages.py	sha256=2OA3Q3WTksq0mw3BnajHXqmxceQL9EaW7Ixcn_RBbvw	28781
jax/_src/test_util.py	sha256=FrwR4m1TeA6w22EiNlYsUxPxkGYwfP1u8Cu8UpHOVXw	75596
jax/_src/tpu_custom_call.py	sha256=D1UC8fReRpSRDG5e2-KTFA6odta3-fOAxK8ZV1aG3X8	23934
jax/_src/traceback_util.py	sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo	9103
jax/_src/tree.py	sha256=pQY3bwJ4puT4TZqPcAmmeAdoeHsoG4EkMwFPN-0iwqQ	9310
jax/_src/tree_util.py	sha256=Ujqh2mtzAdQ4tl2GhA5CBMDd40Li_6APKKNlsEMbIRo	47480
jax/_src/typing.py	sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw	3287
jax/_src/util.py	sha256=CGCfqhXavF6h-ZFzML16cAY8r1cJQvAFO3Ygua7XWj8	19655
jax/_src/xla_bridge.py	sha256=y46ilaFIJbcMEHnvsngq1W-Phj9182odUyaiIi8TGn0	45243
jax/_src/xla_metadata.py	sha256=8I7y4gHQPVPksnC8T_jEcH2jHB-wJZLET6WtLhSoe5k	1589
jax/_src/clusters/__init__.py	sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94	1283
jax/_src/clusters/cloud_tpu_cluster.py	sha256=qwH3okD1Zv6ozm9WC8GOxWPvPwB_g13XCnmbxlmiFyU	8232
jax/_src/clusters/cluster.py	sha256=3cxnC1kNeMWfWt0JiMKRlyRq1YrHD199ifPAg4tSdZU	6061
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=lp7io8tsE1PDXZvhlb30WVAu5j7v2OUNRU-wVA3Xl_8	41096
jax/_src/cudnn/fusion.py	sha256=y3uCu2pohDXpND3RoycHtHVv0agyKvlSJPkPjsbydeQ	2803
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=as8c9MPD1_vZhgNX4EpjGei6J2ndH2dgxZ3JZ5A__uQ	69944
jax/_src/export/serialization.py	sha256=85LqMILNiiARTNjojYb8ansLAowCrJIcUTzjBW4cwrs	18773
jax/_src/export/serialization_generated.py	sha256=xqg9DDaCZv1E2bHJoWXcw8T5ukxwIRk3PkQnfo9Vb9A	27738
jax/_src/export/shape_poly.py	sha256=zjbgDZDjlFpV9zadsFVx8Aop3Ea7q4HZRX2qtZOcJXQ	86089
jax/_src/export/shape_poly_decision.py	sha256=7cEAydAto4UiXlOYdq7oHqp4eXwp4lc3OAY-KKgy9Bo	20904
jax/_src/extend/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/extend/ffi.py	sha256=2eA87QBKz9MMfC_z-ZmRokl5rZImjjiyhv1vuIPjEnc	15876
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=bRi3Tq9Jm6oHFSdRElv7Ao-9BESGEuUTDrxa85jI1h0	15345
jax/_src/internal_test_util/lax_test_util.py	sha256=TCvYcOZfaAJ6ktX952MtC3vE1f0Ce0iIE_FhEdRkJIo	12140
jax/_src/internal_test_util/test_harnesses.py	sha256=KTmkB03OGmEyzb_siBYEjst5dUUhXI93RihQUr8CjPc	125985
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=eYOMEHtp-fCfbfH0d_OpSqvxfrhWmkJY_A6ZiuT85Do	32367
jax/_src/interpreters/batching.py	sha256=sXXbdhKca1tsDMXFXHqh8eRiJISy0zILMsYsreTeskM	52572
jax/_src/interpreters/mlir.py	sha256=oMSc0Dp63BqqMkyP83Onpa0tj3OdxJT8B_gHy5WYNrA	128306
jax/_src/interpreters/partial_eval.py	sha256=yGEghhtYqnzO0_wOjfZvAGauBozGWRjGvQegybU1cQ8	123220
jax/_src/interpreters/pxla.py	sha256=jbM0i_aS2AQ_vs2L9ZbtdUJboN1ZScwXanIJTyZKGWQ	129796
jax/_src/interpreters/xla.py	sha256=9_KiDbk4PnOgUZRwKI88E9TQLES9fpGMuZV2HyeJ4oE	7055
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=Zd_8c2S3lsALdY_shghkjLMJ_CZwKtOodjeG0j9ZJeU	253200
jax/_src/lax/linalg.py	sha256=K3Eml3xicmdZATVu4ESE1XD1pDKWElPv7VXPfOeEYcw	100845
jax/_src/lax/other.py	sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc	12497
jax/_src/lax/parallel.py	sha256=KzUk_rImpu1WZoC5xNHYMR0Z7kG2AuPo0vmgvqCut0g	65716
jax/_src/lax/qdwh.py	sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo	9391
jax/_src/lax/slicing.py	sha256=iGjCEGXav-OEXIim_Q0JtONeI6IPIWGFaRXY09lQwAQ	125928
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=tE2Ypu43ifsgr9aC7WyeJsdu6Cg_9dz4ljpSkGkHkv4	4346
jax/_src/lax/windowed_reductions.py	sha256=C9wqD6qF71C-zcNdNdNzRqvam138wfFBM03kttacM_A	42356
jax/_src/lax/control_flow/__init__.py	sha256=oXltllSCun6yudA6TPZQZCLiV-Bd4vyAJOkOFb6Uxns	2186
jax/_src/lax/control_flow/common.py	sha256=Wpr7T_Mm9I3WHLCO_aH-LxjyVWK6h0DtuO_BpMzLlMc	11041
jax/_src/lax/control_flow/conditionals.py	sha256=Nz_nMuN0BCE7jh3hWGUStfz9uc_b3kLR89wj94v0Eak	44072
jax/_src/lax/control_flow/for_loop.py	sha256=10MAG6zkwTPRADA57HuFXL0QiyFIh52Na5GAB_D5tWo	36058
jax/_src/lax/control_flow/loops.py	sha256=u5ctTANfViakStOMZn9Z4OvuvFRfV5KhSXXMDPjP6MM	110528
jax/_src/lax/control_flow/solves.py	sha256=s_mk-cxA63Sg1gn_I3v1T8PWxj-YL1tkp-7MxDPXVFU	18959
jax/_src/lib/__init__.py	sha256=Bj4lVVAP3goWvuah54hgpuvLnddelGiiCYagjEqLc2w	6108
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=6KUUW3fBmKZn0eM7VlIFh5yiSLRQ9PNC1Cl9tAxoF2o	2032
jax/_src/nn/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/nn/functions.py	sha256=v0d93B1NxfuiEJT1HXeR4uwfagrj0FzBRh_X5ZgQleo	36416
jax/_src/nn/initializers.py	sha256=DzkIkSdy2_5s6iwNGzNo9gut0nu-Nwhuij4K9hkoDDk	25503
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=H5tbL-8617EPScRkiZM-cF2ORQnawjZzv9iZYo07fNQ	45688
jax/_src/numpy/fft.py	sha256=ZZih9b0NDQYrD82RwdR4UwqrbXvIKv-px18hyANG9wE	51814
jax/_src/numpy/index_tricks.py	sha256=y5xMJNntEu86_18ALVKc5sB_xNE83XqLpVNLYSsT10g	9888
jax/_src/numpy/lax_numpy.py	sha256=SZm6cGe4rYENpA4jBqjuYTL4E986JIcCJr1KzSnQHSI	468037
jax/_src/numpy/linalg.py	sha256=tFjsnJLvsCf5rDH5w-ZC2zN3bykTrg_met2ASKMbaNI	77349
jax/_src/numpy/polynomial.py	sha256=t1ZaTSKRgP636uvJyQov6PubHflDEPt61z05f2Yqdro	27754
jax/_src/numpy/reductions.py	sha256=OYwls5unD3mODhYOhANAP6tGoa7d13PdpFLSmP_Xk_w	100063
jax/_src/numpy/setops.py	sha256=TAk4vQ6N9-P533Mhl0CQzEBFIQHCmYLTP5Px9SFeLiw	47279
jax/_src/numpy/ufunc_api.py	sha256=ik6r9MIGuPA5Q8CneGrTWTt-NVVaai-bgd4W6KAygoI	23508
jax/_src/numpy/ufuncs.py	sha256=IEbh7e2yY3uxFh6ZyzsTUS4_mSHb6mVKVYvvpwUsMaU	115758
jax/_src/numpy/util.py	sha256=_T43zT7tIzKJFYhP_KHwJ061d_qeRzVQCIRcIH2XHPw	18045
jax/_src/numpy/vectorize.py	sha256=kkpGhVAIzikmzmc3BxbHC-WwsTpYvOy9bt26UrldlOo	13964
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=Ypg7maHS3_-cklEXoTSwrxReDVaEsUHx15D3Zk9BeYA	4151
jax/_src/pallas/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/core.py	sha256=dg7QcHXPAqduS-66I7vDP9_R51uQebXcSzaWFtqYfyw	37191
jax/_src/pallas/pallas_call.py	sha256=vdfH3sGAeRf-EphP0B2xukyMT--eQ-AZSShv0Y2CI-0	74819
jax/_src/pallas/primitives.py	sha256=TJOMDXQ088xE_X6PsBWHyVnvoBZKbxm37AI2pT3Y0CM	32138
jax/_src/pallas/utils.py	sha256=nT2Kxs4KIhPVana1v6tvm3HAEnWO6y_y5GWx-kPc-AI	11060
jax/_src/pallas/mosaic/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/mosaic/core.py	sha256=M8x4zhOVOXKGVtVBPbA411WxWw2TVx2FBVKWkS2SCio	8792
jax/_src/pallas/mosaic/error_handling.py	sha256=nmjLsqhnE9-foi7lwMAWdWNQMdHd2jOMqJtDxhBhr1o	5480
jax/_src/pallas/mosaic/lowering.py	sha256=OU03QCWFlAx4xP8Ttnmq1lkA93FFuTZkUkgApkkiw20	109651
jax/_src/pallas/mosaic/pallas_call_registration.py	sha256=9llakMF8RN6vZz_RkRrVzommgWmJkCZMxC0ftL06P54	9629
jax/_src/pallas/mosaic/pipeline.py	sha256=Mftbmqs3Cpd0nX3f9F1Q_LsRSlWQaVy7hS2moCLFVLg	44006
jax/_src/pallas/mosaic/primitives.py	sha256=dm6_aoERl5SwUxR8J8nIyf1hmoqKoYD74A7BzWCPHp8	30454
jax/_src/pallas/mosaic/random.py	sha256=yBP4tvNPUodwWMSssHDUJkwW92uU3K2wtvziRg00CpM	7989
jax/_src/pallas/mosaic/verification.py	sha256=7uKOMn-UohXETxuS9OPH33I_RBx-PQNQklNEE8qyDA4	21825
jax/_src/pallas/mosaic_gpu/__init__.py	sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU	581
jax/_src/pallas/mosaic_gpu/core.py	sha256=pElIcNUFo_RVJcBBKo_ukKxsItwr8LPPKOz8Y6CKtGw	15986
jax/_src/pallas/mosaic_gpu/lowering.py	sha256=WyF4zhYlaG2JT-WhstqTx1wbVt9dcf17v0m0e_FXX9M	49425
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py	sha256=iVRKIrWxEMYScZ9K-5ASabpeIOpgqg3vfjL6hayIgtU	2303
jax/_src/pallas/mosaic_gpu/primitives.py	sha256=5TDYV6ab24g_66-UBSYUCa5uNWRi9YyacUxg4YeBgyw	20186
jax/_src/pallas/triton/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/triton/core.py	sha256=O9OecyoUxyRJXmHHN4Yh4HcqjCDA4L9BrvMaYLZnIFY	1372
jax/_src/pallas/triton/lowering.py	sha256=BV7g3W-cwpENeqBP6ayH3DSvDpP19b7zqn7uKsEtDeE	88277
jax/_src/pallas/triton/pallas_call_registration.py	sha256=lX-Z9cj2H6imyOYMaxA2UmWZeLybltVNBOTKv_LfK6c	4395
jax/_src/pallas/triton/primitives.py	sha256=on8o0zgU7H19hChzji_twx3QAORFLwOEDxj4Zgd52V4	4223
jax/_src/scipy/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/fft.py	sha256=rEOuNrkRnMOofq0i8XunDVONpgMyUws_vn7HgGCjHAs	15266
jax/_src/scipy/integrate.py	sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0	2346
jax/_src/scipy/linalg.py	sha256=cAf3vBZgDFu71-VVbQa1WBgbbeB29HASp_OKS7wXN84	75010
jax/_src/scipy/ndimage.py	sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc	7092
jax/_src/scipy/signal.py	sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs	47453
jax/_src/scipy/special.py	sha256=PncIKVBzp3ixbQRUWP0uQhChVLvii-kYgvf9ejZgt8U	80205
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=hz2kXBC7VlrJfs_phRbioCiO2xWDt6N0Fo-KtiLfST0	17513
jax/_src/scipy/stats/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/stats/_core.py	sha256=dUhvAaJQoGBeTEdmlZdfd8JPI69M-duDxa3ITV6ax34	10620
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=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=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=yFb9oTCNes0Ir4BljpNJURthrcPPHP3npE1r39nR_dI	40531
jax/_src/state/indexing.py	sha256=9rtibPMDHrTan3EoDTQCbvLtc339H8Nhpa11dZ26hJM	9400
jax/_src/state/primitives.py	sha256=8JH6Ak_bwHhFrRB7CAhOPZBEz8-FmiL32Gm532Mzfss	24974
jax/_src/state/types.py	sha256=sKKZzVlslsExRDItjmX3DyMv3QYjXGc_Ea_9Fg4a_20	12074
jax/_src/state/utils.py	sha256=fv7oalhAmGZhOmhdvGcZo8Z5vbtlda-fx8mPfXx07p8	4100
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=q4iyf45X5uCnPiRSYLYIQokxA3Z2aqg9pAZWCgQZtQY	9463
jax/experimental/checkify.py	sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0	1213
jax/experimental/compute_on.py	sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE	644
jax/experimental/custom_partitioning.py	sha256=E0IrgU4LX1P-khUlG69nf9d4SPi9wbaBmyTiyvE6PbU	856
jax/experimental/host_callback.py	sha256=cuZhGDILVkhFgEFjgwSRH4C__ySPwxeJjxvPVNXf-b8	1207
jax/experimental/jet.py	sha256=IUxcSJxmHvshrQk0KFA2El5ipEOjlVfeD1exwaAE5pM	26855
jax/experimental/layout.py	sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY	675
jax/experimental/mesh_utils.py	sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY	930
jax/experimental/multihost_utils.py	sha256=wF6jU0maOyc3f8L4u-etOGgHi5298YPMzQL9fKHv5yo	18546
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=kxuSVk5trxvOkWqPLUJ6iZqROD3lSKxcsMA7Uq5IRok	18777
jax/experimental/serialize_executable.py	sha256=DYfargD1RSL8Uc65iFWOjm1MU1WArjt4Q4t5AZTTz_0	3006
jax/experimental/shard_alike.py	sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M	646
jax/experimental/shard_map.py	sha256=U7wITt0bYzVgC3RhmyWCk_gP8nCCP8smpzOqqlHUD68	95485
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_api/__init__.py	sha256=JYhDscKw62F_3Y0uaTh2SrJADrKBOOQwwavlvXo0KKA	1090
jax/experimental/array_serialization/__init__.py	sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw	581
jax/experimental/array_serialization/serialization.py	sha256=VS2pBc2z5E7QPQ5-Xk8HmNwI5_kR7kOjvyMdguQyb8k	26055
jax/experimental/array_serialization/serialization_test.py	sha256=y8VsLgbJNULIHi7_os5z8soAAcjnt0Ed0D0Uea7qI-s	22946
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=MAjDiHvy9Nz-Yv9GNB0zL6JVMdHC1tfSY1ZeUmcAmyE	3342
jax/experimental/jax2tf/__init__.py	sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw	944
jax/experimental/jax2tf/call_tf.py	sha256=sG0sTtog95aDvRMHsy5vH1cJ-yhZC-ICLeyDedVw9-M	27754
jax/experimental/jax2tf/impl_no_xla.py	sha256=b7HE4DSON0pum1CbtF_QuKSQoab9zI4cAeWAnzVAJWc	54442
jax/experimental/jax2tf/jax2tf.py	sha256=n4kI4sX1ze-O_iknesSWlktw_XAX3WwW_2P_3R0DTNs	145406
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=SDD02UKUgG8O-G-cLWcrDml2BaNy2fGmuu2TvqeylnE	66083
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=_Powd4Sx2Lxuch5XZCNKRQnnf_V3FZu8CZPQf1VdxT4	66948
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=DUaRRB6pQ7ObD-wDAVXlVafYYAuvX9OaxFLX4CbBfeM	14278
jax/experimental/jax2tf/tests/savedmodel_test.py	sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0	13067
jax/experimental/jax2tf/tests/shape_poly_test.py	sha256=wIOh9tnJnjBGeu9TWf7Mkvi2qk4_SIM05fsILN1eZfM	127563
jax/experimental/jax2tf/tests/sharding_test.py	sha256=cGnkFXOsvqP20Q5IRaht_znOJJmVMQUQMeTZyvddkOc	25883
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=ki3VqNGkO0anW0SxgdQx_rroXnPwt0-6srLVWhZ1AsU	2302
jax/experimental/mosaic/gpu/core.py	sha256=Tg2pPzb_mIEdQTG0fUqoG4G-ZtmVxTGqMa3eKVkx6o4	35420
jax/experimental/mosaic/gpu/fragmented_array.py	sha256=6AyeOfvU4p2McZphzJmD-JZ2kCBQ6mR7iCezUlpMqUE	38949
jax/experimental/mosaic/gpu/profiler.py	sha256=9fy3sPtNADYPoRDEUGYNcf3lvnWHjEvqDSRj13U1wv8	10892
jax/experimental/mosaic/gpu/utils.py	sha256=ugK0G7qhStFJeameotOVYRsPl0IpS3kev2UDp0tkKfQ	34423
jax/experimental/mosaic/gpu/wgmma.py	sha256=sbHVdhp957LVl-iVtQPLoq0Kv4zd2ap5Yoni8rjaa84	18025
jax/experimental/pallas/__init__.py	sha256=lCesVyibtLTCr1_tFr3BbNVvgt05ubrr6MlSEzvLGos	3222
jax/experimental/pallas/gpu.py	sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I	865
jax/experimental/pallas/mosaic_gpu.py	sha256=PVvqNbzkTbQN3MP8Jig1TKAVoYDebV4ag0nla6lZdiY	2657
jax/experimental/pallas/tpu.py	sha256=YYnEWhJ5E160WlH3aKoUI1Z5AMtWnl4OhWP01uaw_jE	4193
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=W5du5PhAET7TUIlj6C0BghgdF9s15aVedNNCk-Mr3lk	19432
jax/experimental/pallas/ops/gpu/decode_attention.py	sha256=c-KfY9ZS28YG2MD1EgwpuLxPICBIifRIXlSke3j4-KE	10238
jax/experimental/pallas/ops/gpu/layer_norm.py	sha256=JXixlFl-AB7m3KQMnUG136ZKhNu3dxx9MOsUz3KJJIc	11124
jax/experimental/pallas/ops/gpu/rms_norm.py	sha256=RJUyOd1PqNR4WcKhwYc6rHERXEBhWjrutmkdGaDYuug	10024
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=HDWf7scXNZg4HyJrcFdbmuiWbqtiVCJFFu0y40pTPMA	50881
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=RoYvEAClmyg5mhWDNY6q-KRUfNgygvIn_cEo8rBzT6Y	73483
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=epW1K_v5OPLTqMr9dFDIjh89sxb3BX7ReDNTi47gDB4	33385
jax/experimental/sparse/__init__.py	sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs	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=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE	7784
jax/experimental/sparse/api.py	sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g	6596
jax/experimental/sparse/bcoo.py	sha256=ngz0oZj-QW3NfOD1qKJjENba_sLL8bisMnq-hFA-raA	128904
jax/experimental/sparse/bcsr.py	sha256=ic0JHXVfYBuR9CHoJRK-JmX3-TsRfyrev3HJOsCBMTc	38249
jax/experimental/sparse/coo.py	sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg	24143
jax/experimental/sparse/csr.py	sha256=bBNOl1KR8WwAW0MtiBNKfebV3W2qTW58mA7lsH6Kotg	24396
jax/experimental/sparse/linalg.py	sha256=G7ndX6rm2nds_e3DglFPogdcNq4vHOc1kmpZd3RHl0A	23435
jax/experimental/sparse/nm.py	sha256=_yfZ2HKjZmZUicFG5BOC_sqlKfo79K7nCvw9sU4Ih3M	8563
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=i6fzxSub8SlogK2eOiaD8PcP9stEKD0WcXv-wGPjkII	37152
jax/experimental/sparse/util.py	sha256=TcMp_1-74yfTUU1q35lXL5ufjSBZ12xFof3-Vpnbd8E	4328
jax/extend/__init__.py	sha256=rzXCgE1QFpnc0HPRj52vJOxD9_a5LzALYh_bQrYCHIU	1265
jax/extend/backend.py	sha256=kn0ghtRuNckWaiLKIAonFzcaBL0_Gm7GiRmsxrRI7L0	968
jax/extend/ffi.py	sha256=Bo8zJigDbyClf75aMNeDPKKJoWQTNdJsp2fvctaVH2g	915
jax/extend/ifrt_programs.py	sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc	801
jax/extend/linear_util.py	sha256=Fi4NWtIl3vvh-6GonAS1PmPIseris_w-ZG_lJsV01oU	986
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=aFeGezH4qE5D7ZhPJZFGGkx1Ahv3-drh-1UeGOcf6kI	6352
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=QYDYW-0BjdeWUd_6bf1e-a7Ku-UfgRl7yHRmqLIjF90	3224
jax/interpreters/batching.py	sha256=qyVCJDVfj2AC7lCBsb50RXnaqvZyQ_sdddEBPZL4PmQ	2582
jax/interpreters/mlir.py	sha256=9Kg6EyCFqLqVIxvAo6euAo_o3UYVmZxJcfWPgcFSrBk	2924
jax/interpreters/partial_eval.py	sha256=Q_igvKqlBbumnFmxG0fSYlhXA7UGc_o8H55Tgp2eehU	4471
jax/interpreters/pxla.py	sha256=lDuw87eRS_K5Wz5JfkSTLEZkDyNAuS7tgG9Hch_19qU	1852
jax/interpreters/xla.py	sha256=jeBirvMzH6DZQZpY-gxH7EABKsePx_v_7ZF0CJY2hDI	3003
jax/lax/__init__.py	sha256=NBNt456gAiBzgBlZWsyfYR8cetYWVWwtXZwijPdkPIk	11068
jax/lax/linalg.py	sha256=xvbvXnAOPZ_z97hjoaYzqlyM9verSaMqaL5ZrEsulK0	1397
jax/lib/__init__.py	sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M	774
jax/lib/xla_bridge.py	sha256=RlgLzihbRKJNPKVumdx1QojpZ-xf9XScRmJO708ZjRc	1767
jax/lib/xla_client.py	sha256=fudTPOfusxghBADe2wPyPlYfAOjzMqwSts0XmBsQU8s	3735
jax/lib/xla_extension.py	sha256=bTVt2zrN5alcfoTpH6M7Oo4TeajM2QYNlR-9ZLfK9bw	1358
jax/nn/__init__.py	sha256=ucg5g4JQjNZnTsFEGi4CtIyGS7E2m9htEehaF6nRBCg	1930
jax/nn/initializers.py	sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g	1469
jax/numpy/__init__.py	sha256=iB4KdqmHFnrSMOJtNKpcp5N6P3L9YTHpnXvS5vHAdJk	12213
jax/numpy/__init__.pyi	sha256=KTEq3q940vtzfNcJNe5GoOCGqxJxaZnw9h6RAyOX5MA	41287
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=hT-5s-5ki7yXeQG3ZO-GMTjSrx3JHla41UNnQ0Njwsk	1718
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=zqgkBIwuwOlCGd-aMwRPdsx9JbykpLYMvfzwEHiMDA8	4194
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.4.35.dist-info/AUTHORS	sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY	313
jax-0.4.35.dist-info/LICENSE	sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA	11358
jax-0.4.35.dist-info/METADATA	sha256=ELKXv8Ch8Nh21ly26BW5tyRAx4XmbMmcSU9_Ej01izU	22670
jax-0.4.35.dist-info/WHEEL	sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA	91
jax-0.4.35.dist-info/top_level.txt	sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE	4
jax-0.4.35.dist-info/RECORD	—	—

top_level.txt