Lianmin Zheng 64f80eabbe Register aten::rms_norm and aten::mm.dtype in batch invariant mode (#24459)

2026-05-05 17:21:34 -07:00

..

__init__.py

Add Batch‑Invariant RMSNorm (#12144)

2025-10-28 21:05:57 -07:00

batch_invariant_ops.py

Register aten::rms_norm and aten::mm.dtype in batch invariant mode (#24459)

2026-05-05 17:21:34 -07:00