Add EddyFormer#1237

Open

mrlazy1708 wants to merge 12 commits intoNVIDIA:mainfrom

mrlazy1708:main

Commits on Nov 20, 2025

implement EddyFormer
Yiheng Du
committed

Commits on Nov 21, 2025

fix format issue
Yiheng Du
committed
verify rope dimension
Yiheng Du
committed

Commits on Nov 23, 2025

fix device and docstring
Yiheng Du
committed
fix import and remove comments
Yiheng Du
committed

Commits on Nov 24, 2025

Commits on Dec 8, 2025

add amp
Yiheng Du
committed

Commits on Dec 9, 2025

implement C0-Legendre SEM basis
Yiheng Du
committed
fix C0 Legendre SEM basis
Yiheng Du
committed

Commits on Dec 16, 2025

add test every 100 steps; add torch.compile; support amp training
Yiheng Du
committed