Research Engineer, Pre-Training
Jump TradingAbout the Role
As a Pre-Training Research Engineer, you'll be at the forefront of developing massive-scale foundation models that fundamentally transform how we understand and predict markets. You'll own and drive the entire training stack: building fault-tolerant infrastructure that scales across thousands of GPUs and TPUs, engineering data pipelines that stream terabytes per second, and designing custom kernels that unlock significant efficiency gains. You'll co-design novel architectures with researchers and pioneer cutting-edge approaches to mixed-precision training and model parallelism, with the latest generation hardware at your disposal. Your improvements directly impact live trading at scale.
Requirements
- Expertise and track record of significant, measurable performance improvements in large-scale distributed training (MFU, throughput, convergence, cost-per-token)
- Published research in efficient training methods, scaling laws, architectures, or systems for ML
- Background in numerical computing, HPC, or distributed systems, including familiarity with GPUs/TPUs, high-performance networking (NVLink/InfiniBand), Kubernetes/Slurm, and OS internals
- Expertise in Python and deep experience with modern deep learning frameworks (PyTorch and/or JAX)
- Advanced degree (MS or PhD) 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
- Strong problem-solving skills, results orientation, and excellent collaborative communication
- Reliable and predictable availability
What You'll Get
- $300k–$350k base salary
- Discretionary bonus eligibility
- Medical, dental, and vision insurance
- HSA, FSA, and Dependent Care options
- Employer Paid Group Term Life and AD&D Insurance
- Voluntary Life & AD&D insurance
- Paid vacation plus paid holidays
- Retirement plan with employer match
- Paid parental leave
- Wellness Programs