yarongmu-google 0a049c7d86 [CI/Build] Add tests for the V1 tpu_model_runner. (#14843) ...

Signed-off-by: Yarong Mu <ymu@google.com>

2025-03-25 12:27:16 -04:00

..

worker

[CI/Build] Add tests for the V1 tpu_model_runner. (#14843)

2025-03-25 12:27:16 -04:00

__init__.py

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

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

test_basic.py

[V1] TPU - CI/CD use smaller model (#15054)

2025-03-18 21:39:21 +00:00

test_mha_attn.py

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

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

test_sampler.py

[V1][TPU] Speed up top-k on TPU by using torch.topk (#15242)

2025-03-20 19:19:40 -07:00