jax/__init__.py,sha256=OTKbaYeT-ekG1YMHKMjPBBI50IrTFVt4dSPIgrQWE2M,10192
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=yHNh-x_OytwKQ9Oj5MSCqMAzmPhogJK9wLMUjw5radU,9446
jax/sharding.py,sha256=16l0TCfLgpzmf0IwPvlBzfsOi3yzYJRv9rJp82P2sLI,1721
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=xPy7SdLMkwyHNHFvF8RpaaEwng7uFly5Q8-Aw0mYMvE,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=uRzcjs3OHuOT96OJq58_CeWq6-B59b1WRmKBDQDwpe8,39710
jax/_src/ad_util.py,sha256=n14jlpCC3f7itcVEL8ijSM_C4ofyB8wbKKTzwWzxbr0,4164
jax/_src/api.py,sha256=6nv2gtKhzXQwHCNKmJJ840SU-YNMUae30rYdbAbe4qY,125063
jax/_src/api_util.py,sha256=O3yJ8XZzN_f9OX-pgmCCtJjRTF1wrKMdLEamUwTvdbM,28268
jax/_src/array.py,sha256=e-d61KyrxljTcoRgI4m6nYCqQ8xjfS77wpDT5ZfJ1Nk,49323
jax/_src/basearray.py,sha256=12hYSzFDsq0LVpPX8nQ-gH5mLE253PBWRSnYvQsuVwQ,6503
jax/_src/basearray.pyi,sha256=D9Q7W2_DF0LkP5BpXRFZROcV2VzNBBgAMCo_QLax2ZE,13146
jax/_src/blocked_sampler.py,sha256=OtV8uaYBfdGsWOpYD6LG9JPK1yknKkrJuLPAJ_LJPeM,5915
jax/_src/cache_key.py,sha256=8ytVe0xpceRekMKyEUxC0T3GaEKAv_JgreT7xPuCB5M,12883
jax/_src/callback.py,sha256=sdJxpUp2grl0Gus6quVAvERsYaQRpq0a_g8HbleUTzA,33775
jax/_src/checkify.py,sha256=xTGl8P5OZk21qJ3rW-Yi_b-uLaH6f_v45pf1zhxlqXU,57396
jax/_src/cloud_tpu_init.py,sha256=nHNo6ZZV6r3mvv8vCDp6Dzva6bhirmN-MXoAh8u8Dns,4807
jax/_src/compilation_cache.py,sha256=gWbFDIUV4sVCpk6QEdaTJmKRdvww9YOo2t2nfhhSzRs,11940
jax/_src/compilation_cache_interface.py,sha256=ONHvsIC3Kzoda0TcQWnhhjxtOWYixjkO9Or__BAW3_0,865
jax/_src/compiler.py,sha256=lyYHVH4nxDq6gziqNM73_lquUqapxd4oQgIDLDmcfVY,29520
jax/_src/compute_on.py,sha256=0b74TSV1R31kJTR53thafdtB0sYiFGqmRoAbZwAFcps,1911
jax/_src/config.py,sha256=uTY24NxEOUpsvLNOi0ghx6q0Yqxqy9ScNYqmSLYJvCc,69339
jax/_src/core.py,sha256=Fix1n2WfOApe7IGl0dV27daOE8WKopjoZLswGq6CrPo,119532
jax/_src/custom_api_util.py,sha256=uv6uyrXU1BHGzKuAe498W2PD9bcWt1PEa-8UAqr9Zaw,876
jax/_src/custom_batching.py,sha256=wK4iEi0Fk8SL_Le3oenJwzBUO6vTmfJRwqiX4z0gTA0,15177
jax/_src/custom_dce.py,sha256=XkVPIEBiiZ05Z4BIC8Ajbt6nmkE8qIAccDTLQ24sU2s,17478
jax/_src/custom_derivatives.py,sha256=HIhJbppP5JCYyH_KHUSgeb3FmlpvxC-wNAoQNtvrWDE,78126
jax/_src/custom_partitioning.py,sha256=4mjU8OlTsNRjwPCT8pgz9FyPTfIpAEzT70SttW1fdt8,25737
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=dXz95VFUkKTFuPg3xn-QHwEVupRnuEypxigcdwYNTLc,27792
jax/_src/deprecations.py,sha256=FO0YxHbuhn-h2kFyzUH3Aj93PdQUSFqWyMjFRuyyFiQ,4784
jax/_src/dispatch.py,sha256=pxVg1YgVyXjZrd5Kk2dXY5eeTpWwRc5agGqElfscc3Q,25276
jax/_src/distributed.py,sha256=NoqKqUg-KUD1JVBSZdsLLfdTjPcKFcCvHHVf75rpD0k,12768
jax/_src/dlpack.py,sha256=2krUhEhrQI0qOb40T0dlcRAaQjtMGOUgi8Ss0JtOJ_U,11410
jax/_src/dtypes.py,sha256=US75MBDvPauGmaF0gvSaQqoAm28IwKjVYXNMwCFad3U,37298
jax/_src/earray.py,sha256=6KsYjNLzWaT1S0IKDwmZMAsa94E8yg9TU2Yb3-2QN70,4412
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=1aoniSHk_3TGUTyg3prsNafXVUyFmyi0ivZRK_YcWLI,5869
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=0-8OyFvMYu5l-C0e3VObpQMkLGg-BBtzP56CAdFEIxE,2295
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=IFmyAx2kRqjVBCfNDb5Syhz8HoqIP5XPDg-y01Bp6Ig,18503
jax/_src/logging_config.py,sha256=zKijn3gjyw21ljvWjTsBX1bE6AyIa3zHRQ15YvaDDmk,4452
jax/_src/lru_cache.py,sha256=NejEgo8BmO7xy8uORQrAHsN0HpCnY8qsEUnuEhDCkPs,6741
jax/_src/mesh.py,sha256=6m9mkhv7G3W5hvB51WZFBZU1n7xHtOK5MA7Z3RGnCE0,18700
jax/_src/mesh_utils.py,sha256=3FsEOtQfebQX5VjUAVeT8JxNVT8MoJndG37to4JGbpM,32336
jax/_src/monitoring.py,sha256=Z5ol4haHZeaj11ZHKFNgBOUXXSjeOHYmg-xS9ZNQpZw,5188
jax/_src/named_sharding.py,sha256=GgD0efCpVlfErkYjNTEfoqg7G9HZfPRM_wy9csmmqlk,21723
jax/_src/op_shardings.py,sha256=nRnGgfaKz7mIB7sHxh_QkiCa1oPMA9VsDpPyQkivtlU,3857
jax/_src/partition_spec.py,sha256=hNaAcVufeEvxjxxQsnw8gvU7UOI8yCK7vWFyMHMlm7E,2785
jax/_src/path.py,sha256=VkrwcBMWagZZQBBQYhdSFr7pQzoxwBnNUEfKfEbN-nU,1596
jax/_src/pickle_util.py,sha256=gE8IrNWtg4hfu84T5rQ2DVAdpozh5xwSzAlRi7UIWcQ,2312
jax/_src/pjit.py,sha256=1k8DpdoPEZk7PODgiTjPcKIxqiVNadwgbWE8JFvvTLo,130917
jax/_src/pretty_printer.py,sha256=iRZHOftfSZUYXoFmBxLWM58CXEuwvlDVhJk0Njhc_QY,15460
jax/_src/prng.py,sha256=vI4-GnKzaNTHFZBBAkvHfYX598CwY6fVJxkkDXb0luw,45147
jax/_src/profiler.py,sha256=7WtOfvJ46GGqgPwfHHGu_I5dXuX-rkyUNmkhkAm2jkk,16768
jax/_src/public_test_util.py,sha256=6kBfGDhMAWm5grEDdQ-gREvWOETA0nSuzARd3pssowc,13241
jax/_src/random.py,sha256=v4XoxJHYfZYOjbqV7T0XpG9Y_h7L2F4zYtVV8zLkMNo,104889
jax/_src/shard_alike.py,sha256=wG_kUvyH6aNRjgViwVffEAz1tbfhBF66kf3vw0XexIg,3989
jax/_src/sharding.py,sha256=rCq32k0vOyt75z0GLtFYN3RTRrrVhntd9bqvvkyHLd8,8165
jax/_src/sharding_impls.py,sha256=REwInqyh4Z9QYDbaKL5AhGf8GFG486MezKbDl9tY9Ds,53502
jax/_src/sharding_specs.py,sha256=UCYUIcQfS6lcA993W33zZI5tfAg5zcefP9cZUyW9NoI,8617
jax/_src/source_info_util.py,sha256=O8Ff0QonDyYKecTH8a2KdqBe7HuQHrIDTkb-OShHnpk,10852
jax/_src/sourcemap.py,sha256=5s4zTThhX6PZwgQ4v4UbYL4norg4uHipqgUGdHBn76Y,6640
jax/_src/stages.py,sha256=87dxrFZE2pOUsyMxtpGjE-S2AXmIs05zYAfWtlR-1wM,29048
jax/_src/test_util.py,sha256=oynyLPBu-YhuqCn-PzEEDwi-sawKm9_O66MCbUe4mN0,85806
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=xSOQ3HQbvE-zFEOmx6FXhnEOpUvrW36D1pTi03utmFg,21179
jax/_src/xla_bridge.py,sha256=-iq__WOwFRa-njYQTeu15UqwiaD1htVhV7vWcHcup-Y,47139
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=ZVfq6whNkDmnPd1N-cUdfIREkAkqGpKyeUcTZpestWw,71837
jax/_src/cudnn/fusion.py,sha256=3ifqe61DuTyoOXQL7Nelacz2KRdcJdPRI__ub4s_vI8,2808
jax/_src/cudnn/scaled_matmul_stablehlo.py,sha256=OCZrOeV2idoRfyURTufw00SZLzNkUC7cmuCXvlMnhK4,23286
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=-EChG8sl5LE3kKAwqjAVOnK-gYIWlLDMYIj_d3KjoCc,69426
jax/_src/export/serialization.py,sha256=lrYYbwYZV1z5Pg5YGoXGICtiiE0L6tffh_GgiS-Ri-0,19457
jax/_src/export/serialization_generated.py,sha256=48WdUMEmJTc04nWhHvJGFWPEyXe2vx6kNkC-1p4ndlg,27811
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/interpreters/__init__.py,sha256=hnuhN7h1jY5__Kt2rBvTrk0o8nX4fQl244fDCIMh0qk,690
jax/_src/interpreters/ad.py,sha256=J5MC7IgQ0Uc5lYJq7biPdTDa5B980c-HP-EZpF4M7Q4,55631
jax/_src/interpreters/batching.py,sha256=ctezIPsdMuMS1JRNdUBT1QenLzO8y2tO5A_nMX1N7wo,50002
jax/_src/interpreters/mlir.py,sha256=NvaRt_o6dSHtjgjP9RCWP37s40LtWBzkGZuo-_UIWPg,126367
jax/_src/interpreters/partial_eval.py,sha256=2jrThFip5nuAgDBAxb5Vv7MBwEMN2V7O9X8giaROso4,116120
jax/_src/interpreters/pxla.py,sha256=T1x4PJCvBVkxGjFJbyWkLhjDAlrNXkG7SgaVCFN4y9k,135409
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=H4efmu6VZM7WIctXAA9aajf_bglZEx5dCPXNX4y9S8k,341517
jax/_src/lax/linalg.py,sha256=R9TElylfLfk1uACUoKsWYfFdwZ02M0cwjQf8CeykPo0,104721
jax/_src/lax/other.py,sha256=k5QaTGae3ps89MM2UzOG1_foHg_W3C5Ysq3BMKJWdyc,12497
jax/_src/lax/parallel.py,sha256=aZsbnojvYecJxwiYe7h42Zx_rjr7_YU2V3aeaTTTEVo,78503
jax/_src/lax/qdwh.py,sha256=dmslOVnzg-DhxjSIHV9jHCOKopVosYD-xe1FPmSgnCo,9391
jax/_src/lax/slicing.py,sha256=-a_4yrSp6tkLL4zZuYJ6JL3wWHBkcx9m1ten0R4XwtA,133065
jax/_src/lax/special.py,sha256=dPQDWjoYTPy8DnGlglYL_JhMc9c51-ZJ5C1PvdbgW3Y,27247
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=s37gkemCrCy_96bCpBMEhs2UwV6ExFdV-3wulwzKCqs,7031
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=75qJrTdI-QLmKGcjkC1T8cECWKgHYQDtzX1UOHXMRe0,11565
jax/_src/lax/control_flow/conditionals.py,sha256=0LDvAORybpF5SwdAsO4FJQbX6Pt-NsreioUieDg_eeI,44082
jax/_src/lax/control_flow/for_loop.py,sha256=9L4Zv3LTKKCW3mlce9cFZRrK6zYb5UIiBWNdgyUy9-A,36370
jax/_src/lax/control_flow/loops.py,sha256=HCoP8y7cydXfDPXBWEk_hJwPsRVSQyLv3rEDsbV92aA,114010
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=LoyCZJaPgryOXO9QhPSDyNyJy0x8v3Z-wHUKuUVFMC8,41766
jax/_src/nn/initializers.py,sha256=QtFWAlSp4ljUWjsqmNXYdjVIQmW91Fk3EOBaF2K9QN0,25341
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=tya3wnoBIkI45gj0OOHN3IJx2fikfrjDFXb7sc_WTFo,23410
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=JwLWvr73IOhJzL2VLuTJa2y8SX3DDjMp4i6PkpNe0dk,50883
jax/_src/numpy/lax_numpy.py,sha256=pcZSpAS9Q_7KOjFqo3Uvg0zDv8EE0F-02eymf7GHjGo,358687
jax/_src/numpy/linalg.py,sha256=ViYABXPkqPzNb5ZYGSK274tRUAj8yOdWCWx9CHGSQGM,78248
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=cdnoLqrIyEZEcp8JR-9GtV6cwm9tJauFWVyTkv-gzOI,4016
jax/_src/numpy/setops.py,sha256=QUbc1j-EMKEcK79Ron__r7zA0P87hoSr1SmMrcWB1m4,47867
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=JMVnPBn585gC7jIoSNOrcFx_h5T0lQ5a5JOtQoiI-IE,16111
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=gFop2N7kIF0Z46VMMDRsMRcg7odeYONt7Xelw5raz4E,4244
jax/_src/pallas/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/core.py,sha256=DWS3fVeWWZZITaSsKMntWXoRHfE4bar6Gx6RPbH1aJY,39433
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=CLHfSX7iIChICjyHcUfuTXhJVTvkMHnFmeWWNdww2cE,18953
jax/_src/pallas/pallas_call.py,sha256=j22uGiexAam1bU2VifmiONKlsJfSxfmHVzU3JStRMwU,67068
jax/_src/pallas/primitives.py,sha256=-8XKbto7_3BwxAzPsSVNNvlXk5rkZqJKIELpLUZxGDM,34093
jax/_src/pallas/utils.py,sha256=7Tor_59JAGvLDRbFMMLCqIcDQ-8y8UOrSjVlwND5kKw,13887
jax/_src/pallas/fuser/__init__.py,sha256=WeYuHahJepC-QEZ0ttZV531UBPOB-9DQJHpNjSBsb28,1182
jax/_src/pallas/fuser/block_spec.py,sha256=2MXtBmreMBpzuVWfLTca_sWo9U38UXDzNDM1klHo4s4,47935
jax/_src/pallas/fuser/custom_evaluate.py,sha256=jv5JjL5BlUUK89ZKx9sufff0bVyqJA3s5ag6Bpus5ys,2823
jax/_src/pallas/fuser/fusable.py,sha256=gPgIWbEQ7ldYa-ZZIW953PDqxpdZg7VptxFwOnkoFLA,2487
jax/_src/pallas/fuser/fusable_dtype.py,sha256=Le1G258yoGwUTxEUhl3TNVLKZc8FvmAWOzU0LRczXG0,14361
jax/_src/pallas/fuser/fuser_utils.py,sha256=UY_9Ly9Z863mHX3R_EdLw4H1jLsTCeaeTDMV1Dg9l-s,1273
jax/_src/pallas/fuser/fusion.py,sha256=W1NBIzSqOkBk8ADty9HfjnmePZXKBsPxFUiM_k3qtC8,1488
jax/_src/pallas/fuser/jaxpr_fusion.py,sha256=lmPSWuNxVtgVSr5tFr1r2NywbeWxUUUH6iXuGdh_9dM,5249
jax/_src/pallas/mosaic/__init__.py,sha256=t5wRc_z8eIDwV0votvqn8WtOUEd53ZNf_J4VRYLJepc,581
jax/_src/pallas/mosaic/core.py,sha256=qIoE2266MSqyzCWQsXOISlHiSGx6Cimh60Rge7EHG7g,9149
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=XnnAa03HMY05WVfh3DbO0LmfmMP-qR7braMumGerc6k,59955
jax/_src/pallas/mosaic/lowering.py,sha256=zLsLXICWZKzH4y-n_hmDscCAKCPjshKDsl3E0f4g2Do,135089
jax/_src/pallas/mosaic/pallas_call_registration.py,sha256=_bOH6AftHEsn5-ajdtXv7vDG74aj3kiaQJTCZz0FwKU,9387
jax/_src/pallas/mosaic/pipeline.py,sha256=tqF96rTDRqT4gvY8dU_MQgQoPZjjqNGvuOnpH2GJ1Lo,47487
jax/_src/pallas/mosaic/primitives.py,sha256=YEmMvIjsOM5c7VcwhOkWmABpBxdLRx9cro6RQ9JmfKk,31905
jax/_src/pallas/mosaic/random.py,sha256=Dya2ppf5XvHF263ZBvYNwc1Y5C7jWA30jfAGCiykEqc,8460
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=K6O8jjPet3qoI8cap6PVKdcPYJmXG3Ww5jzdDILI53U,18770
jax/_src/pallas/mosaic_gpu/lowering.py,sha256=DKiQlhb-Q-G7Kq2a7mO6zjgldNvAwDdLuS890qh2A6Q,81688
jax/_src/pallas/mosaic_gpu/pallas_call_registration.py,sha256=0J_mYHXAIj4K-QKov3xSoTZI06gcl7QdNNGhygMRjCs,3818
jax/_src/pallas/mosaic_gpu/pipeline.py,sha256=9m47NAeToeKiEKX64EHjhS3eYhYwbqz9wbaEIWBS_as,25060
jax/_src/pallas/mosaic_gpu/primitives.py,sha256=3Sa1uU6YhfJvuRApWxAOvXuQgPLwVe67tDemUh032OI,33430
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=bOG12XAP7ge1HI9NVN7KQczSlGJtchfeu4y8PyeurNs,95201
jax/_src/pallas/triton/pallas_call_registration.py,sha256=Pwdik0DdNkBZ0a_sdwUZNULPQNtmfbW_GdYsTnFwSnE,6768
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=Zk6rMFtDXTDME86XLitZvE366y7rMw4AE0YOFwkrdss,77810
jax/_src/scipy/ndimage.py,sha256=IuE4PMzY5HuxST9D0G0On1SVFRpNF367Y2cIQv8duTc,7092
jax/_src/scipy/signal.py,sha256=E01k5pOeAtg2has0eb8GRoaogdfBWNYFzhcCdHt7dAs,47453
jax/_src/scipy/special.py,sha256=BzI93hH5u6F_g1-jw12ZGw5IVV2GUBDv5w4wXkeyzJg,82605
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=Nf0qgFD7K-2H4R581g1Kb0RQ_WHZrl81A1vgyS_PdxM,7038
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=rpqhet2XWQttwnixuCiZBOvw93MpCnFbjQJbkM_tbms,45559
jax/_src/state/indexing.py,sha256=ptIrT7WCeiA6IVEIPQGwr9QOKe30Mxi-10MfFbEhUf8,10563
jax/_src/state/primitives.py,sha256=xHCoto0hNROzHdHjs0hu0fVSQOgAEEoqBpv2j8hW9dk,25788
jax/_src/state/types.py,sha256=n0LZDi-7f1nm5I2LtKTdF9iL86_ECkcm9RXfeBd6810,13887
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=YBlmf1imil4GCcP0PBF7PdsAhLaAaOjbqfA-AcUMUFk,27252
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=TPgTtVTgePcDRx2xOnDWyr7o8F3K7P_DxVXtKqt9CdU,100742
jax/experimental/topologies.py,sha256=Lxxb2W6nT6AEhM2ekkNUKm6W1tgh5i-_49454NuxCkQ,2101
jax/experimental/transfer.py,sha256=E8dFkOjF0jnpEYk6--VEdnozMEcC-ntdSG-GkKln9eI,2327
jax/experimental/x64_context.py,sha256=lUrmtrYhu0ZGwd_j8vrzFqQnwfDptgpHxZbqV94Hb30,1710
jax/experimental/xla_metadata.py,sha256=YoNAzPbdARYfGzGTUVoxTfHG9dDrobUvWaPI4uvHRJk,658
jax/experimental/_private_mm/__init__.py,sha256=VIjw6l_CQ5-xaicoK8-2QgM5W7GYrBvv4xASfTB0ehU,775
jax/experimental/_private_mm/mini_dime.py,sha256=cEFmbyq5mMdOBXergCqdrqoh7WMSNgi0OINZKt4zbzE,9736
jax/experimental/_private_mm/mm.py,sha256=sI1wL8y9YpZRN5IgZd0-wX_ho2K5uP2-Ivrze3PfckM,11426
jax/experimental/_private_mm/profile_utils.py,sha256=dZ1wvrhpaRNbEyiU7FpIuIi6qUfOvBHdMOcXqsYNRrg,2169
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=PZnhfzgIMLlay2bPih6fpF-r9XvTdVx66U4IlTC7nWY,24237
jax/experimental/colocated_python/__init__.py,sha256=Cp1CgJT_Sn5qLkiO7jW5S0HPWieBmOdpEE-AaYpKs9E,980
jax/experimental/colocated_python/api.py,sha256=UdsHC3bp63LQxDOdC3jPKHa-BnEegbwX_UAAZvWGWlY,2307
jax/experimental/colocated_python/func.py,sha256=8nb9UB39UUexX7wNVvfir5TbWUTRgzkZinry8McQqrk,17437
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=ZTsGn5ueJ9UNvKNuay-NkMHsY4aD4ssl1UyThkh7044,8338
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=6AP4RMS7rPfav4rmXaJnrt3rLl90OULtEpghzLZeOCU,28081
jax/experimental/jax2tf/impl_no_xla.py,sha256=7ztEeEhJycNm_yA263efIAa-FsuYeOI2cvvgrqmmgNY,54478
jax/experimental/jax2tf/jax2tf.py,sha256=XjV0sX6xsc9C94mJ29DGxsKqW6SwmvEDr-I3tYpQ2ow,145130
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=20ZieTGRBNlpld8UXZu61BX905Q5TPhS3SB0RD-WySQ,66913
jax/experimental/jax2tf/tests/control_flow_ops_test.py,sha256=NlaRUQa0B43nQtJ71gRX3E16QYwXjwgCaxVEeDyFZoI,10041
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=8kvyo8TjjTyIe8steWsggUlQ-OyRuh9QBP8Likg3uSI,60794
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=QkF2fxh1gzd0nkAroynwLyKr3CbRiXBA1ZzzBkyGrr4,14643
jax/experimental/jax2tf/tests/savedmodel_test.py,sha256=Ch1luggaR0F9_jRSLAovuqRSCnVw3EIUmYBYfKKJRE0,13067
jax/experimental/jax2tf/tests/shape_poly_test.py,sha256=whsGtGDjM9UzOd3UZM4fpC_qyhJXuPKepE1eSChU-Do,126488
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=Rjh2P-IIJf6F73C_iXG7kAiXpccRxPUWwprSkRXYFkM,2919
jax/experimental/mosaic/gpu/core.py,sha256=fDxSYhJHKpfZQlGRILHMsdtPyiv9mFehnbzKhPA7S_8,25562
jax/experimental/mosaic/gpu/dialect_lowering.py,sha256=C8tEYgesYco_kKz9sLiXVnvYb6ACm0X__KRa_FqYiSc,35985
jax/experimental/mosaic/gpu/fragmented_array.py,sha256=XgfnMzYlYjIWlSJFNsX-s3bwiRHIVdJi4wxa4RLjRho,99846
jax/experimental/mosaic/gpu/inference_utils.py,sha256=iJLnSlnI1E1BkfhKdxI-BNOr4oymgKi3fKkefdaF0cE,7052
jax/experimental/mosaic/gpu/launch_context.py,sha256=1JtAO3CzJBfP-xpu_pJ5CFaO8q5QkW-Djfcw7wQGtqk,27605
jax/experimental/mosaic/gpu/layout_inference.py,sha256=jX_7bvIn5yNU0xyUQoSAVdkysAFTC_TIUb_5Z1Ec068,17301
jax/experimental/mosaic/gpu/layouts.py,sha256=Ikn2bUFZIb_Ku4wgGgaPNPQwvthdmI6z8WEBqJkOTVY,6411
jax/experimental/mosaic/gpu/mma_utils.py,sha256=_Aoyrzdo3FWifCBUsSrlKMGyJ5NfxhLL01o3jlF1WGc,8664
jax/experimental/mosaic/gpu/profiler.py,sha256=0u-Os0I2OFyB2O7OoWlkxrwyzXiH8rj4rT2q1w0rVpA,14632
jax/experimental/mosaic/gpu/tcgen05.py,sha256=rd4TxgLjfYQLOXiGadd2Xi5jgDyJKsGZmSItRsp9CYI,22157
jax/experimental/mosaic/gpu/transform_inference.py,sha256=mOOOU9ALM30RmYNQtbXgUCP84EkSNUREy3OeUeX0K3w,10511
jax/experimental/mosaic/gpu/utils.py,sha256=Mq_VenHDw366x-XILN1Zd1Wy51vLmef4YnvuPrzcR3w,42781
jax/experimental/mosaic/gpu/wgmma.py,sha256=KIkEwHi8gqdlMOgxg0pfQXjCwAp384xrW90E45aMDRE,15159
jax/experimental/pallas/__init__.py,sha256=Ehc7toK4u5j50yBzcbf8jZ9o33C2_-Hp4oNqBvFkvm0,3591
jax/experimental/pallas/fuser.py,sha256=4P2m-h9UDjddPkGL-kYIYTrfMveANo5BLijTtzYM1jM,1211
jax/experimental/pallas/gpu.py,sha256=Tswdk79SmB-Otqbp0s-FLoyQmMGPrrVQHHUNUBDWT_I,865
jax/experimental/pallas/mosaic_gpu.py,sha256=59oELiEXXl-SxtNeNO3oVWDR25m1_rKER5ifvutZyNs,3182
jax/experimental/pallas/tpu.py,sha256=mGAiLffeD0GfP5mBeWkzLyEXTXRKI8usqhdoGF6itgk,4263
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=cuzBkk0M6p6hrKOUjtXOAAKH5Lu6PLWpE5pFyl-f2sI,17086
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/ragged_paged_attention.py,sha256=s-zdh-kqLcSAWqs6e06nE6kCRO9xoTfVKBBtQyxH9_c,25799
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=rCP4uS9rufa71F3kt2jAvcUaMDlTt021f3lpiohyFgs,74456
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=eryt0VhACHNSaZwMH8KCNNGWa5jwfjZGMuAZ4wqJrBM,10747
jax/experimental/roofline/rooflines.py,sha256=4p15pBmacsARcDHvwmxNcATV5Y8eU5bVu7qL5lLabPU,13828
jax/experimental/source_mapper/__init__.py,sha256=g5b_71QFD84RscRTmCDm7l_TfUYkcW6gZuRd8rgFg9Q,1698
jax/experimental/source_mapper/common.py,sha256=2wh5nZMvCdhjDBQKw539R5WTnmFJWNzMBRgaZxqI__s,2481
jax/experimental/source_mapper/generate_map.py,sha256=rCr8YPEZI1OAjw_GZFUndUkhgVVS7gcv6dDVszTey3w,2102
jax/experimental/source_mapper/hlo.py,sha256=ItS2ytK1D4cBQOZsap0UzyWA-saoI_T_fWH2DVoMSFA,4112
jax/experimental/source_mapper/jaxpr.py,sha256=PXa9zMzP2_fD_0vxo8Za-mw21pXp2Aao6IQkQIqmoJU,2595
jax/experimental/source_mapper/mlir.py,sha256=U2YSWC9tp_DlnsrRzuzsAU7W2PY6ZwtWedHiuMSTQ5M,4686
jax/experimental/sparse/__init__.py,sha256=iYfDK0sdq3ErijcE7Va7cRjEHjzuJ3ykCwnHSYxNMPs,11186
jax/experimental/sparse/_base.py,sha256=fXN4gSMRuo1CizZweJDMa_vZvcrJcSfYzlbzysRvWG8,3556
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=m_dBKU7VkkzdEjOROd1mXOjkf5uvt4IYP1Sovq2hG-k,37479
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=koxOYt-TU0_YrfsQyfAC456oqDCE4gtjeKrX1Dwip9w,11138
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=6WS091xJMPX_fjRFDblfN2eza5dX72Kvq9RWQITdAc0,1640
jax/nn/initializers.py,sha256=finfAJmOEZdPIzY2roSIBTOmC4rKEW-rchpH9d6Dc2g,1469
jax/numpy/__init__.py,sha256=B2iz_xUo7bcDf_KJOWE3GbAjpwvv6FSYDfMcFFHpiRU,12887
jax/numpy/__init__.pyi,sha256=l9tgnzbkP78itIdsAYGuMVKPkIljDRTRgxOUzS8-7d0,43781
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.3.dist-info/AUTHORS,sha256=OOBQygrI1Zc94AYJycH1r_BSCwP38Hyv3sludltJ7hY,313
jax-0.5.3.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
jax-0.5.3.dist-info/METADATA,sha256=wUNQxQ7M05JT7qdez5FYqAZ-E-JXbHRPau1rV_zSg9s,22833
jax-0.5.3.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
jax-0.5.3.dist-info/top_level.txt,sha256=DabAh8MXQ-HM0EtjKD-Wx_D6tTIW-_de73pdG_f_KTE,4
jax-0.5.3.dist-info/RECORD,,
