Join as a Staff Software Engineer, focusing on optimizing distributed training systems and enhancing ML infrastructure.