Software Engineer, Infrastructure

The MRS ML Infra team is focusing on ML Infra performance and efficiency for both large scale AI training and inference workflows in the recommendation domain. In this role, you will work on optimizing the e2e stack for model training and inference for large scale recommendation models, with opportunities coming from the domains of distributed systems, model/system co-design, GPU optimizations, and more. While the core of day-to-day work and key responsibility will be to identify and lead the execution for short/mid term opportunities for efficiency optimization, you will also drive long term strategies and shape team direction on things like model/system co-design, performance automation, regression detection and mitigation, etc.