Principal Machine Learning Infrastructure Researcher (AI)

Location: Mountain View, California, United States; flexible hybrid workplace model.

Salary: $300,000–$350,000 USD annual on-target earnings.

Company

Develops photonics-based data center infrastructure for extreme-scale AI and high-performance computing workloads, including 3D-stacked optical interconnect technology.

What you will do

• Track advances in machine learning research and act as the machine learning subject-matter expert.
• Build tools to model the performance of photonics-enabled systems across training and inference workloads for current AI models.
• Design customer reference platforms and high-performance rack-scale systems with product and solutions architecture teams.
• Generate performance analyses and collaborate with research, product, solutions architecture, and engineering functions.
• Publish research papers at leading conferences and journals, and contribute to whitepapers and technical content.
• Lead a small team of machine learning engineers and deliver published research results.

Requirements

• PhD in computer science, electrical engineering, or a related field, plus at least 10 years of industry or research experience.
• Strong background in machine learning systems research, performance evaluation, and performance modeling.
• Hands-on experience with machine learning and deep learning methods, including extensive knowledge of large language model training and inference frameworks.
• Academic publications in relevant journals and industry conferences.
• Strong knowledge of AI compute and networking technologies, including GPUs, TPUs, Ethernet, InfiniBand, and NVLink, as well as their topologies and trade-offs.
• Proven experience leading a team to deliver published results.

Nice to have

• Creative problem-solving and first-principles thinking in ambiguous and evolving environments.
• Ability to communicate complex research outcomes and technical concepts to internal and external partners.
• Experience working across research, product, and business functions.
• Experience managing engineers and leading projects in high-speed, high-precision environments.

Culture & Benefits

• Flexible hybrid workplace model.
• Medical, dental, and vision coverage.
• Retirement savings matching program.
• Vacation, sick leave, public holidays, and paid family leave.
• Life insurance, disability coverage, training and development, commuter benefits, and equity grants for full-time employees.