Machine Learning Performance Engineer

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.

\n

\n

The role

\n

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

\n

\n

What's on offer

\n

\n
• 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.
\n
• Colleagues drawn from 40+ nationalities
\n
• Highly competitive compensation
\n
• Firm-wide profit share plus performance-linked bonus
\n
• 401(k) matched up to 50%
\n
• Full health, mental health, dental, vision, disability and life cover
\n
• 25 days' paid leave plus market holidays
\n
• Generous office perks — breakfast, lunch and snacks, social events, clubs and sports leagues
\n

\n

\n

What you'll bring

\n

\n
• Strong low-level GPU programming in CUDA — Tensor Cores, cooperative groups, graphs, warp-level intrinsics
\n
• Deep familiarity with the internals of frameworks such as PyTorch, JAX or TensorFlow
\n
• Solid grounding in computer architecture
\n
• C++ and Python
\n

\n

\n

Bonus points for

\n

\n
• The JAX ecosystem (XLA, Flax and similar)
\n
• GPU tooling — Triton, CUB, cuDNN, cuBLAS
\n
• Linux systems programming
\n
• Large-scale distributed training
\n
• Open-source contributions in ML or data science
\n

\n

\n

About the firm

\n

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.