Zijie Tian 965c8aff12 [WIP] need change flashattention to debug.

2026-01-01 00:58:22 +08:00

..

activation.py

fix

2025-06-15 13:28:29 +08:00

attention.py

[WIP] need change flashattention to debug.

2026-01-01 00:58:22 +08:00

embed_head.py

simplify

2025-08-31 20:02:51 +08:00

layernorm.py

[refactor] Translate into english, void Chinese due to claude.

2025-12-11 00:30:24 +08:00

linear.py

simplify

2025-08-31 20:02:51 +08:00

rotary_embedding.py

simplify

2025-08-31 20:02:51 +08:00

sampler.py

[feat] Added bench_offload.py and GreedySampler.

2025-12-12 00:24:08 +08:00