Skip to main content

This job is no longer actively hiring. Talk to Jack to find live roles.

ML Engineer at fast-growing AI startup

Are you an ML Engineer with a track record of training Large Language Models at top-tier tech companies? Join this well-funded NYC AI startup to design and evaluate cutting-edge deep learning systems using PyTorch and JAX. This is a rare opportunity to move beyond generative models and apply broad ML literature to solve complex problems within a high-caliber engineering team.

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

See Open Roles

Role overview

As an ML Engineer at this fast-growing AI startup, you will design and train advanced deep learning systems at scale. This role focuses on pushing the boundaries of modern architectures and optimization techniques. You will work alongside a high-caliber team in NYC to evaluate models and contribute to the broader machine learning landscape.

About the company

Fast-growing AI startup

What you will do

  • Design, train, and evaluate sophisticated deep learning systems using PyTorch or JAX.
  • Implement modern deep learning architectures and optimization strategies for high-performance models.
  • Stay abreast of current ML literature to integrate novel research into production-ready systems.

Who this is a fit for

  • At least 2 years of experience at top tech companies or startups specifically training Large Language Models.
  • Deep proficiency in PyTorch or JAX and modern deep learning tooling and infrastructure.
  • Breadth of knowledge across ML literature, including optimization, architectures, and evaluation frameworks.

===

Why this role is remarkable

  • Join a high-performing team of engineers from top-tier technology companies and research labs.
  • Work at a well-funded startup backed by top-tier VCs in the heart of NYC's tech scene.
  • Influence the direction of cutting-edge AI products by applying deep learning literature beyond generative models.

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.

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.

Learn about Jack

Ready to find your next role?

Talk to Jack for 10 minutes and see your first matches.