Michael Goin 3b17ea26e4 [TPU] Re-enable the Pallas MoE kernel (#18025) ...

Signed-off-by: Michael Goin <mgoin64@gmail.com>

2025-05-20 19:52:27 -07:00

..

fused_moe

[TPU] Re-enable the Pallas MoE kernel (#18025)

2025-05-20 19:52:27 -07:00

mamba

fix: typos (#18151)

2025-05-15 02:16:15 -07:00

quantization

[Minor] Rename quantization nvfp4 to modelopt_fp4 (#18356)

2025-05-20 09:08:37 -07:00

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

activation.py

[New Model]: support GTE NewModel (#17986)

2025-05-14 01:31:31 -07:00

layernorm.py

Update deprecated type hinting in model_executor/layers (#18056)

2025-05-13 04:17:23 -07:00

lightning_attn.py

[Bugfix] Fix triton import with local TritonPlaceholder (#17446)

2025-05-06 17:53:09 +08:00

linear.py

[Bugfix] Fix QKVCrossParallelLinear::sync_weight_attrs for PyTorch compile (#17844)

2025-05-14 09:39:51 +00:00

logits_processor.py

[Misc] Slight spelling modification (#18039)

2025-05-12 20:36:27 -07:00

pooler.py

[Model] Use sigmoid for single-label classification (#18313)

2025-05-18 07:00:09 -07:00

rejection_sampler.py

[Sampler] Adapt to FlashInfer 0.2.3 sampler API (#15777)

2025-05-16 15:14:03 -07:00

resampler.py

Update deprecated type hinting in model_executor/layers (#18056)

2025-05-13 04:17:23 -07:00

rotary_embedding.py

[New Model]: support GTE NewModel (#17986)

2025-05-14 01:31:31 -07:00

sampler.py

[Sampler] Adapt to FlashInfer 0.2.3 sampler API (#15777)

2025-05-16 15:14:03 -07:00

spec_decode_base_sampler.py

Update deprecated type hinting in model_executor/layers (#18056)

2025-05-13 04:17:23 -07:00

typical_acceptance_sampler.py

Add full API docs and improve the UX of navigating them (#17485)

2025-05-03 19:42:43 -07:00

utils.py

Update deprecated type hinting in model_executor/layers (#18056)

2025-05-13 04:17:23 -07:00

vocab_parallel_embedding.py

Update deprecated type hinting in model_executor/layers (#18056)

2025-05-13 04:17:23 -07:00