Skip to main content

Confidential company, Engineering

Staff Machine Learning Engineer at well-funded consumer AI startup

Location
Remote (United States)
Compensation
Not Disclosed

Are you ready to move beyond basic AI prompting and build a truly proactive consumer application? This well-funded startup is looking for a Staff ML Engineer to architect the end-to-end systems powering a next-generation productivity platform. You'll lead LLM fine-tuning, optimize GPU inference, and build the foundation for an app that automates workflows for millions.

Backed by a profitable parent company, this role offers the fast-paced innovation of a startup with the stability and scale of an established market leader. If you are a PyTorch or JAX expert looking to ship products that redefine daily digital interaction, this is your chance to lead.

What's the next step?

Confidential company are hiring from Jack's network. Talk to Jack for free and get a direct introduction to the team, if a good fit.

Role overview

You will lead the design and implementation of end-to-end machine learning systems for a next-generation consumer productivity platform. This includes building LLM fine-tuning pipelines, evaluation frameworks, and high-performance inference systems. Your work will directly shape the technical foundation of an AI-native application designed to automate complex daily workflows for millions of users globally.

About the company

Well-funded consumer AI startup

What you will do

  • Design and scale LLM training, fine-tuning, and evaluation pipelines for production-grade reliability.
  • Optimize GPU-based inference systems for maximum throughput, low latency, and efficient resource utilization.
  • Architect robust data processing pipelines to validate and manage high-quality training and evaluation datasets.

Who this is a fit for

  • Deep expertise in Python and hands-on experience with frameworks like PyTorch or JAX for large-scale models.
  • Proven track record of deploying and monitoring production ML systems, specifically LLM-based applications.
  • Strong background in distributed computing and optimizing inference engines like vLLM or TensorRT-LLM.

Why this role is remarkable

  • Build a ground-up ML architecture for a high-scale consumer application used daily by millions.
  • Backed by a profitable, large-scale parent organization, providing rare long-term stability and significant resources.
  • High-impact leadership role shaping the core technical strategy of an ambitious, proactive AI-native ecosystem.

How Jack & Jill work together

Jack
I get to know what you’re great at, then find roles you’d never find yourself.
Jill
I recruit from Jack’s network and make the intro when I spot a great match.
Thumbnail for Meet Jack
Jack gets to know what you're great at and what you want next, then searches 15 million jobs daily and helps you discover roles at companies like this.

What happens next?

  1. Jack’s an AI agent for job searching and career coaching. He works for you.

  2. Jill is the AI agent for hiring, working for the company. She recruits from Jack’s network.

  3. If your profile’s a match and Confidential company wants to meet, Jill will make the intro. In the meantime, Jack will send you excellent alternatives.

Learn about Jack