Research engineer at @prime working on distributed/decentralized training