Machine Learning Infrastructure Researcher

Skills

About the Role

You will research machine learning systems and model the performance of photonics-enabled hardware for training and inference workloads. You will develop rack-scale reference designs, collaborate on product and architecture decisions, communicate technical findings, publish research, and lead a small group of machine learning engineers.

Requirements

• PhD in Computer Science, Electrical Engineering, or a related field
• At least 10 years of industry or research experience
• Machine learning systems research, performance evaluation, and performance modeling experience
• Hands-on machine learning and deep learning experience
• Extensive knowledge of large language model training and inference frameworks
• Relevant academic publications
• Knowledge of AI compute and networking technologies, including GPUs, TPUs, Ethernet, InfiniBand, and NVLink
• Experience leading a team to deliver published results
• Capacity to comply with U.S. export control laws

Responsibilities

• Stay current on machine learning research advancements
• Build tools that model system performance for AI training and inference workloads
• Design customer reference platforms with product and solutions architecture partners
• Provide machine learning expertise
• Publish academic papers, whitepapers, blog posts, and marketing collateral
• Lead a small team of machine learning engineers

Benefits

• Comprehensive medical, dental, and vision coverage
• Retirement savings matching program
• Basic, voluntary, and AD&D life insurance
• Vacation, sick time, and public holidays
• Paid family leave
• Short-term and long-term disability coverage
• Commuter benefits
• Flexible hybrid workplace model
• Equity grants for eligible full-time employees