Hyesoo Yang 47195057e9

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

Signed-off-by: Hyesoo Yang <hyeygit@gmail.com>

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

..

attention

[V1] Scheduler Refactoring [1/N] - Add Scheduler Interface (#15250)

2025-03-20 17:50:43 -07:00

core

[V1] Scheduler Refactoring [1/N] - Add Scheduler Interface (#15250)

2025-03-20 17:50:43 -07:00

engine

[V1] Scheduler Refactoring [1/N] - Add Scheduler Interface (#15250)

2025-03-20 17:50:43 -07:00

executor

[Attention] Flash Attention 3 - fp8 (#14570)

2025-03-20 01:14:20 -04:00

metrics

simple bugfix: Update stats.py (#15139)

2025-03-19 18:26:27 +00:00

sample

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

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

spec_decode

[V1][Spec Decode] Optimize Rejection Sampler with Triton Kernels (#14930)

2025-03-18 14:31:54 -07:00

stats

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

structured_output

[BugFix] Lazily import XgrammarBackend to avoid early cuda init (#15171)

2025-03-20 01:30:43 +00:00

worker

[V1] Scheduler Refactoring [1/N] - Add Scheduler Interface (#15250)

2025-03-20 17:50:43 -07:00

__init__.py

[V1] AsyncLLM Implementation (#9826)

2024-11-11 23:05:38 +00:00

kv_cache_interface.py

[Bugfix][V1] Handle MLA in kv_cache_interface (#14462)

2025-03-07 22:18:25 -08:00

outputs.py

[V1][Spec Decode] Optimize Rejection Sampler with Triton Kernels (#14930)

2025-03-18 14:31:54 -07:00

request.py

[V1][Minor] Minor code cleanup for scheduling metrics (#14800)

2025-03-14 08:21:28 +00:00

serial_utils.py

[V1] Use msgpack for core request serialization (#12918)

2025-02-10 11:35:56 +08:00

utils.py

[V1][Minor] Add __repr__ to ConstantList (#14907)

2025-03-16 20:20:15 -07:00