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)

- Strong C++ & Python programming skills

- 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

Bonus points for:

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

- Linux system programming expertise

- Contributions to open-source AI projects

Why Join?

-High-impact role optimizing AI infrastructure

-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. ([email protected]) or give me a call on (07834635117 / 737 256 7409).