Skip to content

use torch.gather() to simplify the process of broadcasting a token embedding to an atom embedding and gathering the frame atom coordinates#269

Closed
OccupyMars2030 wants to merge 4 commits into
bytedance:mainfrom
OccupyMars2030:use-torch-gather
Closed

use torch.gather() to simplify the process of broadcasting a token embedding to an atom embedding and gathering the frame atom coordinates#269
OccupyMars2030 wants to merge 4 commits into
bytedance:mainfrom
OccupyMars2030:use-torch-gather

Commits

Commits on Mar 14, 2026

Commits on Mar 15, 2026

Commits on Mar 16, 2026