Skip to main content
Back to all jobs

This job is no longer actively hiring. Open Roles to see active jobs.

Confidential company

Job listing

New York, United StatesNot Disclosed

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.

Overview

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.

Company

About the company

Fast-growing AI startup

Responsibilities

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.

Candidate profile

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.

===

What makes it remarkable

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.

Jack & Jill

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.

Meet Jack

Thumbnail for Meet Jack

Jack gets to know what you're great at and what you want next, then searches 14 million jobs daily and introduces you directly to hiring managers.

How does this work?

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 it's a match and the company wants to meet you, they'll make the intro. In the meantime, if you'd like, Jack will send you excellent alternatives.

Find a job withJack

Ready to find your next role?

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