Principal Machine Learning Infrastructure Researcher

Job summary

Responsibilities

If you're passionate about tackling complex challenges, making an impact, and being an expert in your craft, join our team of brilliant scientists, engineers, and accomplished industry leaders.

• Stay up to date on the latest advancements in Machine Learning research.
• Build tools to model the performance of Passage-enabled systems when running both training and inference workloads for the latest AI models.
• Collaborate with our product and solutions architecture teams to design compelling reference platforms for our customers, powered by Lightmatter technology.
• Function as the Machine Learning expert in the room.
• Publish academic papers at top industry conferences and journals. Help support the publication of whitepapers, blog posts, and other marketing collateral.

What they're looking for

• PhD in Computer Science, Electrical Engineering, or a related field and at least 10 years of industry or research experience in the field.
• Solid background in Machine Learning systems research and experience with performance evaluation and performance modeling.
• Hands-on experience with machine learning / deep learning methods and extensive knowledge of Large Language Model training and inference frameworks.
• Academic publications in relevant journals and industry conferences.
• Strong knowledge of AI systems, including compute (GPUs, TPUs, etc.) and networking (Ethernet, Infiniband, NVLink, etc.) technologies, and their various topologies, paradigms…