Woosuk Kwon b3856bef7d [Misc] Use torch.compile for GemmaRMSNorm (#7642)

2024-08-22 01:14:13 -07:00

..

guided_decoding

Support for guided decoding for offline LLM (#6878)

2024-08-04 03:12:09 +00:00

layers

[Misc] Use torch.compile for GemmaRMSNorm (#7642)

2024-08-22 01:14:13 -07:00

model_loader

Revert "[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)" (#7764)

2024-08-22 03:42:14 +00:00

models

Revert "[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)" (#7764)

2024-08-22 03:42:14 +00:00

__init__.py

[Performance] Optimize e2e overheads: Reduce python allocations (#7162)

2024-08-08 21:34:28 -07:00

custom_op.py

[XPU] fallback to native implementation for xpu custom op (#7670)

2024-08-20 00:26:09 -07:00

parameter.py

[Misc] Update gptq_marlin to use new vLLMParameters (#7281)

2024-08-13 14:30:11 -04:00

pooling_metadata.py

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

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

sampling_metadata.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -07:00

utils.py

[Hardware][Neuron] Refactor neuron support (#3471)

2024-03-22 01:22:17 +00:00