Research Engineer, Pre-Training (AI)

TL;DR

Research Engineer, Pre-Training (AI): Developing massive-scale foundation models for financial market prediction with an accent on fault-tolerant infrastructure and high-performance data pipelines. Focus on scaling training across thousands of GPUs/TPUs and designing custom kernels for maximum efficiency.

Location: Must be based in or able to work from New York, London, or Chicago

Salary: $300,000 - $350,000 USD

Company

hirify.global Group is a quantitative research and trading firm that applies cutting-edge scientific research and technology to global financial markets.

What you will do

• Own and drive the entire pre-training stack for massive-scale foundation models.
• Build fault-tolerant infrastructure scaling across thousands of GPUs and TPUs with near-linear performance.
• Engineer high-throughput data pipelines streaming terabytes per second from global market data.
• Design custom kernels and pioneer cutting-edge approaches to mixed-precision training and model parallelism.
• Co-design novel architectures with researchers to fundamentally transform market prediction.

Requirements

• Track record of measurable performance improvements in large-scale distributed training (MFU, throughput, convergence).
• Published research in efficient training methods, scaling laws, or ML systems.
• Background in numerical computing, HPC, or distributed systems (NVLink, InfiniBand, Kubernetes/Slurm).
• Expertise in Python and deep experience with PyTorch and/or JAX.
• Advanced degree (MS or PhD) in Computer Science, Machine Learning, Physics, Mathematics, or equivalent frontier lab experience.
• Reliable and predictable availability to work in New York, London, or Chicago.

Nice to have

• Expertise in CUDA kernel development, Triton, Pallas, CuTe DSLs, or XLA optimization.
• Knowledge of reinforcement learning, post-training, or fine-tuning techniques.
• Familiarity with financial markets or trading.

Culture & Benefits

• Competitive base salary with discretionary bonus eligibility.
• Comprehensive medical, dental, and vision insurance including HSA and FSA options.
• Retirement plan with employer match.
• Paid vacation, paid holidays, and parental leave.
• Wellness programs and a culture of innovation and intellectual honesty.