Robert Shaw 18fecc3559 [ Kernel ] Fp8 Channelwise Weight Support (#6487)

2024-07-18 03:18:13 +00:00

..

fused_moe

[Hardware][TPU] Support MoE with Pallas GMM kernel (#6457)

2024-07-16 09:56:28 -07:00

ops

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

quantization

[ Kernel ] Fp8 Channelwise Weight Support (#6487)

2024-07-18 03:18:13 +00:00

__init__.py

Change the name to vLLM (#150)

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

activation.py

[Kernel][CPU] Add Quick gelu to CPU (#5717)

2024-06-21 06:39:40 +00:00

layernorm.py

[Model] Add Gemma 2 (#5908)

2024-06-27 13:33:56 -07:00

linear.py

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

logits_processor.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00

pooler.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

rejection_sampler.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

rotary_embedding.py

Support Deepseek-V2 (#4650)

2024-06-28 13:24:57 -07:00

sampler.py

[Core] draft_model_runner: Implement prepare_inputs on GPU for advance_step (#6338)

2024-07-17 14:30:28 -07:00

spec_decode_base_sampler.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

typical_acceptance_sampler.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00

vocab_parallel_embedding.py

[CORE] Quantized lm-head Framework (#4442)

2024-07-02 22:25:17 +00:00