R0CKSTAR d3cdee0a04 [MUSA][4/N] Add common device utilities, distributed backend, and custom op wiring (#17246) ...

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

2026-01-28 23:13:24 -08:00

..

attention

Move fa4 from sgl-kernel to jit kernel (#17353)

2026-01-24 15:25:03 +08:00

lora

[diffusion] fix: fix the LoRA weights mismatch caused by weights packing (#17355)

2026-01-21 18:17:16 +08:00

quantization

WIP: initial multimodal-gen support (#12484)

2025-11-05 12:28:52 -08:00

__init__.py

WIP: initial multimodal-gen support (#12484)

2025-11-05 12:28:52 -08:00

activation.py

[diffusion] perf: support FFN pack gate and up proj for Z-Image(#15201)

2025-12-16 01:18:47 +08:00

custom_op.py

[MUSA][4/N] Add common device utilities, distributed backend, and custom op wiring (#17246)

2026-01-28 23:13:24 -08:00

elementwise.py

[diffusion] perf: apply mul add fusion for Qwen-Image (#16299)

2026-01-28 09:40:13 +08:00

layernorm.py

[diffusion] perf: apply mul add fusion for Qwen-Image (#16299)

2026-01-28 09:40:13 +08:00

linear.py

WIP: initial multimodal-gen support (#12484)

2025-11-05 12:28:52 -08:00

mlp.py

[diffusion] feat: support TP for Flux.1.dev (#15666)

2025-12-25 18:00:44 +08:00

rotary_embedding.py

[diffusion] model: LTX-2 (1/2) (#17495)

2026-01-24 11:59:48 +08:00

triton_ops.py

[diffusion] perf: apply mul add fusion for Qwen-Image (#16299)

2026-01-28 09:40:13 +08:00

usp.py

[diffusion] comment: fix typo (#16257)

2026-01-03 11:42:04 +08:00

utils.py

[diffusion] chore: clean srt imports (#17252)

2026-01-17 15:47:49 +08:00

visual_embedding.py

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

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

vocab_parallel_embedding.py

WIP: initial multimodal-gen support (#12484)

2025-11-05 12:28:52 -08:00