8 Commits

Author	SHA1	Message	Date
Woosuk Kwon	90bab18f24	[TPU] Use mark_dynamic to reduce compilation time (#7340)	2024-08-10 18:12:22 -07:00
Woosuk Kwon	73388c07a4	[TPU] Fix dockerfile.tpu (#7331)	2024-08-08 20:24:58 -07:00
Earthwalker	7f8d612d24	[TPU] Support tensor parallelism in async llm engine (#6891)	2024-07-29 12:42:21 -07:00
Woosuk Kwon	fad5576c58	[TPU] Reduce compilation time & Upgrade PyTorch XLA version (#6856)	2024-07-27 10:28:33 -07:00
Woosuk Kwon	c467dff24f	[Hardware][TPU] Support MoE with Pallas GMM kernel (#6457)	2024-07-16 09:56:28 -07:00
Woosuk Kwon	4552e37b55	[CI/Build][TPU] Add TPU CI test (#6277) ... Co-authored-by: kevin <kevin@anyscale.com>	2024-07-15 14:31:16 -07:00
Woosuk Kwon	08c5bdecae	[Bugfix][TPU] Fix outlines installation in TPU Dockerfile (#6256)	2024-07-09 02:56:06 -07:00
Woosuk Kwon	1a8bfd92d5	[Hardware] Initial TPU integration (#5292)	2024-06-12 11:53:03 -07:00