Lianmin Zheng 76e4a8662c Replace clamp_position with JIT kernel + platform dispatch (#20999) ...

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

2026-03-21 21:26:26 -07: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 clamp_position with JIT kernel + platform dispatch (#20999)

2026-03-21 21:26:26 -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