Join a high-caliber team in New York City focused on pushing the boundaries of machine learning. You will lead the design and training of advanced deep learning systems, leveraging your expertise in PyTorch or JAX. This role offers the opportunity to apply broad ML literature knowledge to build innovative, production-grade models that define the next generation of AI technology.
Back to all jobsSee Open Roles
Learn about Jack
This job is no longer actively hiring. Open Roles to see active jobs.
ML Engineer at fast-growing AI startup
Are you a machine learning expert with a track record of training LLMs at top-tier tech companies? A fast-growing AI startup in New York City is looking for an ML Engineer to lead the design and optimization of next-generation deep learning systems. This is a rare opportunity to join a well-funded team backed by top VCs, where you will apply broad ML research to solve complex, real-world challenges. If you are proficient in PyTorch or JAX and want to build production-grade models at the cutting edge of the industry, this is the role for you.
Want to apply for this role?
C
This role is no longer actively hiring, but Jack can still help you discover similar open roles that fit.
Location
New York, United States
Compensation
Not Disclosed
Company
Confidential company
Role overview
About the company
Fast-growing AI startup
What you will do
- Design, train, and evaluate sophisticated deep learning systems using modern optimization techniques and specialized tooling.
- Implement and scale models using PyTorch or JAX to solve high-impact machine learning problems.
- Stay current with and apply the latest research from ML literature to enhance system performance and architecture.
Who this is a fit for
- At least 2 years of experience at top tech companies or startups specifically training Large Language Models (LLMs).
- Deep technical knowledge of modern deep learning architectures, optimization strategies, and the broader machine learning landscape.
- Proven proficiency in PyTorch or JAX with a track record of deploying robust ML systems into production.
Why this role is remarkable
- Work alongside elite engineering talent from top-tier technology companies and research institutions in a collaborative NYC environment.
- Join a well-funded startup backed by top-tier VCs at a critical stage of architectural development and model scaling.
- Tackle complex challenges that go beyond simple generative models, impacting the future of modern deep learning architectures.
How Jack & Jill work together
I get to know what you’re great at, then find roles you’d never find yourself.Ok, I'll go first. I'm Jack, an AI that gets to know you on a quick call, learning what you're great at and what you want from your career. Then I help you land your dream job by finding unmissable opportunities as they come up, supporting you with applications, interview prep, and moral support.
I recruit from Jack’s network and make the intro when I spot a great match.And I'm Jill, an AI Recruiter who talks to companies to understand who they're looking to hire. Then I recruit from Jack's network, making an introduction when I spot an excellent candidate.
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.
Meet Jack
What happens next?
Jack’s an AI agent for job searching and career coaching. He works for you.
Jill is the AI recruiter 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.