Principal Machine Learning Infrastructure Researcher

Mountain View, CAfull timeOn Site

Apply on Employer Site

senior

$300k–$350k

H1B: yes

Lightmatter Principal Machine Learning Infrastructure Researcher at Lightmatter leading a small team to invent novel hardware systems and model performance of Lightmatter-enabled systems for AI training and inference workloads. Collaborate with product and engineering teams and publish research.

Responsibilities

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.
Lead a small team of Machine Learning engineers

Qualification

PhD in Computer ScienceStrong knowledge of AI systemsAbility to work across researchAdaptable and versatile

Required

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, and trade-offs.
Proven track record leading a team to deliver published results.

Preferred

Demonstrated ability to creatively problem-solve and apply first-principles thinking to generate novel ideas, particularly in ambiguous, evolving environments.
Exceptional skills in communicating complex research outcomes and technical concepts to partners and internal stakeholders, while embodying our value of speaking truth.
Ability to work across research, product, and business functions, rolling up your sleeves to bridge gaps and accelerate outcomes, while demonstrating the core value of "We are one Lightmatter."
Adaptable and versatile, thriving in a high-speed, high-precision environment.
Experience in managing engineers and leading projects, with the ability to lift up the team, and work with high standards and high collaboration.

Benefits

Dental
Vision
Equity/RSU
Parental Leave
Life Insurance

Company

Lightmatter

View company page

Source: greenhouse

Active within the past hour

More roles at Lightmatter

Chip Firmware Validation Engineer

Mountain View, CA

Signal Integrity Engineer

Mountain View, CA

Package Layout Engineer

Mountain View, CA

Power Integrity Engineer

Mountain View, CA