DarkSharpnessandXiaoyu Zhang 2dd9196079 [JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880) ...

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>

2026-03-20 18:24:07 +08:00

..

diffusion

[Diffusion] Add diffusion time embedding to jit kernel (#17658)

2026-01-24 14:27:08 +08:00

distributed

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

elementwise

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

fast-hadamard-transform

Remove unused fast-hadamard-transform PyTorch extension sources (#18927)

2026-02-18 15:51:07 +08:00

gemm

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

lora

[LoRA][I] Add MOE LoRA JIT alignment kernel and tests (#19710)

2026-03-12 12:23:46 -07:00

moe

[JIT Kernel] Reland NVFP4 kernels to JIT (#20012)

2026-03-07 10:31:08 +08:00

nsa

[DeepSeek-V3.2][JIT-kernel] Support nsa fuse store indexer k cache (#19148)

2026-02-26 10:23:10 +08:00

add_constant.cuh

[Refactor] Clean up JIT kernel utilites (#16884)

2026-01-13 17:54:16 +08:00

hicache.cuh

[Refactor] Fix test and clean up hicache code (#18555)

2026-02-18 14:37:46 +08:00

ngram_embedding.cuh

Feature/support longcat flash lite (#17838)

2026-03-09 23:00:11 +08:00