Demo

Machine Learning Engineer, Distributed Training Infrastructure

Twelve Labs Inc.
San Francisco, CA Full Time
POSTED ON 1/16/2025
AVAILABLE BEFORE 3/24/2025

Who we are

At Twelve Labs, we are pioneering the development of cutting-edge multimodal foundation models that have the ability to comprehend videos just like humans do. Our models have redefined the standards in video-language modeling, empowering us with more intuitive and far-reaching capabilities, and fundamentally transforming the way we interact with and analyze various forms of media.

With a remarkable $107 million in Seed and Series A funding, our company is backed by top-tier venture capital firms such as NVIDIA’s NVentures, NEA, Radical Ventures, and Index Ventures, and prominent AI visionaries and founders such as Fei-Fei Li, Silvio Savarese, Alexandr Wang and more. Headquartered in San Francisco, with an influential APAC presence in Seoul, our global footprint underscores our commitment to driving worldwide innovation.

We are a global company that values the uniqueness of each person’s journey. It is the differences in our cultural, educational, and life experiences that allow us to constantly challenge the status quo. We are looking for individuals who are motivated by our mission and eager to make an impact as we push the bounds of technology to transform the world. Join us as we revolutionize video understanding and multimodal AI.

About the role

As Machine Learning Engineer, Distributed Training Infrastructure, you will be responsible for ensuring that compute performance and ease-of-use never delay our research timeline. You will own strategy and implementation for all compute & training infrastructure optimization, observability, scaling, and orchestration. You will collaborate closely with other engineers and scientists to define and implement your chosen roadmap. This role is a perfect fit for research-minded compute specialists who want to build SOTA video, vision, and video-language modeling systems!

In this role, you will :

  • Own our compute strategy e2e
  • Partner with researchers to understand our future research roadmap and to identify scaling limitations which will most imminently block us from achieving our goals
  • Be a hands-on leader who is excited to debug perplexing node failures at odd hours
  • Mentor junior engineers / researchers, and hold a high bar around code quality / engineering best practices
  • Work across teams to understand and manage project priorities and product deliverables, evaluate trade-offs, and drive technical initiatives from ideation to execution to shipment

You may be a good fit if you have :

  • 7 years of industry experience
  • Owned large scale distributed training efforts across thousands of accelerators
  • Experience with a panoply of HPC related tools and have developed strong opinions about how we should build our stack
  • A passion for solving the most pressing technical challenges, as opposed to the most intellectually satisfying ones
  • Strong Python and infrastructure-as-code expertise
  • Excellent communication skills in written and spoken English
  • Interview and Onboarding Process :

  • Recruiter Phone Screen
  • Initial Technical Assessment
  • Final round technical assessment & culture interview
  • Reference Checks
  • We're also excited to share that we'll do global onboarding in Seoul for all new hires (paid company travel!).

    Even if there are a few checkboxes that aren’t ticked through your prior experience, we still encourage you to apply! If you are a 0-to-1 achiever, a ferocious learner, and a kind and fun team player who motivates others, you will find a home at Twelve Labs.

    We welcome applicants from all walks of life and are committed to equal-opportunity employment. We cherish and celebrate diversity not just because it is the right thing to do, but because it makes our company much stronger.

    Benefits and Perks

  • An open and inclusive culture and work environment.
  • Work closely with a collaborative, mission-driven team on cutting-edge AI technology.
  • Full health, dental, and vision benefits
  • Extremely flexible PTO and parental leave policy. Office closed the week of Christmas and New Years.
  • Remote-flexible, offices in San Francisco and Seoul and coworking stipend
  • VISA support (such as H1B and OPT transfer for US employees)
  • J-18808-Ljbffr

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Machine Learning Engineer, Distributed Training Infrastructure?

    Sign up to receive alerts about other jobs on the Machine Learning Engineer, Distributed Training Infrastructure career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $123,167 - $152,295
    Income Estimation: 
    $146,673 - $180,130
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $184,796 - $233,226
    Income Estimation: 
    $77,900 - $95,589
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Twelve Labs Inc.

    Twelve Labs Inc.
    Hired Organization Address San Francisco, CA Full Time
    Who we are Is this your next job Read the full description below to find out, and do not hesitate to make an application...
    Twelve Labs Inc.
    Hired Organization Address San Francisco, CA Full Time
    Who we are At Twelve Labs, we are pioneering the development of cutting-edge multimodal foundation models that have the ...

    Not the job you're looking for? Here are some other Machine Learning Engineer, Distributed Training Infrastructure jobs in the San Francisco, CA area that may be a better fit.

    Software Engineer, Machine Learning Infrastructure

    DoorDash USA, San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!