search

pytorch / torchtitan

A native PyTorch Library for large model training
BSD 3-Clause "New" or "Revised" License
1.28k stars 115 forks source link

use local_map to annotate fusedrmsnorm #346

Closed wanchaol closed 2 weeks ago