Xiaoyu Zhang 4373df5525 add flashinfer mxfp4 (#8847)

2025-08-06 16:23:41 -07:00

..

configs

Support OCP MXFP4 quantization on AMD GPUs (#8255)

2025-08-04 18:14:52 -07:00

connector

feat(remote_model): support variable remote backend for model loader (#3964)

2025-03-14 00:40:44 -07:00

constrained

bugfix: Fix XGrammar backend to use model's EOS tokens for constrained generation (#8422)

2025-07-28 10:01:02 +08:00

debug_utils

Minor tool for comparison of benchmark results (#7974)

2025-07-27 00:27:50 -07:00

disaggregation

[PD] Refactor parallel sizes and add pp support for mooncake (#8571)

2025-08-04 20:18:11 -07:00

distributed

Add support for NCCL symmetric memory for TP allreduces (#8238)

2025-08-01 23:30:55 +00:00

entrypoints

Support v1/responses and use harmony in serving_chat (#8837)

2025-08-06 16:20:34 -07:00

eplb

[2/3] Optimize Slime Update Weights: Avoid GPU-to-CPU Device Sync when update expert weights (#8753)

2025-08-05 22:09:52 -07:00

function_call

Support v1/responses and use harmony in serving_chat (#8837)

2025-08-06 16:20:34 -07:00

layers

add flashinfer mxfp4 (#8847)

2025-08-06 16:23:41 -07:00

lora

[Fix] Fix several issues preventing gemma3n LoRA support. (#8776)

2025-08-04 17:11:46 -07:00

managers

Support v1/responses and use harmony in serving_chat (#8837)

2025-08-06 16:20:34 -07:00

mem_cache

fix args typo in memory_pool_host (#8662)

2025-08-03 13:47:29 -07:00

metrics

feat: add production metric for retracted requests due to insufficient kvcache (#7030)

2025-07-17 11:59:05 -07:00

model_executor

Use reduce scatter for DP (#8539)

2025-08-06 16:21:26 -07:00

model_loader

Support OCP MXFP4 quantization on AMD GPUs (#8255)

2025-08-04 18:14:52 -07:00

models

Use reduce scatter for DP (#8539)

2025-08-06 16:21:26 -07:00

multimodal

feat: Support DP Attention for step3_vl (#8699)

2025-08-03 19:35:26 +08:00

sampling

fix: prevent crashes due to logit bias dimension mismatch (#7685)

2025-07-23 15:30:55 -07:00

speculative

[CUDA Graph] save cuda graph memory by using next_token_logits_buffer (#8579)

2025-08-03 03:06:47 -07:00

weight_sync

Add tensor.detach() back to update weight util (#8691)

2025-08-02 00:41:05 -07:00

_custom_ops.py

[Feature] Integrate quick allreduce and select the best allreduce implementation (#6619)

2025-07-24 20:48:42 -07:00

aio_rwlock.py

A better aio rwlock that guarantees the order (#2547)

2024-12-22 15:44:32 -08:00

code_completion_parser.py

feat(oai refactor): Replace openai_api with entrypoints/openai (#7351)

2025-06-21 13:21:06 -07:00

constants.py

Multi-Stage Awake: Support Resume and Pause KV Cache and Weights separately (#7099)

2025-06-19 00:56:37 -07:00

conversation.py

[feat] Add detail in image_data (#8596)

2025-08-05 14:01:38 +08:00

custom_op.py

[1/N] MoE Refactor: refactor select_experts (#7966)

2025-07-19 00:51:15 -07:00

hf_transformers_utils.py

model: adapt mllama4 to VisionAttention (#8512)

2025-08-02 00:39:40 -07:00

jinja_template_utils.py

[feat] Add detail in image_data (#8596)

2025-08-05 14:01:38 +08:00

model_parallel.py

Make torch TP composable with torchao (#2436)

2024-12-11 04:21:42 -08:00

operations_strategy.py

[5/N] MoE Refactor: Update MoE parallelism arguments (#8658)

2025-08-01 01:20:03 -07:00

operations.py

Support overlapping two batches (#4068)

2025-05-24 17:39:07 -07:00

patch_torch.py

Fix torch.compile cacheing (#5259)

2025-04-10 18:08:45 -07:00

poll_based_barrier.py

Support colocating requests (#7973)

2025-07-28 22:51:49 -07:00

reasoning_parser.py

model: support Step3V (#8583)

2025-07-31 02:41:00 -07:00

server_args.py

add flashinfer mxfp4 (#8847)

2025-08-06 16:23:41 -07:00

torch_memory_saver_adapter.py

Multi-Stage Awake: Support Resume and Pause KV Cache and Weights separately (#7099)

2025-06-19 00:56:37 -07:00

two_batch_overlap.py

[Feature] improve TBO: two chunk overlap (#8144)

2025-08-05 21:11:01 -07:00

utils.py

Support v1/responses and use harmony in serving_chat (#8837)

2025-08-06 16:20:34 -07:00

warmup.py

Improve streaming, log_level, memory report, weight loading, and benchmark script (#7632)

2025-06-29 23:16:19 -07:00