Research Engineer, Pre-Training

Home > Python programming jobs

Research Engineer, Pre-Training in UK

Jump Trading
Full time
Email

UK
10/06/2026
Report

London

Responsibilities

Own and scale the full training stack across thousands of GPUs and TPUs with near-linear performance.
Build fault-tolerant distributed training infrastructure and high-throughput pipelines streaming terabytes per second from petabytes of global financial-market data.
Design custom kernels and optimize mixed-precision training, model parallelism, and training efficiency.
Co-design novel foundation-model architectures and apply research improvements to live trading.
Collaborate with quantitative researchers, engineers, and ML experts, and perform other duties as assigned.

Requirements

Demonstrated expertise and measurable performance improvements in large-scale distributed training, including MFU, throughput, convergence, or cost-per-token.
Published research in efficient training methods, scaling laws, architectures, or ML systems.
Background in numerical computing, HPC, or distributed systems, including GPUs/TPUs, NVLink/InfiniBand, Kubernetes/Slurm, and OS internals.
Expertise in Python and deep experience with PyTorch and/or JAX.
Advanced degree in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier lab.
Ability to balance ambitious research goals with practical engineering constraints, with strong problem-solving, results orientation, collaboration, and communication skills.
Reliable and predictable availability.
Preferred expertise includes CUDA kernel development, Triton, Pallas, CuTe, PyTorch/JAX internals, XLA optimization, FPGA/ASIC acceleration, reinforcement learning, post-training, fine-tuning, or financial markets and trading.

Benefits

Discretionary bonus eligibility.
Medical, dental, and vision insurance.
HSA, FSA, and Dependent Care options.
Employer-paid group term life and AD&D insurance, plus voluntary life and AD&D insurance.
Paid vacation, paid holidays, and paid parental leave.
Retirement plan with employer match.
Wellness programs.

Salary: $300k - $350k/yr

Jump Trading

Jump Trading builds innovative trading technologies that leverage advanced research in mathematics, physics, and computer science to optimize performance in global financial markets. We cater to talented professionals seeking to push scientific boundaries and collaborate within a dynamic environm...

Similar positions

Lead Engineer, Machine Learning

Bjak
Full time

Remote
10/15/2026

Worldwide

Deployed Engineer, Professional Services

LangChain
Full time

Remote
10/15/2026
Salary: Competitive

Singapore, US

Senior Software Engineer, Network Development

CoreWeave
Full time

USA
10/15/2026
Salary: $165k - $220k/yr

Bellevue, WA

Founding ML Research Engineer

Clera
Full time

USA
10/15/2026
Salary: $200k - $300k/yr

San Francisco, CA

Operations Engineer, Tech Foundations

Airbnb
Full time

Remote
10/15/2026

India