jax

View on PyPI — Reverse Dependencies (1254)

0.5.2

jax-0.5.2-py3-none-any.whl

Wheel Details

Project:	jax
Version:	0.5.2
Filename:	jax-0.5.2-py3-none-any.whl
Download:	[link]
Size:	2366814
MD5:	7d40716a42fdb3a15af4debdd284244b
SHA256:	d80954d90e2cce57498b6e9c23a6402c2c39edf86d4f808813ab1d1e97672e34
Uploaded:	2025-03-05 02:35:59 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

WHEEL

RECORD

Path	Digest	Size
jax/__init__.py	sha256=dNQGi_iJM4BTa3f5P8jKxFqkwDAcOQ53MB7JWUvdKMc	10126
jax/ad_checkpoint.py	sha256=jnNhXa-FiezkCFP9BlCul194Pz11olQPdZEeoCJQi-U	932
jax/api_util.py	sha256=avYenJ6674XMZHYtWzplyMPO0j2NXOk13BYb1o3vnCc	979
jax/cloud_tpu_init.py	sha256=r_UB06-qStzlo7BWtH7eZ7OMl3ayrG_CQXWQJJwWsYs	651
jax/collect_profile.py	sha256=44nqGZ4fjcCEEqVGpn6Z8_nDuGFLRtTfUyar9YfGKHk	4730
jax/core.py	sha256=sc46ldkH2iTHKe5pktGUw-u7xAixeLTfa-2Bscv5Hbo	12256
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=CP5IDXwA4QzxXilcTCM1F6435U3dYsdWkDzO4hr_evE	710
jax/dlpack.py	sha256=TWzMqAIedhARYUrnreJ35tEYt0SepzrpvTQtmnSdhEw	710
jax/dtypes.py	sha256=iaMlUUaiUH8zFsy5IJEff4s5QTu0EzD5msV-JYAyD6I	1188
jax/errors.py	sha256=MaZbeSxeRyWRMfq4cj6P0GA9UFykLgX3aURPAKhqbts	1363
jax/export.py	sha256=OzLiZq2zwjC4gDUs-3VcHVFF5IUZtIhMaaTdnV89qIA	1537
jax/ffi.py	sha256=S4EjUxeohrl-Kmmnu8rykPCTxMl2LAxGdaerWLhd5ac	1052
jax/flatten_util.py	sha256=jGwQUB80s2jr78XuEaT1_OP2OUyVm8Pk-yAQq-C1SsY	645
jax/monitoring.py	sha256=U87O_UHrN3UhWxY91BJi7r26ZsGPi0zoJODNVjIv5K0	1321
jax/profiler.py	sha256=w7VxxnWqxWNV5LRSvIofTb11I4ndaUQKptxjaUq3P4c	1119
jax/py.typed	sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU	0
jax/random.py	sha256=TG_JPGzvJzNvfy2OB_24N-5m3e9U7lcwgyQ5U03jhAY	9416
jax/sharding.py	sha256=tCEKtRHY6fAij2jyyyYJDfbqgQJ3EJGsI69NEBWdnLE	1686
jax/stages.py	sha256=lcGVsE8R-t3t5Pb6BWLzNelU-_USwzQ8Qj8bgMdM218	1314
jax/test_util.py	sha256=DVJAPxzZMKdM01Nhx7s2kPe5mcdv8ZL8a_37sgyZPUs	835
jax/tree.py	sha256=tI9yJ2PIEhAFiSBbVKnxV1Wfly6mb3b3mJXECzugJs0	1078
jax/tree_util.py	sha256=ZCuOkjx2V6A0EVV3e9sp7JtvhI5sxHF7u1-Vqpak-es	3125
jax/typing.py	sha256=CrQpi9UTB9MYfg-zEsqP3n2qe28hlY_QEsgY-9mrtAs	3331
jax/util.py	sha256=OXU0bu3n2-y901BwEhxd1CvOeGoEBaXPdvYaVNHOT3E	1135
jax/version.py	sha256=ZiVWIKvpZvq6OV1JeqMVBgvpYmDz15aMcRglbc68WM4	6520
jax/_src/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/abstract_arrays.py	sha256=1kxQA1qnwmri9_GeZHerQmghGZ-zHOy-etSyGHIhTIE	2889
jax/_src/ad_checkpoint.py	sha256=5y3PPwoOcyd2Zau-sK1FFhPJXEWQ9jV62it_7AuZaEs	39695
jax/_src/ad_util.py	sha256=n14jlpCC3f7itcVEL8ijSM_C4ofyB8wbKKTzwWzxbr0	4164
jax/_src/api.py	sha256=61XJ_4wlsp6M_I2wTLzmOreWMC2lfNFUXM6UW5ORLLA	123699
jax/_src/api_util.py	sha256=nop8oIqJ-rlsABgfnS3H4YoklSUGjou040TXPvwJpgw	28222
jax/_src/array.py	sha256=pbyHfoWOA3GczkZiKlCsn0EAmXvmM2c3E5Uxqane2lU	49927
jax/_src/basearray.py	sha256=12hYSzFDsq0LVpPX8nQ-gH5mLE253PBWRSnYvQsuVwQ	6503
jax/_src/basearray.pyi	sha256=D9Q7W2_DF0LkP5BpXRFZROcV2VzNBBgAMCo_QLax2ZE	13146
jax/_src/blocked_sampler.py	sha256=rYk-BNwg0fIXWqgMXG5v3kjJw66Bod7vjNSQTa8atuw	5666
jax/_src/cache_key.py	sha256=8ytVe0xpceRekMKyEUxC0T3GaEKAv_JgreT7xPuCB5M	12883
jax/_src/callback.py	sha256=sdJxpUp2grl0Gus6quVAvERsYaQRpq0a_g8HbleUTzA	33775
jax/_src/checkify.py	sha256=FeqF_eF5RlIKC7AvRg6E2hzbJjSBPaHhvAJw96wO_eI	57387
jax/_src/cloud_tpu_init.py	sha256=Kq4h_zy2YD0XqK3xrAJpCGwY50yItlgI4vj2pNyiZns	4272
jax/_src/compilation_cache.py	sha256=gWbFDIUV4sVCpk6QEdaTJmKRdvww9YOo2t2nfhhSzRs	11940
jax/_src/compilation_cache_interface.py	sha256=T-39MQ5ygMI96IHmJIyBwUobcQwMTXkANIGpCyr0eoo	887
jax/_src/compiler.py	sha256=4ofeDZ4kNIlpS_EAbpWAeIrNTjVfq1nd8hwS_zzYqx0	29657
jax/_src/compute_on.py	sha256=0b74TSV1R31kJTR53thafdtB0sYiFGqmRoAbZwAFcps	1911
jax/_src/config.py	sha256=no_Q0Hmyz6B8xx_O07-Qe4SYO_p5D3bdk3CJX8Qyk1o	68257
jax/_src/core.py	sha256=3eTvzCbAIlmdJrbrsOEr62Gp1jvbZYQ5LxEek-gZusM	115036
jax/_src/custom_api_util.py	sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw	876
jax/_src/custom_batching.py	sha256=0IBvj1I4PTcThLLOfEKUOQ383zIT4oJ7mZllr4LArAs	14892
jax/_src/custom_dce.py	sha256=b4KLWdL5BOrnPk88UObGgU-zXmBENVqaDUGNnVZ7xvM	17198
jax/_src/custom_derivatives.py	sha256=SbEmVs75IxkjJF91mgqz42AgduHE7oVh-8NT0B5L6lU	77565
jax/_src/custom_partitioning.py	sha256=9tdwAV21y-VrDmkEq5TSnM57O1cC0H60c_nS09pH3os	25652
jax/_src/custom_partitioning_sharding_rule.py	sha256=T-X4fqF11n1fs9g4npjgW3e2qzKv9gxxxqFLFPg4EaI	17976
jax/_src/custom_transpose.py	sha256=kpLDKCbbcHBW2Lr5yru0Izzo0pw-KzC-9LykrkNbnw4	8708
jax/_src/debugging.py	sha256=AGrKfIh8Xl0MlaYeBEXEK822vLmZMx1GB6xIbnmZpbI	27819
jax/_src/deprecations.py	sha256=FO0YxHbuhn-h2kFyzUH3Aj93PdQUSFqWyMjFRuyyFiQ	4784
jax/_src/dispatch.py	sha256=_1IHrTv4oxJbQftHkkxdauZdhg9QwNiHyGliuahYvvA	25055
jax/_src/distributed.py	sha256=NoqKqUg-KUD1JVBSZdsLLfdTjPcKFcCvHHVf75rpD0k	12768
jax/_src/dlpack.py	sha256=2krUhEhrQI0qOb40T0dlcRAaQjtMGOUgi8Ss0JtOJ_U	11410
jax/_src/dtypes.py	sha256=fYGDlGoo-4CZJS8nJbN-9oklmNKpsdY75nUqs99-tuI	36460
jax/_src/earray.py	sha256=3xAOp69wfWoKhBSu1pBIWlXKfO9CR72NPApGdVQCTTY	4399
jax/_src/effects.py	sha256=ZqHKVxNurnN2g3-PJNdvYeut8_XEjNul8h_Yjojx18Q	4682
jax/_src/environment_info.py	sha256=TLwjyscnAh8rrQWePe1b14-IeSc-z-XP51Y92dMQeGI	1873
jax/_src/error_check.py	sha256=EtRCdD5WUU4itD6iJ-3OCUzTbfvva-sHR47pVWzzkXk	2863
jax/_src/errors.py	sha256=s3zDLKcfvcCVSrJgNON0YF7bb7RB8Gp-xWinVHcvsNU	24741
jax/_src/ffi.py	sha256=apkqJumxJ3yos9n4p3wIqH0-EtdejUskEyZQOoYMUfg	30864
jax/_src/flatten_util.py	sha256=t0PwBrSY7U22mLp_-xz_UJju-SBsCgtKDJwCRyPQllE	3556
jax/_src/hardware_utils.py	sha256=L03Dapnam8MdnSyBGq2d8nio5gPF51xd_Ufxw2pVjCE	1728
jax/_src/jaxpr_util.py	sha256=OZ4Xa3wvMlr1MwxoiAlEURFZQYcse70duyVe2jvWti4	6495
jax/_src/lax_reference.py	sha256=YPZTCI52PssuXAEnMn4r44ZZ2jDHB0wYj8d_fJDrCnw	19282
jax/_src/layout.py	sha256=Km3wPwPjWyBE37S4V0prIvE0FqI56Nq1u0TCVhuL3tE	5375
jax/_src/lazy_loader.py	sha256=G6czW2Uq6WRC1EjDiX6rslMu02cZUGRkKNrvIhXksxw	1842
jax/_src/linear_util.py	sha256=tZDP_FoHSbaXNDt2OvJjWAUtLhmHflkYh2wmAH3046M	17741
jax/_src/logging_config.py	sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk	4452
jax/_src/lru_cache.py	sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs	6741
jax/_src/mesh.py	sha256=Ia2abnBeq5vMAjGzAdH39wEOv93_v9nQEPCNvvkAQVw	19241
jax/_src/mesh_utils.py	sha256=3FsEOtQfebQX5VjUAVeT8JxNVT8MoJndG37to4JGbpM	32336
jax/_src/monitoring.py	sha256=Z5ol4haHZeaj11ZHKFNgBOUXXSjeOHYmg-xS9ZNQpZw	5188
jax/_src/named_sharding.py	sha256=cyqIPKRjHjEcDwQU3Mbh4zZyYQAeYgTigsSlvdX8kL4	21456
jax/_src/op_shardings.py	sha256=ZYx-ASvqGPXifwEBC1utdbAJA-3ZCgWM037y6l4Wp4I	3865
jax/_src/partition_spec.py	sha256=NJMVJ7lMeB9xC3d-pS1o6msWjLa2xJPDEXuw7GOrNCk	2342
jax/_src/path.py	sha256=ot2lEjzaYVYCCGoOI8YhZYT01HgME-6dtGpmckXd41A	1452
jax/_src/pickle_util.py	sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ	2312
jax/_src/pjit.py	sha256=KUXMOCT8zLE3mSpY_GB9u03edsJdgom7vQcztt5p6v0	130891
jax/_src/pretty_printer.py	sha256=iRZHOftfSZUYXoFmBxLWM58CXEuwvlDVhJk0Njhc_QY	15460
jax/_src/prng.py	sha256=lEsvN_n4RaDRko1-G_qhhuZOgrh9aj6TMD0OZxSgSAo	44596
jax/_src/profiler.py	sha256=7WtOfvJ46GGqgPwfHHGu_I5dXuX-rkyUNmkhkAm2jkk	16768
jax/_src/public_test_util.py	sha256=p-eJnGvUjSPq-gdRfOJ3528GQqxsmxv5CsrOgs3ZKhY	12978
jax/_src/random.py	sha256=Zmm3zyoJugfaJTSSSxRlVVT4_3EfmSlKMFBL93XVGM8	100268
jax/_src/shard_alike.py	sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg	3989
jax/_src/sharding.py	sha256=rCq32k0vOyt75z0GLtFYN3RTRrrVhntd9bqvvkyHLd8	8165
jax/_src/sharding_impls.py	sha256=1twLgGbF5XD5cFA8XuW03NsbirvCrkv5PlpOK9YLCxM	52268
jax/_src/sharding_specs.py	sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI	8617
jax/_src/source_info_util.py	sha256=A1Z0uQ27GRNezdJov3WJsVo0atOmT_9Gh0Mwp0L5tz0	9833
jax/_src/sourcemap.py	sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y	6640
jax/_src/stages.py	sha256=uFxepDUrIweHoxobO6BhYOAY796kTiIZ7pOogTQEZ80	30070
jax/_src/test_util.py	sha256=RpudS01f-mEs209O0FuawpzK9Re-DXOP17el0yEJBRg	85894
jax/_src/test_warning_util.py	sha256=9CvuFdeyGcmTH5xfz3Wqs0qOUGpOVcOwh4SnJsp6FbE	4074
jax/_src/tpu_custom_call.py	sha256=TnHQmWohsoo3GB4lV2CtaqrO0P7AX8fTAk-aWsPXUtY	27743
jax/_src/traceback_util.py	sha256=5DgCqqOxQ9JNrreoZx7u3ZglimkfIFxHT2tVrlnOcKo	9103
jax/_src/tree.py	sha256=SIwfh6UL_xEuEMq-ky4bmOmoq1LllYa0ZUhmh4l4-gM	12592
jax/_src/tree_util.py	sha256=ZXnw6yfNQQWY3eVNxGl99n3Xgb5-hlgGBcCoACnzvd8	47698
jax/_src/typing.py	sha256=_yNwYCbn30QE4Ordkc5d4Y2bF2M78C5LhK42EggAetw	3287
jax/_src/util.py	sha256=gyh2p23EYo4cXo9Hsilp56G8ZGkDsyaL9VecHIo01lg	20238
jax/_src/xla_bridge.py	sha256=Xl29uaxYdpdzJFCpFwr-amVDY8dHus6s2_JnNIC1JaA	46409
jax/_src/xla_metadata.py	sha256=cwgOkbnHA09Loo7DTwZVYz4NzVmRyfDiHj5V7XDSCAg	1977
jax/_src/clusters/__init__.py	sha256=blIgNT92mYE9Bst7obP5_xhZ2fE2hKqwxH6R0cAQZ94	1283
jax/_src/clusters/cloud_tpu_cluster.py	sha256=xpeiJ-pVH6uNypUfIoxwNVaRuATiEtIqGVtNrvYN_lE	8405
jax/_src/clusters/cluster.py	sha256=4_heeOPzwH7NHumJKZspEfCReABCJC9SDfW8z3ORZp0	5856
jax/_src/clusters/k8s_cluster.py	sha256=TzUXU4ksY11X2f6iYjhb5LEGZfr6r5WZM5tc9DQhvBQ	4188
jax/_src/clusters/mpi4py_cluster.py	sha256=ugDhirp00Hh3knxX3qVb5yxSz1d0_O1DuQUcVfyO8rg	2608
jax/_src/clusters/ompi_cluster.py	sha256=Vfk3AimGa2jtDEwjpJJXuYP9p5uv1tBwEKtpGn1f4IQ	2317
jax/_src/clusters/slurm_cluster.py	sha256=hsZQJXznkCXGDTfKgMDgjVf4bmbAtSne6pa0wW4iJT0	2321
jax/_src/cudnn/__init__.py	sha256=03MR0G6UMqHXi-JBeyj6tipV6ioP9a-iSN1mW-8Ws5A	631
jax/_src/cudnn/fused_attention_stablehlo.py	sha256=p5Z7a3noeK57664nqmNBbvm681FAWOMmbo3FQcLxRcc	71831
jax/_src/cudnn/fusion.py	sha256=3ifqe61DuTyoOXQL7Nelacz2KRdcJdPRI__ub4s_vI8	2808
jax/_src/debugger/__init__.py	sha256=iZU1X5nkpSBK5pwLSuWhZsPZgtLUovBF3tyWPmQoh1Q	899
jax/_src/debugger/cli_debugger.py	sha256=C9FQoMwTZOesOTG0X5dVDT_3nGTExqKZixdPPzTDkew	4780
jax/_src/debugger/colab_debugger.py	sha256=PHmiL-QES4t97iICJt34a5FXSNTm-AaSUERPGCZ3QHA	7823
jax/_src/debugger/colab_lib.py	sha256=jCbwj2Q4f8SEo7CBr6AeiUtzlsIv4BfKa5CVbq4Bn2U	4298
jax/_src/debugger/core.py	sha256=WilcRMqpzssKa5YL_P7SFtKaUVUaXqDutxO_N-ZRnp0	8593
jax/_src/debugger/web_debugger.py	sha256=Or-gJL40oBUfClu8mg1YUBoKA9cQytZgJAeGGrPPK_w	3433
jax/_src/export/__init__.py	sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU	581
jax/_src/export/_export.py	sha256=5ATxCvLMyJ0NL0ZKg_qssSxigk1B6YBG-CWBumNVCRo	66075
jax/_src/export/serialization.py	sha256=IJLWeBn2fOAprd8AvEnKrXTLokxjNQXwUU1_N1d80_A	19331
jax/_src/export/serialization_generated.py	sha256=74xkuHItNLL3ZLy9wC1Hi91Ii1ueK04FKAOYhsoAfKg	27792
jax/_src/export/shape_poly.py	sha256=9QoxUe9expfO8ClENW_a4yjxpTbkyDGdNtvwaX1vQYE	85956
jax/_src/export/shape_poly_decision.py	sha256=x24hHa1CGTskS83TMeo7xGILKsrgufYsGReiKLQAJtc	20476
jax/_src/extend/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/extend/random.py	sha256=1KmQCCYxFMJO8KB0qJz9ZVQhnK-05Sv4Jn0jXLPpWQw	1258
jax/_src/image/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/image/scale.py	sha256=XL1VSkCM4gtOOFurIElk8nXPfUB802nw3P8wlqquP4Q	13774
jax/_src/internal_test_util/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/internal_test_util/deprecation_module.py	sha256=iZ6fRYNaUcOwrftwpDXMnlxrLrrQbdMrbybmmQFwql4	851
jax/_src/internal_test_util/export_back_compat_test_util.py	sha256=S8wDJb98vYYeIq-omRo_b8jErOBa2DJAoX9mO1V-cI8	15336
jax/_src/internal_test_util/lax_test_util.py	sha256=H924uHW6vU2ez3tN-9mrT2AXnqeaHSB95Kzn_PWRcPo	12998
jax/_src/internal_test_util/test_harnesses.py	sha256=d3BdtWFyEKoB8v8vWTbp01UfhQkzHagQkpSptra28zc	126010
jax/_src/internal_test_util/lazy_loader_module/__init__.py	sha256=UP_c2VfWHRhfEi6n4F6qLHZuageXhPiVSAcUKHrrxFs	705
jax/_src/internal_test_util/lazy_loader_module/lazy_test_submodule.py	sha256=luZv4DGyWwdcBX0_NBVwu81IE4jFYVbRkKo9gQPmxPk	610
jax/_src/interpreters/__init__.py	sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk	690
jax/_src/interpreters/ad.py	sha256=ETt6FnKdRel1zdL-tLOegCuzT2mDqd-cUvDE4tE7QwM	52567
jax/_src/interpreters/batching.py	sha256=gYX28O9P7pC0EI59n1nO0tdQdJmriYbHycmQg0k9fAI	49904
jax/_src/interpreters/mlir.py	sha256=6pvlFnA9ztinjq1EgInCSMVMMX_Ut6EG3pkyN5bUvO4	126062
jax/_src/interpreters/partial_eval.py	sha256=RGKG6fiUEzu2BfiC95AGGinWmAwqSnvBGxnrYPM2YNI	115220
jax/_src/interpreters/pxla.py	sha256=CVHmj4XRCjV5FT5uNqvgi4HBszb1TqOPaksRLQLaCks	135178
jax/_src/interpreters/xla.py	sha256=h7Sbg8U01o0vXNCMN-cX7TBX89i7cl01tXt-xLAM8Ys	5424
jax/_src/lax/__init__.py	sha256=jRL3qQoUco8BVUAh8lV-KuY38h_4GxXcb41asljW2yI	690
jax/_src/lax/ann.py	sha256=53ap20vxBD-hV6L1JZsLNfzyACtK8qkE2fmPqsJivvk	16761
jax/_src/lax/convolution.py	sha256=kxGQ5J18irVz3S1sxqoqlTW6otJO_5g3zPCHi09FRhY	44551
jax/_src/lax/eigh.py	sha256=MAWftHqilChJ0yd2-aJxSpiu6b9RYnY_y_SUI5OfrnU	22418
jax/_src/lax/fft.py	sha256=fIrp5WrJCZM0FYZpZnmC67Y7rsWCthbOwK3GXryoEqQ	7258
jax/_src/lax/lax.py	sha256=_kVlC62n_JecsubbTabePhiJEhONsVFjcRCIwokj_yo	318229
jax/_src/lax/linalg.py	sha256=0GB6Ue6tE1HjiYjd_Wt-EHwC8etl1cmVn-JyXWFrpNM	104403
jax/_src/lax/other.py	sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc	12497
jax/_src/lax/parallel.py	sha256=6gFfqLA_R0xWVMIHO4dGILjsqpQ7Kgf50FfLZSvNUMc	73554
jax/_src/lax/qdwh.py	sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo	9391
jax/_src/lax/slicing.py	sha256=rZR13Hznsr0ioCSXzHqELbtYuIIB7s3E2n_dyG5iL8Y	129385
jax/_src/lax/special.py	sha256=djqaB1CuQX53mqxeHQhz5hV6kZGN-HVhtzAwqezU5dY	26940
jax/_src/lax/stack.py	sha256=7hxK1zlVa_Q9gO48Ymj8S1nA8DN6keYgcOY0DAtbjOY	2560
jax/_src/lax/svd.py	sha256=3ol5rMgm_uy40yZzKs-9BoDwFn4oUqbcDWbKcvvVok0	8826
jax/_src/lax/utils.py	sha256=ukeYvooKG3spoADeMKef5Gqe8rrnrj-ZdZxt0pOHU2s	6075
jax/_src/lax/windowed_reductions.py	sha256=bveO1IxCntSGUlQeSW3rwnnP7-at0D3mUXmtOqEgQ9w	44109
jax/_src/lax/control_flow/__init__.py	sha256=SUQsNr-Ctcs2Vgmb4HIzqVidxEpAHFe1J9WsM0aMegw	2200
jax/_src/lax/control_flow/common.py	sha256=Og8kjaUhkBW-KI53J5-cNl-3cG5NJ7ofn1kSNfW3cPU	11273
jax/_src/lax/control_flow/conditionals.py	sha256=L75bvoxc8iojwifM1BBxmmsw7n5gXtq_G86bVPPR7og	43912
jax/_src/lax/control_flow/for_loop.py	sha256=9L4Zv3LTKKCW3mlce9cFZRrK6zYb5UIiBWNdgyUy9-A	36370
jax/_src/lax/control_flow/loops.py	sha256=xQKTwpgtso9MXaC6om_3BoBfY1KhLsHJcUFfjfmC_60	113665
jax/_src/lax/control_flow/solves.py	sha256=pD2XMF6cPBZCYudC6oel_Q3RpqVBMCqqOiuANvnxsQ8	19983
jax/_src/lib/__init__.py	sha256=b8QrDVINTxFtbEe5SzUOsK6vOYLeafuUTyEkY7JAVt4	6621
jax/_src/lib/mosaic_gpu.py	sha256=lHnM852MFg7K5UUwry1zN_r5nrPSWl6ZVwOxVnsWAz4	898
jax/_src/lib/triton.py	sha256=lMwAguUTQxYs9HkdLqeracg8sz5RL7gm1W6PpzSrzdk	2202
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=pISrxFaPYtuBASFC4Pxl4U7rMdREjZrddIfdRUcvMrE	37500
jax/_src/nn/initializers.py	sha256=BC3FncBzhB7dKEuZfqFpcyKEuNMbIOyGukbs_5o-kBQ	25612
jax/_src/numpy/__init__.py	sha256=-QhmzWj2qFbKVIT9TDAIvQWaKQibiRXy3r1zQkI3X4M	581
jax/_src/numpy/array_api_metadata.py	sha256=je15NAJj_CFlgqeDqo5iSg9EzmN9IwCkZ_E8TtStKcg	3347
jax/_src/numpy/array_creation.py	sha256=c-Rjys1Fg9Rvls30ujL-0qFwQYxlM-NAOg--HhV6aWY	14214
jax/_src/numpy/array_methods.py	sha256=wJJ9lBf6r7pOU_r8iN0qCRVsX0TYHtiYC-w_4aJwRPw	46143
jax/_src/numpy/einsum.py	sha256=eUa5qD7I5_UuGJCIyiwQ2YI-pymntvQj1UhERiFhJro	23115
jax/_src/numpy/fft.py	sha256=VRX4rk-0SClGXZaup_O4mcPWhoSON3pf7GtvwV2grAs	52578
jax/_src/numpy/index_tricks.py	sha256=aRFc9So43wmgzk-eVGBeN16BI4Ny7eXYFr7yEtrgDqM	9992
jax/_src/numpy/indexing.py	sha256=IJfONyGzsGuVAOrhdbRi5QSP3zLGV48RYTB8KNJdclw	49936
jax/_src/numpy/lax_numpy.py	sha256=9vQgSK3q41QOsGCNwJ5E_xFx7WAxHPZh951XCv3mHIg	359119
jax/_src/numpy/linalg.py	sha256=n-ibZ8YjOAqr3ocUqQD2CBSFXceapNJzBs8L3QbWuOw	78306
jax/_src/numpy/polynomial.py	sha256=uvJnJgbZh3OfMYT2wEKFKEWZH_fSdaFyfknKaHb3NpE	28106
jax/_src/numpy/reductions.py	sha256=GWd4COgUXLlDm3r235dZ2fLFpJ9KXqa-26Od_h-V8g4	109348
jax/_src/numpy/scalar_types.py	sha256=TVAucYbP2u0RfH3AepWS_OhFQ_UFTrrhwNu9aJ3tUE8	3929
jax/_src/numpy/setops.py	sha256=dwD1X4yXyUuMP6BCdnsCkZS8iKfCQoFDTaBoUA-ZS84	47594
jax/_src/numpy/sorting.py	sha256=DvYC8Te38zxcBImc0egun35xGcD2zNpj2KP1c2SvY54	16296
jax/_src/numpy/tensor_contractions.py	sha256=pRMGmFpDvHx6ouWE3sz-MTLMtW3wqmYegzDYoxuX6HM	24176
jax/_src/numpy/ufunc_api.py	sha256=kRvpLgvTjIwNrV7DBtDpDJ1EkmFu1twcDT8TtOfc5h8	24284
jax/_src/numpy/ufuncs.py	sha256=i8KwYXArPAPHt9N9DsvHo3N357vv-6D5L2hRX1LJWLM	116554
jax/_src/numpy/util.py	sha256=PChYTY9Cdv8a4UpJPQpVN8CNz7shBUKEfGjlik3ipHA	13342
jax/_src/numpy/vectorize.py	sha256=FLPnVlL9HSddfiu_mvZo4WWBjTD67OYONJkXuXS8hZs	14037
jax/_src/numpy/window_functions.py	sha256=_xI_pbsvC8Ai_8hVdchPx8gQNWYgKGPED8TR2GRsqXY	5524
jax/_src/ops/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/ops/scatter.py	sha256=t3jQFLGKvbm2gIoFZH7Qx9Ws0w3lcKsI-cPZtABXKqU	17685
jax/_src/ops/special.py	sha256=tm7RbrE3fHxkBi3eU9UjYUP0zm69aQYQ_3Mql6wYnRg	4206
jax/_src/pallas/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/core.py	sha256=E8CSLgmv-xrgRmv3kZb1Ntj2AjBXp_hRio1xuWS_5U8	40243
jax/_src/pallas/cost_estimate.py	sha256=zjl4nUbPh-BVZS2iMltWxWArWzmQQ2HwDcCDKFP_GAQ	8384
jax/_src/pallas/helpers.py	sha256=Qq1v8UeE0RyzVbEvGLye_LufSFXW6daz2SDx6bdqKgI	1689
jax/_src/pallas/hlo_interpreter.py	sha256=_W_24-K83LNO4-3oyGeXOG4tiBRebKKnorQlyDYoVcU	18899
jax/_src/pallas/pallas_call.py	sha256=up7zfc8aSM2IwyH4HVnt3d3gpAhj1Iat5z2G7VuGIV4	66415
jax/_src/pallas/primitives.py	sha256=Z58oI6UE3Q6T2fvHlN-x_UiLncLxLGzXYILGZ9k7XPU	33109
jax/_src/pallas/utils.py	sha256=7Tor_59JAGvLDRbFMMLCqIcDQ-8y8UOrSjVlwND5kKw	13887
jax/_src/pallas/mosaic/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/mosaic/core.py	sha256=Kb7csjsO43k1O3oR8dOuYYIn2kBgJ72T17YmIm3AL-I	8948
jax/_src/pallas/mosaic/error_handling.py	sha256=FCbB-b5Dl75erMmr-FMvc4QJwCXqn2z7ioNC2Eh0vQc	5542
jax/_src/pallas/mosaic/helpers.py	sha256=TH8O3K4k9TMhxXKZvshmOjEBPjoZuxor3TOVrPhIjKM	2765
jax/_src/pallas/mosaic/interpret.py	sha256=Ukr6T8ZuXP6pT6ir9dzn-Z7zETVKMYQSU49IH6XKGLA	37227
jax/_src/pallas/mosaic/lowering.py	sha256=wTAH2qfPqKY6r1FPDIWwj3fcJI7C_w0p1b9IpN8xBIM	128937
jax/_src/pallas/mosaic/pallas_call_registration.py	sha256=owMx7meUeCpMl30WMYnH0QjgFHuQJtiukqd-sW9HKzk	9307
jax/_src/pallas/mosaic/pipeline.py	sha256=bu2aJcvWNFheJq-pnPDRpULGmhv2ZMe_UeH49eqbAaI	44520
jax/_src/pallas/mosaic/primitives.py	sha256=YEmMvIjsOM5c7VcwhOkWmABpBxdLRx9cro6RQ9JmfKk	31905
jax/_src/pallas/mosaic/random.py	sha256=yBP4tvNPUodwWMSssHDUJkwW92uU3K2wtvziRg00CpM	7989
jax/_src/pallas/mosaic/verification.py	sha256=fQAVzbETThqBWA_yp2jLbeX_5meoVShh3flYgQWiF9g	24540
jax/_src/pallas/mosaic_gpu/__init__.py	sha256=mS3YSYUZFIG68T5Zo9CFGkT7wCyWRycnWV4L36CnHYU	581
jax/_src/pallas/mosaic_gpu/core.py	sha256=ELHtbEkY1rIukRQLCCPhrxRwVyaxmqru4FHbkr_aquc	18619
jax/_src/pallas/mosaic_gpu/lowering.py	sha256=_hzEQDgAqkna0opE3echanQz2dYllBUMXh1mdQvzaSk	76773
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py	sha256=DdJbOpeZbT5KJuvqay2JxqNZSE_DKct10LtfBh69vTU	3759
jax/_src/pallas/mosaic_gpu/pipeline.py	sha256=-Fzbx2lfeNm23qwkbrVN64cMSf9BNMywPOHHPN5_gUc	24126
jax/_src/pallas/mosaic_gpu/primitives.py	sha256=hTA1Ip25MY9M19Xsj7qRpo35MDh5gEHwv5ANu7QuU9s	25251
jax/_src/pallas/triton/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/pallas/triton/core.py	sha256=5u6hubCM5N7aSYK5iExZpffEGrmlpqR6Op1xd3dOhDk	1374
jax/_src/pallas/triton/lowering.py	sha256=gn9BdzquEMvlWjiOsTco1cOrQ-Ni4qfnURuTQzKAmw0	92183
jax/_src/pallas/triton/pallas_call_registration.py	sha256=i3kTBTkaEgYhW8lzhUjsb3xBTCLtYjfjn5g2pnaFW4M	6831
jax/_src/pallas/triton/primitives.py	sha256=_UjvgpOsEaRzul3xizwzPdP1blZ1-NzChzZ4bsVmSaA	4228
jax/_src/scipy/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/fft.py	sha256=NFy_bDIzx44Lpkw1-74fBm2jxeXmzIUp8LTVh3-6Z5s	15281
jax/_src/scipy/integrate.py	sha256=4VtR2wYVO037Z0yiOdQbniKtpxL4boZQNIppy4J9qy0	2346
jax/_src/scipy/linalg.py	sha256=Du93VsPxiHA1t5iCG6-DjYMPXr1UrpS5eELjlRdts3g	77680
jax/_src/scipy/ndimage.py	sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc	7092
jax/_src/scipy/signal.py	sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs	47453
jax/_src/scipy/special.py	sha256=-F5ApAboZ3LyQFwzWWOUBtEdt3tgm-ZIXOs8VhVzpBY	82636
jax/_src/scipy/cluster/__init__.py	sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8	581
jax/_src/scipy/cluster/vq.py	sha256=3Xfr7esTCzSawrDGZ37hu_MXJ-EqnZtff6hYmXhz0Fc	3043
jax/_src/scipy/interpolate/__init__.py	sha256=S-o0CP72882Xv3l6k2PlrEx6eNvXBTk5Oaed5KF4iL8	581
jax/_src/scipy/optimize/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/optimize/_lbfgs.py	sha256=kqKkdDeRYiH5prawdntOXfCs-g311NYwnY57jV8DDbw	7769
jax/_src/scipy/optimize/bfgs.py	sha256=JRdDBxnfNrVxK5UDyehFGF3eQTJenWfdZVmKOMheenE	5609
jax/_src/scipy/optimize/line_search.py	sha256=E51ROBUJYmanL1byuPpIa28HmN1vCf_BAaBC_dQZgoc	13235
jax/_src/scipy/optimize/minimize.py	sha256=mCojZkL0TXQhix6w0-GroM7ITU2rodYdDwmvynLsqWs	4979
jax/_src/scipy/sparse/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/sparse/linalg.py	sha256=IGX8lZ2il77S3WA50Lz7MdBiw7lqdD8mHgWRaCUHX6k	27245
jax/_src/scipy/spatial/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/_src/scipy/spatial/transform.py	sha256=yD9HOfgQpQO3-eACTh5zW8qzV-WLcrTHY_5ZlsnBoSk	17525
jax/_src/scipy/stats/__init__.py	sha256=9AYvmtjvwOmouHXvS-uU2Id1YBE7CUbYVE__dR1koWg	581
jax/_src/scipy/stats/_core.py	sha256=Gq8H86uI_wWodhyvdEy5GiNulBfW3kJZvm8Z0Qw6Vdc	10653
jax/_src/scipy/stats/bernoulli.py	sha256=zOvo5ZLr8y3O0byR0cp3JRPKPPHoz8oVyG3-C_5-NJo	4541
jax/_src/scipy/stats/beta.py	sha256=h2LF8lIveFCa-qNZ_XLD1hQIK9_dLSmlCjrqeSdz_ME	7988
jax/_src/scipy/stats/betabinom.py	sha256=dmhOAUk6py5Fnu-jR95KnwJ6suvvfwNF9CXNHz73CSc	3447
jax/_src/scipy/stats/binom.py	sha256=E7a3go8bCM5bSiC72l1pDQEuuzSpXLmgfG40iPF90wA	2825
jax/_src/scipy/stats/cauchy.py	sha256=26eHaNIfxwh1og3kTqnPYgivH0GTIuWVwElai4VJwv8	8884
jax/_src/scipy/stats/chi2.py	sha256=vPFrhpbLRPDheYenmHQ7QvZrRt8cAyRtOpU9VmemNIY	8008
jax/_src/scipy/stats/dirichlet.py	sha256=_wWyx07z62_EH25i3J4jS5q53oj0UH5A3-XgQzapwbY	3198
jax/_src/scipy/stats/expon.py	sha256=W0tt3HtC5jB5mEM-bQbAC4U9FKVJFspsqnPpOAN6_FE	7813
jax/_src/scipy/stats/gamma.py	sha256=W5B8z6LPD3Zi3cbcWHDB-RU_OuKjXopJIspf_VKZMw0	6987
jax/_src/scipy/stats/gennorm.py	sha256=kp22L5H-rR5n7sn9rpeWsWta-F76yZaXkO0Ovvf9zc8	2989
jax/_src/scipy/stats/geom.py	sha256=2hQ4LoaqNHFBahdRTGS9oyWMYdFQufJb2ByqOw9709s	2243
jax/_src/scipy/stats/kde.py	sha256=qY2ZLH8HVOU50DhQBDrhH2XTLT1eUHAFx7qGuvhgSmc	10139
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=S8nPOcsDpEJCuiMjy4IuA-ivwVvUdOt1oPWjkHurSkI	45534
jax/_src/state/indexing.py	sha256=zsOFen-6wn-fOGa5oRpVp6i0g6PyM1WGbWK_-7XuHws	9317
jax/_src/state/primitives.py	sha256=CuEVSWlUSKDgLqDtmZYnh6zH6mUTgsJqQV_Y7iwnt2Y	25227
jax/_src/state/types.py	sha256=ulentcXvyKoZC2BttsM1UoR8jg8VYGtHav03Lhwdhu0	13003
jax/_src/state/utils.py	sha256=BBCCdjdCyaZmnB1Y6FSfSZ9JyWiAbTX8c1ZaXfR-lt4	4107
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=weGxx01LOfJDklgdX8vKJVUpTy1_FV_9PWEX2hgNtB0	9136
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=hDJ7FjYhNftLa10Vcf0R2iwktQ26C3EGyPpJW8QVh7Q	9899
jax/experimental/checkify.py	sha256=V84hTTNrwnckaRzgvWDmpcKKA95kAakjZwwlfnRaXt0	1213
jax/experimental/compute_on.py	sha256=Mdg2n8JtAsrq1ap_PwUPwoqpoCNGO5rjf7T47e_iaAE	644
jax/experimental/custom_dce.py	sha256=ORRk_kBjbseQ06T_LmLPxYCO7-Rby5U8Erro5U5Gi4A	682
jax/experimental/custom_partitioning.py	sha256=RlaIvfdM8VqlEJC3FajqCmB_EOHyk6iyfGUgaUyR61g	1054
jax/experimental/host_callback.py	sha256=cuZhGDILVkhFgEFjgwSRH4C__ySPwxeJjxvPVNXf-b8	1207
jax/experimental/jet.py	sha256=h60mC5MyamuE6zVLbtv35Sm0bd8x7WCUYSISVXGZHxA	27229
jax/experimental/layout.py	sha256=q2i-_DZq6t3TaTmtkRSwh5CiseuwbH-VMqIMBUL4ItY	675
jax/experimental/mesh_utils.py	sha256=X4EYn9ZS-cEOBncDPtqJZA7A4FDEOWCAw2YP7qoJMNY	930
jax/experimental/multihost_utils.py	sha256=mz6TzO1_GisaP0X4hNUjhAItbp2Bp-reHXwL1W-le5A	21708
jax/experimental/ode.py	sha256=VmaWgwXaD1N9WspkCJ7pvdbUtYRuGtp4cpQaqJkf__8	10859
jax/experimental/pjit.py	sha256=2Qi4A4tKn1p6pjxysiCY5UAsfQq_KVZ7QpMX3gRJw30	749
jax/experimental/profiler.py	sha256=DJ0ETyXEZ56c6LYWRY_ankgPOs5oWZejGtGXS-Ttbus	1424
jax/experimental/rnn.py	sha256=-7OVX35DS1GfTWxTU5quyhKxdrVzkeaJ_-ci-pGdvoQ	20207
jax/experimental/serialize_executable.py	sha256=DYfargD1RSL8Uc65iFWOjm1MU1WArjt4Q4t5AZTTz_0	3006
jax/experimental/shard.py	sha256=xTXar7eZIDvVAXyXUdgCPaqHEUmbl8zPh6iF5oLt380	696
jax/experimental/shard_alike.py	sha256=dbrhVCnWEX39TsQYRyGoClFByDmqtmDlKGtOmlT-m2M	646
jax/experimental/shard_map.py	sha256=QFXWDNi8T2VB_S4qfjKxSa_l9SwqgWFOLJvxYRAodWA	99056
jax/experimental/topologies.py	sha256=Lxxb2W6nT6AEhM2ekkNUKm6W1tgh5i-_49454NuxCkQ	2101
jax/experimental/transfer.py	sha256=XBkE6sLLjmVj5kXD2xzXYFKem5ezJ6M2cY-rFlFkEAw	2475
jax/experimental/x64_context.py	sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30	1710
jax/experimental/xla_metadata.py	sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk	658
jax/experimental/array_serialization/__init__.py	sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw	581
jax/experimental/array_serialization/serialization.py	sha256=Fv3JGT287UWmx9YD9qGJBsZL9yAcamDcbiBRmOI2m-0	26066
jax/experimental/array_serialization/serialization_test.py	sha256=t9v6BZXYAQTlpHJGf0Vi-apf-5wiklWuj-yh1AlRoS8	24295
jax/experimental/colocated_python/__init__.py	sha256=0RICZxmbab6vzz9vVHDqxITH8xM9FbCVfXHZZ8CA7fM	980
jax/experimental/colocated_python/api.py	sha256=UdsHC3bp63LQxDOdC3jPKHa-BnEegbwX_UAAZvWGWlY	2307
jax/experimental/colocated_python/func.py	sha256=OusSq72YbwE3wpP9BDOXL3LOhA81g1JDfAcnoWbpj_0	17447
jax/experimental/colocated_python/func_backend.py	sha256=lXN-FRbWIWezsICyIc3QYwjpF9-tYgZV1Ut7TBN_dpE	1373
jax/experimental/colocated_python/obj.py	sha256=NmkXkY5-pDuX1JtwcXl6imJxehUcuOq7uCaH3Itshf4	5631
jax/experimental/colocated_python/obj_backend.py	sha256=50I4Ttb17qRk7IUrtufYMNlXqyl4KY6bBc61lGQBWfM	2283
jax/experimental/colocated_python/serialization.py	sha256=EOEhFL_X9GSO-3ino2KngYsrPjtGn1vukmdFpC6jFU0	9055
jax/experimental/compilation_cache/__init__.py	sha256=9jtOmvvDMLWHLUuqHkvlxcTzvN1yoUxeFmRAxYu7lMw	581
jax/experimental/compilation_cache/compilation_cache.py	sha256=OIZLvS1c3Yw_EbbqlV28pC40g5dPdBB-XDrXnHcT6As	820
jax/experimental/jax2tf/__init__.py	sha256=ZqsPlkalGAkzdTwRIutfW1k8NmOHzJn6kfIB1A98pdw	944
jax/experimental/jax2tf/call_tf.py	sha256=FG8j8FhPEv7WiW-kAG-ujSzWGpTC4IKoAIgyFuKvhco	28339
jax/experimental/jax2tf/impl_no_xla.py	sha256=7ztEeEhJycNm_yA263efIAa-FsuYeOI2cvvgrqmmgNY	54478
jax/experimental/jax2tf/jax2tf.py	sha256=fkOoNRhsn5kgvImLn6ycGANlRnj9tP5q_v5cMj7JXFc	145108
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=j59m4moGqfRxXehwVrd5aErwRYoSIiV5QFHK_7k4yR8	5476
jax/experimental/jax2tf/tests/call_tf_test.py	sha256=HxoQ9V7DL4LiI6aUeXBHrwW8x2zfN7ZUmACbfwSzWRE	66938
jax/experimental/jax2tf/tests/control_flow_ops_test.py	sha256=sqYQJWhcY-awqj6u6IF26qRBF7xPzodkN2XnZpRMleg	10108
jax/experimental/jax2tf/tests/converters.py	sha256=uYSiDOPUNaxcgVwLExAgquiK6M0VFH7j55OnbruOyaM	1998
jax/experimental/jax2tf/tests/cross_compilation_check.py	sha256=holjJ2qso1R1VD_QSmKj-HnSoh_dkJgkOVLVqZqPIcA	7400
jax/experimental/jax2tf/tests/jax2tf_limitations.py	sha256=FWkCsZeRHZq93XP2No2CtiO4uLq24AcLbh6kLns1Frw	60812
jax/experimental/jax2tf/tests/jax2tf_test.py	sha256=Qo_6eyZuKlpn2L6s8gWKFUD7ml8hOglt_tNproLt5jY	67012
jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py	sha256=2ouDpIkKBvLLomEsmYU8jMKIm4iy4xKjONdlEG1Fy-s	6645
jax/experimental/jax2tf/tests/model_harness.py	sha256=-rgFOZu0GAtUx2ozfe9q-QpJ-7CHT3NxsUS8CJrbCsg	13870
jax/experimental/jax2tf/tests/models_test_main.py	sha256=qk_lozA90084XzxnrkmjzLd7RvEGfUIMWNM3v2FG-ac	8928
jax/experimental/jax2tf/tests/primitives_test.py	sha256=T0TVQqQsFzNNZ27c3WIbXMP9Re3wV6jI1bUa3wh28LQ	14582
jax/experimental/jax2tf/tests/savedmodel_test.py	sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0	13067
jax/experimental/jax2tf/tests/shape_poly_test.py	sha256=qiCrR8up-_lPyYFGQ0w6YbPRavWVEt0TkxY1lkGdzSw	126545
jax/experimental/jax2tf/tests/sharding_test.py	sha256=6BzliayHqYimzNPsJOJJvHk7Fjmvu_E1BmMziI0qyos	25810
jax/experimental/jax2tf/tests/tf_test_util.py	sha256=YIli6L4hltJraWYyPbK4hOTZZ_64UafOn45MbMNpkGg	19962
jax/experimental/key_reuse/__init__.py	sha256=5CjqLFSLnAhm_YyrowsWA48ENwxcvcD5KQrNQJTgLpQ	1688
jax/experimental/key_reuse/_core.py	sha256=X_5kJxhc7zgosxVWt5lWaBOhAmY581dqYfBnJPj_evA	23025
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=uuY1iC4kY9bCPHZ4OO6fgiwlyOj78TEudis376J9-P4	2925
jax/experimental/mosaic/gpu/core.py	sha256=Q0CGzI8hFYpUKwT_-AP9YccWfjduxuLUo8UdYuxVp64	25435
jax/experimental/mosaic/gpu/dialect_lowering.py	sha256=YqFjCLWmHhzxe_V5yuTgPioJjcFUxEOMPwNE2YWTuko	20302
jax/experimental/mosaic/gpu/fragmented_array.py	sha256=eV5I14KchDIqGfKPq8PQBAfGDi5OZK59OYhDQTDLBMI	82134
jax/experimental/mosaic/gpu/launch_context.py	sha256=JL0HzKmcd5eIX-ugLzWrvMFOtcdls0N209z4T32UPHg	27590
jax/experimental/mosaic/gpu/layout_inference.py	sha256=9myJ3qIRd7YPU8TGFJTTcnvkWW0mwxSLGPUZAog8Wnw	13725
jax/experimental/mosaic/gpu/layouts.py	sha256=3PAdOdot08L61y_lTbVVOnDBrSjKqoz-0txXTo29UXg	6265
jax/experimental/mosaic/gpu/profiler.py	sha256=0u-Os0I2OFyB2O7OoWlkxrwyzXiH8rj4rT2q1w0rVpA	14632
jax/experimental/mosaic/gpu/tcgen05.py	sha256=ZqNjLKMQU6jhXqkQmrI2GZaJ8Lsn-XUy10HBdoVCP34	18669
jax/experimental/mosaic/gpu/utils.py	sha256=VpIJNe-b_uenppyWw9__faizd31h_s5WjrluKo8vwwM	39311
jax/experimental/mosaic/gpu/wgmma.py	sha256=Pm3lkh2f7yIuZD7zpW-QEo3aHF30qwhtAXIgOaRAruw	19165
jax/experimental/pallas/__init__.py	sha256=9AMxQmcUKiyWo6xE8B5hvZ_jOBBzTYu49X9wHY4pik0	3527
jax/experimental/pallas/gpu.py	sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I	865
jax/experimental/pallas/mosaic_gpu.py	sha256=O52d4ZAfBSncr2LLJUYYLP2tjqMALGJW25IeoPu7KnY	3077
jax/experimental/pallas/tpu.py	sha256=woM6EOMh7nybebDo-Fyk8xilkj1OaeVO2v6PrAVl8_Y	4111
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=3RC1NexI4oLsYmCPHtaQhDT1WVW6dzXFsm5SDw6JAoQ	22429
jax/experimental/pallas/ops/gpu/attention_mgpu.py	sha256=3SCzp21Mnk9Wfdf8DS--pO-R3AppMC1NoAoNyyqPAL0	18688
jax/experimental/pallas/ops/gpu/decode_attention.py	sha256=X_FCW76hLD98IY5QcwlgT1heF7SWe3Bc65I5vFmE2h8	16482
jax/experimental/pallas/ops/gpu/layer_norm.py	sha256=OAdBY-Ijtai5IOUfXafJAtudlOUrli0g-_Qhc-t9jQU	11142
jax/experimental/pallas/ops/gpu/paged_attention.py	sha256=VXmKKeLEObv-XZg9iweUWaA2BLGfCDYOaUCMFxg-oXE	14058
jax/experimental/pallas/ops/gpu/rms_norm.py	sha256=hNREA4EYeqPbF3RAijhEvtFk4jP8IiNy5LhPGcQldwU	10033
jax/experimental/pallas/ops/gpu/softmax.py	sha256=ydajdpaW_oAOwwdGJUx-t2TmIx9QfHhnXX_lNdtmVL4	2774
jax/experimental/pallas/ops/tpu/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/experimental/pallas/ops/tpu/all_gather.py	sha256=oLZefmIUQ1Ag_C3kz5h114E2l4OiW9-Nn6DkpB4AC8I	5581
jax/experimental/pallas/ops/tpu/example_kernel.py	sha256=jzsyvU-Lk0PIorE9XCw2NDdqdp8Cn4bqsSe76DBxeEM	802
jax/experimental/pallas/ops/tpu/flash_attention.py	sha256=2g9jECYvE3S-HnVPmABhsQgFsiw6JU3wacYhYIGFAWs	50787
jax/experimental/pallas/ops/tpu/matmul.py	sha256=jfpCbL3yQ37So4yOtb9pF6K7PtA0WmV65yrzQ8yp4cs	2726
jax/experimental/pallas/ops/tpu/megablox/__init__.py	sha256=HRWx7FtM9_17OLitbxwIZp-A9RM61uSL6QAcBGJyw_w	650
jax/experimental/pallas/ops/tpu/megablox/common.py	sha256=9zg98VzHJh5l4Sf_SKly1nrXFWeWeyKoe7ZSPn6Ht38	1969
jax/experimental/pallas/ops/tpu/megablox/gmm.py	sha256=vNNgQIirDK7pyDViP2foec7GC8EhJuxNjR1mp3kHwjU	27249
jax/experimental/pallas/ops/tpu/megablox/ops.py	sha256=4Xg3O0bnP5HTWsawS0hAm8mF9HTvRHVK65RlopkA1rw	2903
jax/experimental/pallas/ops/tpu/paged_attention/__init__.py	sha256=Q6gFLakYVRJDRxxvqq6vmtDYup-edUHZy54j9194JfY	700
jax/experimental/pallas/ops/tpu/paged_attention/paged_attention_kernel.py	sha256=f6z0ukhHjObUOYoWsLc8Wxvhv8g3UVpjGxUixdnf_UE	21960
jax/experimental/pallas/ops/tpu/paged_attention/quantization_utils.py	sha256=5y0QHJo2yDsafF_lgeSO_izChspLHpVOvbFWZ-2d0ss	2556
jax/experimental/pallas/ops/tpu/splash_attention/__init__.py	sha256=8y556G1Vpmglvw8FbNzeB4x6QK6PLkKUBnMsL4vleNs	2738
jax/experimental/pallas/ops/tpu/splash_attention/splash_attention_kernel.py	sha256=nAmc3lH3X-gs43dRa2azXKAeXcTQWuivEJ3zWGzyBOk	74036
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=n_N42IhHPuugEXqUeUqxJCqhDIXFYA-AXsvxMuKf3OM	37970
jax/experimental/roofline/__init__.py	sha256=ftyNH95ANQHFYSq9hsweKfJNP47sfipMm-3XFsr32yw	1260
jax/experimental/roofline/roofline.py	sha256=wgavD_DHNYW5e7jBv5Ipr3CEl4_Wh2i99NFDL74OVGY	10203
jax/experimental/roofline/rooflines.py	sha256=dEDES6qL0CiH7sKxEWloEsaVIWTInUD0tvicMgn0xB0	9323
jax/experimental/source_mapper/__init__.py	sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q	1698
jax/experimental/source_mapper/common.py	sha256=Ke2VFVAeqfElMIQl8BpPIRk_oe_fHjLmjZGGMRtMtS0	2471
jax/experimental/source_mapper/generate_map.py	sha256=s2CTsneQbgeilefUHr4PbY1ScqEYUbd0hJq66vhdlDo	2075
jax/experimental/source_mapper/hlo.py	sha256=zqrkCEpKC0RD_XTMEd_FbAwryspaaZjc_QaNig1WP0E	3921
jax/experimental/source_mapper/jaxpr.py	sha256=wuvV9BWN5r2-dcmtf3g8TzTaY2BwPyJBq7bLEJq7aG8	2564
jax/experimental/source_mapper/mlir.py	sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M	4686
jax/experimental/sparse/__init__.py	sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs	11186
jax/experimental/sparse/_base.py	sha256=qyiBgPa12uwS1nLpSDw8dTD1VEq0v9P2_CGTNyPCA_g	3239
jax/experimental/sparse/_lowerings.py	sha256=Wgpt5l3eCB6Vv8PKOrNUh6A_7i4JJ4xyhZxeI7zMvgk	7666
jax/experimental/sparse/ad.py	sha256=YT1jU0dsB3VoX5d0QUx9ihNUx2L_HfAh6ouJuEdhmXE	7784
jax/experimental/sparse/api.py	sha256=DrqEZmlIr6n6kpJjNli9J_rqMX3UO3sPPW7E8_t-u4g	6596
jax/experimental/sparse/bcoo.py	sha256=WYM3Pj5u4kbC3l-9y0cQMJWkDEYD4A0Mhru6uBnReNo	129445
jax/experimental/sparse/bcsr.py	sha256=4mT2jU1EKlftB2x44kzLI8Wz1MUQ5d5qVQSsoFxKQoA	38298
jax/experimental/sparse/coo.py	sha256=2bBCeU2NXmlp7rsGsOQvn--yTrYCbya_ss0K7IjpCsg	24143
jax/experimental/sparse/csr.py	sha256=bBNOl1KR8WwAW0MtiBNKfebV3W2qTW58mA7lsH6Kotg	24396
jax/experimental/sparse/linalg.py	sha256=6fbBA79nY9PASO-E2GIv1s4uD0yxBDwy3uSkMN-huRw	23707
jax/experimental/sparse/nm.py	sha256=ypXA8WaigylNbaRqJRlfXyurJC00f41VF1jxqUAiI3I	8568
jax/experimental/sparse/random.py	sha256=702kXl0gkbcSBthBBbyD3JAUinP_PlRDPCLMZHCdgcA	3882
jax/experimental/sparse/test_util.py	sha256=4JlLtN_KEl_bzF-ts9KRSbMNbzUsjEB_6IyQAhw7WQA	9428
jax/experimental/sparse/transform.py	sha256=_PFhqqW_ZjrOrpvsKO1ZEnvp9iuNASAa56JwGcxqzac	37468
jax/experimental/sparse/util.py	sha256=li_VLh2WCr728hXZ18xVRDpL3qo6tpen-LR7CvruB9I	4332
jax/extend/__init__.py	sha256=cL0QjoXSmMpM_j6_QBlKN6dtl30zMbvHqYA3eoHsd4g	1630
jax/extend/backend.py	sha256=ODjEVf5_5U37PZk5WVp8wI8thqOZO8XAdRV6Rortr8E	1054
jax/extend/ffi.py	sha256=31OhFUi5t8Msov6VnqpTj8U8bpDe9V746tMb1enGh9w	1783
jax/extend/ifrt_programs.py	sha256=oCXLRZL4g3Syl8tQsTnjQNw8AL0zfKBlhUIUQ6Fh9gc	801
jax/extend/linear_util.py	sha256=xag9NUqPndR7mh3NDQ4TluFpc6hFFd4r7Q5LarAFqvY	1592
jax/extend/random.py	sha256=w1OM6RCcGwmzIr9bdGJJTNKIcMO4ypXbzmeEORlBHjQ	1083
jax/extend/source_info_util.py	sha256=IxZYjLDlwOH1uE3A_vc25dy25QZ6FABMb4RqrkiA5hU	1215
jax/extend/core/__init__.py	sha256=o_q4S8jW_Tc0986SMqpjcifigZq-Ez2OhFZo6aBLwzM	1036
jax/extend/core/primitives.py	sha256=Ja7N3kkrO4Qf1HOj5_KSLy9GUfgQfmpthYNEvdaKGGA	6422
jax/extend/mlir/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/extend/mlir/ir.py	sha256=1IirtLROpBL2nVDNSlP_3c6u09xXaKLqBkzF1JtmGz4	631
jax/extend/mlir/passmanager.py	sha256=1rfuCocMdFlx6It95ynBCP-8JC5xEAEG7rBdLJ7SpFI	640
jax/extend/mlir/dialects/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/extend/mlir/dialects/arith.py	sha256=Au6mT9C4OrdpaGGwJ4_bauTFjGFQKF7wilv62mxzP1E	643
jax/extend/mlir/dialects/builtin.py	sha256=Y1QDRes-Arh7uBLJpySSqygBwUovalHyBq71m3uPbiQ	645
jax/extend/mlir/dialects/chlo.py	sha256=sfxvYEEBgGVd8fDXOlyfGtOLCCAW0xquLrSlTJFA81A	642
jax/extend/mlir/dialects/func.py	sha256=sCnRWz5gKtZ8Cf5NpeCIQ0zVh-nM6H95ttjzksTRrXk	642
jax/extend/mlir/dialects/math.py	sha256=G2gFkarEChMUEjv-vjDc4O09e-1orPhh_zHJhjbPPRQ	642
jax/extend/mlir/dialects/memref.py	sha256=nPR3oOrsENZ-mHjHKFSD48PpQMZHsVLfUVCwomBhv_w	644
jax/extend/mlir/dialects/scf.py	sha256=hmcJIF7RUqzeeP_NIOampnvCB5Jb6T8Xex4T03IeKrI	641
jax/extend/mlir/dialects/sdy.py	sha256=rtDlK5Z2OMDjBduAaRn08XBH-yGdZkvZogSB7RoFxBc	746
jax/extend/mlir/dialects/sparse_tensor.py	sha256=TaFY58y273rRITr8WxSM7LB1-sTAa03dCAjH4F8ZWaE	651
jax/extend/mlir/dialects/stablehlo.py	sha256=AbJQc7zghgrJh-BgRB31L7iHrNLVaEk4AELejcq7QiE	647
jax/extend/mlir/dialects/vector.py	sha256=xP03nScJg-t7olD8_P8SiFGVnV6Qi0Fwn9uGK0MHOXI	644
jax/image/__init__.py	sha256=cthOdx-HsZ6SgvW19tdNCDHmC4Proc8da1hY1hRztTI	1027
jax/interpreters/__init__.py	sha256=HNgKpgKpz4svFyTlHE9wMrDp5iQtg6XeeUshYBOnRko	690
jax/interpreters/ad.py	sha256=p0gJFj9VvrsZbZavI-xg00Y-PzEu5Rw_6d3upq5u92Y	2759
jax/interpreters/batching.py	sha256=QqKOP4uX4F1PSf96ZrXi6x3DOIrU8wyWqXGTlt1nC2g	2574
jax/interpreters/mlir.py	sha256=J0KWo8KT-F0EsG8612vktotwDMPJYcd9CPxhgVucRkU	3059
jax/interpreters/partial_eval.py	sha256=Jp-bZC7rkH_O_I6gCn2GhvGvgL2d-Wrhe0jlFasjYcw	3954
jax/interpreters/pxla.py	sha256=VN0Mlb4IreJ9Tqez8i-QWn1EXVcrDWTLPeioTkeL-qE	1801
jax/interpreters/xla.py	sha256=4s2i2F0NnxUevbTeW3BHHM7Nz7yHV7Wkjx9lXUGEIcA	2038
jax/lax/__init__.py	sha256=dAMLafErY2qooz_kITLvhqdPeb9JksCeYrBzREyxZGE	11036
jax/lax/linalg.py	sha256=lpjXQ-7jLL0pTG5IzCz5Z0Usf2l4B5yeuNgieGHjrW8	1624
jax/lib/__init__.py	sha256=wy3yxhlsb9k2th9Dfsr6eS59L2qLIxH_IoTM4cPfx3M	774
jax/lib/xla_bridge.py	sha256=SrvAZ1XaonXyv1MmJv6JT4_eg9q5aa8Gk8Gj2qwH24w	1557
jax/lib/xla_client.py	sha256=jvsBGY7rUMEOSsv10CS8Ehh4NX7zqS4A1ROOUyRXULc	3893
jax/lib/xla_extension.py	sha256=xuSR42eY6codcchsSZx8rJZ2MdokxbJDNZwW9Xldz6o	1942
jax/nn/__init__.py	sha256=4LyqLzosWMtCv_zCd-9Gt9XmfCohbduCQrN_cRPqvVg	1562
jax/nn/initializers.py	sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g	1469
jax/numpy/__init__.py	sha256=V3uu1A7EYHf850z51JQ5e020Z3Ze--VH3TlAm7aqKko	12820
jax/numpy/__init__.pyi	sha256=DlhNI6uoM7fCa1snACp4I3pVY-RNwYtRsrbvw4AE8Gk	43688
jax/numpy/fft.py	sha256=iNatBIYRSwpXquPSgtNRMe6-dhbKkv2yoH6UJ6PFoGM	1084
jax/numpy/linalg.py	sha256=bnzLmLpLYSDRIICkjCEsH1WmA7wNOdSKhmO25gPvTqI	1419
jax/ops/__init__.py	sha256=VIAczIKYAFWawHdNmLmptUYnFQiBwHwlsa4LpgGEbIc	870
jax/scipy/__init__.py	sha256=DXszF0t3ZbfDeoFEP_GyBb0OOr6F-4yqsZzmen7pY58	1474
jax/scipy/fft.py	sha256=PVKQ4r-vl4J9Ct6LrhZWZ7jMF7frIXggiO98Bh0oVIE	809
jax/scipy/integrate.py	sha256=H9iq7YKgsZQ1d0SUvGcgPnEgD1TBdWJXHYjfUOGzT8Y	777
jax/scipy/linalg.py	sha256=jLoDaaxGZTcRu94dbVgk7qa4Nm-CHu_L8rMsvEIwzWI	1350
jax/scipy/ndimage.py	sha256=ei-ZfRiDXrlY4nYC1SFL7nnGHFrC_emD_NAicGhH4vU	788
jax/scipy/signal.py	sha256=nW4h017jShZBLL3J2NMhLwlOv46wK1MhEPeKlPTj710	975
jax/scipy/special.py	sha256=u7tHwfOThzXNgH7N0wSI_ApWNhifGfa9i4BukWG4Wq8	2386
jax/scipy/cluster/__init__.py	sha256=sgIbEEx62OubzDO36E3EkqjbYY_3qsErPOTQE6lsj5E	768
jax/scipy/cluster/vq.py	sha256=lkLSlkWFi6TLjwuzTX7hVSiC1IkCYd3OWgn8YtIAJd0	776
jax/scipy/interpolate/__init__.py	sha256=RI5oUXPC-8tKmYURnnrnA-e1LDbnNVOVVqkGSpagEBM	760
jax/scipy/optimize/__init__.py	sha256=cAw1rT1e0Mu-22vBUuJb6Q-CQF8fbIUBlKYpynB7e2s	822
jax/scipy/sparse/__init__.py	sha256=ugltlihEjwgorQxgfziHC-03UeYy4JzdQmTxrt1cnTo	757
jax/scipy/sparse/linalg.py	sha256=wpjjS1G1UNISyhZVFoJMMgIkqs-L8ASdvtQAbvEBsuk	810
jax/scipy/spatial/__init__.py	sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc	581
jax/scipy/spatial/transform.py	sha256=cm7lKi8aUepI3cOwTnjrldhFh7hBv89CR_m3pFj3r2o	802
jax/scipy/stats/__init__.py	sha256=ssiwAm2AdZ81MPMM11015yMNogj4DGsNfWfq507yhlo	1995
jax/scipy/stats/bernoulli.py	sha256=llkwAlvFH0qNU1BHdxn9CXXOBAPg0ffnbW_WQQ4LKVg	819
jax/scipy/stats/beta.py	sha256=5UIf6_JtqVZfAl1dDc2N3K8SkG0OV1bBkC3vWpY8-Hs	851
jax/scipy/stats/betabinom.py	sha256=5T6965_jHQxG4oR2YocPQRKcDo5AGxkhIyrk2iz-Kk0	792
jax/scipy/stats/binom.py	sha256=XGO5jDk3nRRgGwHQU6KEl9_kQ5X2qF-2TLgvm3ZUvvs	659
jax/scipy/stats/cauchy.py	sha256=MfAMnK9eG2wcutA5FVAyhU6iVYSSNgzI1XFT1xtNIfg	881
jax/scipy/stats/chi2.py	sha256=f-6i-0twBiOrak26tc7bEnWgOODA3iFwEFgi_An9udo	851
jax/scipy/stats/dirichlet.py	sha256=FUHSEmXKmVUneXuxgaIZM28s6GxlusbymdRYEPnDiUE	792
jax/scipy/stats/expon.py	sha256=YClw5PcqAXgq3VD4ukZ19HfTmRCMtrGk6E2BpXb6U9E	866
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/colab_tpu.py	sha256=IPEQ1HZB4xW0MRSPGDeFAw9GoMFaphNQ0kLRIwGypmo	890
jax/tools/jax_to_ir.py	sha256=Z8LkCMaXfqjYFSHPSTS3L5imebmYtV3NPLy7V1WMLPU	8776
jax/tools/pgo_nsys_converter.py	sha256=BCxXmHnvcorM62adEbOFNYcsnTyv-Hkqnf4K7lo_wt4	3413
jax-0.5.2.dist-info/AUTHORS	sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY	313
jax-0.5.2.dist-info/LICENSE	sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA	11358
jax-0.5.2.dist-info/METADATA	sha256=-k-NtDKq60ORUhAaPCyfpCOoR_H8h6jGxNVm2l8pUow	22850
jax-0.5.2.dist-info/WHEEL	sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM	91
jax-0.5.2.dist-info/top_level.txt	sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE	4
jax-0.5.2.dist-info/RECORD	—	—

top_level.txt