ML Engineer at fast-growing AI startup

Jack & Jill/External ATS • New York, NY

2d • Onsite

About The Position

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.

Requirements

• 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.

Responsibilities

• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed
• Number of Employees: 1-10 employees