Staff ML Research Engineer, Pegasus - Training Performance