Back to results

Machine Learning Infrastructure Researcher

  • Toronto, ON, Canada

Applications open in a new tab.

Job details

Experience
PhD in Computer Science, Electrical Engineering, or a related field and at least 10 years of industry or research experience in the field.
Posting language
English
Posted

Original job description

Lightmatter is leading the revolution in AI data center infrastructure, enabling the next giant leaps in human progress. The company invented the world’s first 3D-stacked photonics engine, Passage™, capable of connecting thousands to millions of processors at the speed of light in extreme-scale data centers for the most advanced AI and HPC workloads.

Lightmatter raised $400 million in its Series D round, reaching a valuation of $4.4 billion. We will continue to accelerate the development of data center photonics and grow every department at Lightmatter!

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.

Lightmatter is (re)inventing the future of computing with light!

Lightmatter is seeking a Principal of Machine Learning Infrastructure Researcher to join our Machine Learning team. This team is focused on inventing novel hardware systems using Lightmatter technology and modeling the performance of Lightmatter-enabled systems on both training and inference workloads.

This is primarily a creative role where you need to think outside the box to push the boundaries of systems research with the power of silicon photonics at your fingertips. This role will work closely with our product, solutions architecture, and engineering teams to generate impactful performance analysis, create high-performance rack-scale designs, and publish cutting-edge ML systems 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

Qualifications:

  • 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 Qualifications:

  • 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

  • Comprehensive Health Care Plan (Medical, Dental & Vision)
  • Retirement Savings Matching Program
  • Life Insurance (Basic, Voluntary & AD&D)
  • Generous Time Off (Vacation, Sick & Public Holidays)
  • Paid Family Leave
  • Short Term & Long Term Disability
  • Training & Development
  • Commuter Benefits
  • Flexible, hybrid workplace model
  • Equity grants (applicable to full-time employees)

Benefits eligibility may vary depending on your employment status and location. Lightmatter recruits, employs, trains, compensates, and promotes regardless of race, religion, color, national origin, sex, disability, age, veteran status, and other protected status as required by applicable law.

Export Control

Candidates should have capacity to comply with the federally mandated requirements of U.S. export control laws.

More about this job

Responsibilities

The role involves inventing novel hardware systems using Lightmatter technology and modeling the performance of these systems for AI training and inference workloads. You will also lead a team of machine learning engineers and collaborate with product and architecture teams to design reference platforms.

Requirements

Candidates must hold a PhD in Computer Science, Electrical Engineering, or a related field with at least 10 years of industry or research experience. A strong background in machine learning systems research, performance modeling, and experience with large language model frameworks is required.

Skills

  • Machine Learning
  • Systems Research
  • Performance Modeling
  • Deep Learning
  • Large Language Models
  • Silicon Photonics
  • AI Infrastructure
  • Performance Evaluation
  • Compute Architectures
  • Networking Technologies
  • GPU
  • TPU
  • Ethernet
  • Infiniband
  • NVLink
  • Team Leadership

Benefits

  • Comprehensive Health Care Plan
  • Life Insurance
  • Generous Time Off
  • Paid Family Leave
  • Short Term Disability
  • Long Term Disability
  • Training & Development
  • Commuter Benefits
  • Flexible, Hybrid Workplace Model
  • Equity Grants

Visa sponsorship

Not detected in the job text

Categories

  • Technology
  • Engineering
  • Science & Research
  • Software
  • Data & Analytics

Keywords

  • Machine Learning
  • Infrastructure
  • Photonics
  • Passage
  • AI
  • HPC
  • Silicon Photonics
  • Performance Analysis
  • Rack-scale Design
  • Deep Learning
  • Large Language Models
  • GPU
  • TPU
  • Ethernet
  • Infiniband
  • NVLink
  • Systems Research
  • Performance Modeling
  • Academic Research
  • Whitepapers
  • Team Leadership
  • Computer Science
  • Electrical Engineering
  • Data Center
  • Inference
  • Training

Report this job