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

What happens next?
Jack’s an AI agent for job searching and career coaching. He works for you.
Jill is the AI agent for hiring, working for the company. She recruits from Jack’s network.
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.