3 people 166f4e2dc3 fix: fuse weightless RMSNorms at their declared width (#50867) ...

Signed-off-by: Anuj Bolewar <anujbolewar@gmail.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Anuj Bolewar <anujbolewar@gmail.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

2026-08-04 16:03:09 +00:00

..

__init__.py

Support MLA properly in the Transformers modeling backend (#48250)

2026-08-04 13:24:11 +01:00

base.py

Fix MQA with tensor parallelism on transformers modeling backend (#49987)

2026-07-27 22:51:17 +00:00

glu.py

Improve Transformers modelling backend fx tracer (#49957)

2026-07-27 12:51:16 +01:00

mla.py

Support MLA properly in the Transformers modeling backend (#48250)

2026-08-04 13:24:11 +01:00

moe.py

Fix MLA padding and grouped topk routing in the Transformers modelling backend (#49982)

2026-07-27 18:32:32 +00:00

packed_qkv.py

Fix MQA with tensor parallelism on transformers modeling backend (#49987)

2026-07-27 22:51:17 +00:00

qkv.py

Fix MQA with tensor parallelism on transformers modeling backend (#49987)

2026-07-27 22:51:17 +00:00

rms_norm.py

fix: fuse weightless RMSNorms at their declared width (#50867)

2026-08-04 16:03:09 +00:00