Machine Learning Performance Engineer

Fintal Partners; New York (NY); On-site; USD 180,000 - 280,000; Full time

Benefits offered by this job

401(k) match up to 50%

Health, dental, vision, life

Mental health coverage

25 days paid leave

Office perks & events

Job summary

Fintal Partners seeks a Machine Learning Performance Engineer to shape the flagship AI program. You will optimize training and inference pipelines, explore framework internals, and hunt bottlenecks while collaborating with researchers and engineers across global teams.

The role offers exposure to high-performance ML in a global trading context, with competitive compensation, health coverage, and generous leave.

Qualifications

• Strong low-level GPU programming with CUDA.
• Deep familiarity with ML frameworks (PyTorch, JAX, TensorFlow).
• Solid grounding in computer architecture and parallelism.
• Experience with C++ and Python for ML infra.
• Experience with large-scale distributed training and OSS contributions.

Responsibilities

• Build robust, scalable training and inference pipelines for deep learning.
• Work inside the internals of open-source deep learning frameworks and extend functionality.
• Identify and remove performance bottlenecks in ML pipelines.
• Collaborate with researchers and engineers across teams.
• Develop deep knowledge of trading systems and their ML components.

Skills

CUDA programming

PyTorch / JAX / TensorFlow

C++

Python

computer architecture

distributed training

open-source contributions

Tools

PyTorch/JAX/TensorFlow

XLA/Flax

Job description

A leading global trading firm is hiring a Machine Learning Performance Engineer to work on a flagship AI program. You'll have real influence across ML infrastructure, training and inference, work that feeds directly into the firm's futures trading strategies.

The role

• Build robust, scalable training and inference pipelines for deep learning
• Work inside the internals of open-source deep learning frameworks and extend what they can do
• Hunt down and remove performance bottlenecks
• Partner closely with researchers and fellow engineers
• Build deep knowledge of how the trading systems work

What's on offer

• You'll be part of a collaborative, high-performing culture built around curious thinkers and inventive problem-solvers, tackling some of the hardest challenges in financial markets alongside genuinely strong colleagues.
• Colleagues drawn from 40+ nationalities
• Highly competitive compensation
• 401(k) matched up to 50%
• Full health, mental health, dental, vision, disability and life cover
• 25 days' paid leave plus market holidays
• Generous office perks — breakfast, lunch and snacks, social events, clubs and sports leagues

What you'll bring

• Strong low-level GPU programming in CUDA — Tensor Cores, cooperative groups, graphs, warp-level intrinsics
• Deep familiarity with the internals of frameworks such as PyTorch, JAX or TensorFlow
• Solid grounding in computer architecture
• C++ and Python
• The JAX ecosystem (XLA, Flax and similar)
• Large-scale distributed training
• Open-source contributions in ML or data science

About the firm

A major market maker active on 100+ exchanges worldwide, focused on improving markets by adding liquidity, pricing accurately and increasing transparency in all conditions — with the health and efficiency of those markets a priority for every participant.