Machine Learning Performance Engineer

Digital Republic Talent; New York (NY); On-site; Full time

Job summary

A leading tech company is seeking a Machine Learning Performance Engineer to enhance AI infrastructure in New York City. This in-office role involves optimizing machine learning training and inference pipelines while collaborating with engineers and researchers. The ideal candidate will have expertise in deep learning frameworks and performance optimization, contributing to a pivotal AI initiative.

Benefits

Competitive salary

Performance-based bonuses

Global profit-sharing

401(k) matching

Office meals

Social events

Clubs

Qualifications

• Expert in deep learning framework internals.
• Deep understanding of computer architecture.
• Experience with large-scale distributed training.

Responsibilities

• Build scalable ML training & inference pipelines.
• Optimize deep learning frameworks for performance.
• Identify & eliminate performance bottlenecks.

Skills

C++

Deep Learning

Performance Optimization

GPU Acceleration

Tools

PyTorch

JAX

TensorFlow

CUDA

Triton

Pallas

Job description

Get AI-powered advice on this job and more exclusive features.

Digital Republic Talent provided pay range

This range is provided by Digital Republic Talent. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$300.00/yr - $350,000.00/yr

Direct message the job poster from Digital Republic Talent

Crafting Data-Driven Duets: Bridging Data / Analytics & Programmatic Talent with Thriving Clients

Machine Learning Performance Engineer | High-Performance AI | Located in NYC / Austin / Chicago | 5 Days in Office

Are you an expert in C++, deep learning frameworks, and performance optimization? Do you thrive on eliminating bottlenecks and scaling AI infrastructure? We’re seeking a Machine Learning Performance Engineer to work on a pivotal AI initiative in New York City. This is a 5-day in-office role, offering an amazing opportunity to optimize machine learning training, inference, and infrastructure at scale.

What You’ll Do:

- Build scalable ML training & inference pipelines

- Optimize deep learning frameworks (PyTorch, JAX, TensorFlow) to enhance performance

- Identify & eliminate performance bottlenecks in AI systems

- Work with GPUs & accelerators (CUDA, Triton, Pallas) to push efficiency limits

- Collaborate closely with researchers & engineers to integrate performance improvements

What We’re Looking For:

- Expert in deep learning framework internals (JAX, PyTorch, TensorFlow)

- Deep understanding of computer architecture & performance engineering

- Experience with large-scale distributed training & AI model optimization

- Prior experience in big tech, Autonomous Vehicles (AV), and Hedge Fund/Quantitative Trading firms

- Experience with GPU acceleration (CUDA, Triton, Pallas, etc.)

- Contributions to open-source AI projects

Why Join?

-Competitive salary + performance-based bonuses

-Global profit-sharing & 401(k) matching

-Extensive perks: Office meals, social events, clubs, and more

Who is Digital Republic? Digital Republic Recruitment has been created with the sole purpose of delivering dedicated web and digital analytics recruitment solutions across the UK. Digital Republic aims to use our in-depth web and digital analytics expertise to provide an unparalleled experience to anyone doing business with our passionate and motivated team.

How to Apply? If you are interested in this role and want to find out more, apply via the link at the bottom of this page or contact Digital Republic by email to find out more about the role. ([e-mail]) or give me a call on (07834635117 / 737 256 7409).

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Finance, Accounting/Auditing, and Engineering

Referrals increase your chances of interviewing at Digital Republic Talent by 2x

Sign in to set job alerts for “Machine Learning Engineer” roles.

New York, NY $170,000.00-$250,000.00 5 days ago

New York, NY $56.25-$173,000.00 1 week ago

New York, NY $132,000.00-$184,000.00 1 week ago

New York, NY $180,000.00-$290,000.00 3 weeks ago

New York, NY $46.63-$134,000.00 4 days ago

Software Engineer (L4) - Ads Measurement

New York, United States $100,000.00-$720,000.00 4 days ago

New York, NY $140,000.00-$200,000.00 7 months ago

Machine Learning Engineer, Personalization

New York, NY $140,000.00-$200,000.00 7 months ago

New York, NY $46.63-$134,000.00 1 week ago

New York, NY $160,000.00-$280,000.00 12 months ago

New York, NY $90,000.00-$140,000.00 11 months ago

New York, United States $120,000.00-$240,000.00 4 months ago

New York, NY $164,000.00-$198,000.00 13 hours ago

Machine Learning, Software Engineer (Remote)

New York, NY $190,000.00-$240,000.00 1 month ago

New York, NY $114,000.00-$171,000.00 1 week ago

New York, NY $180,000.00-$240,000.00 1 month ago

New York, NY $85.10-$251,000.00 1 week ago

New York, NY $110,000.00-$250,000.00 1 month ago

Machine Learning Engineer, Stripe Capital

New York, NY $220,000.00-$300,000.00 2 months ago

Machine Learning Scientist, NLP (All Ranks)

New York, NY $200,000.00-$300,000.00 2 days ago

New York, United States $155,000.00-$213,200.00 12 hours ago

Machine Learning Scientist, NLP (All Ranks)

New York, NY $200,000.00-$300,000.00 2 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.