Siyuan Liu 9112b443a0

[Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011) ...

Signed-off-by: Siyuan Liu <lsiyuan@google.com>
Co-authored-by: Hossein Sarshar <hossein.sarshar@gmail.com>
Co-authored-by: Chengji Yao <chengjiyao@google.com>

2025-06-03 00:06:20 +00:00

..

worker

[Bugfix][TPU] Fix tpu model runner testcase failure (#18810)

2025-05-30 18:04:03 +08:00

__init__.py

[V1] TPU - Add tensor parallel support via Ray (#13618)

2025-03-08 08:19:38 -05:00

test_basic.py

[TPU] Enable gemma3-27b with TP>1 on multi-chips. (#17335)

2025-05-05 14:19:58 -07:00

test_mha_attn.py

[TPU][V1] MHA Pallas backend (#15288)

2025-03-21 08:50:39 -07:00

test_multimodal.py

[Frontend] Add missing chat templates for various MLLMs (#17758)

2025-05-07 00:10:01 -07:00

test_pallas.py

[TPU] Update PyTorch/XLA (#16288)

2025-04-09 14:46:32 +08:00

test_perf.py

[TPU][V1][CI] Update regression test baseline for v6 CI (#17064)

2025-04-30 04:03:22 -07:00

test_sampler.py

[TPU] Fix the test_sampler (#17820)

2025-05-08 05:51:33 -04:00

test_spmd_model_weight_loading.py

[Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011)

2025-06-03 00:06:20 +00:00

test_topk_topp_sampler.py

[V1][TPU] Enable Top K (#15489)

2025-04-17 18:18:11 +00:00

test_tpu_qkv_linear.py

[Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011)

2025-06-03 00:06:20 +00:00