Senior Software Engineer

About the role

• Lead a team of engineers focused on enabling new machine learning customers on a specialized SDK and platform.
• Own the customer onboarding journey from model evaluation through production training at scale.
• Drive engineering initiatives to maximize Model FLOPS Utilization (MFU) for customer workloads through performance analysis, profiling, and tuning tools.
• Build and maintain tooling, automation, and documentation that accelerates time-to-first-training for new customer models.
• Partner with compiler, runtime, and framework teams to identify and resolve blockers in customer workloads.
• Develop scalable processes for distributed training enablement.

What You’ll Need

• Experience working with PyTorch or JAX software.
• 3+ years of engineering team management experience.
• 7+ years of experience working directly within engineering teams.
• 3+ years of designing or architecting new and existing systems.
• Experience partnering with product or program management teams.
• 3+ years of experience in deep learning/machine learning, including model training workflows.
• Experience with distributed training at scale.

What’s On Offer

• Competitive annual salary ($212,700 – $287,700) plus potential for sign-on payments and restricted stock units.
• Comprehensive benefits package including health insurance, 401(k) matching, paid time off, and parental leave.
• Opportunities for mentorship and career growth within a dedicated team.
• A culture that values work-life balance and inclusive team dynamics.