Nicolò Lucchesi cfbb8c930f

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

Signed-off-by: NickLucche <nlucches@redhat.com>

2025-03-21 08:50:39 -07: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