Research Engineer, Pre-Training at Jump Trading | Chicago, IL, US | Rezi

Research Engineer, Pre-Training at Jump Trading

Research Engineer, Pre-Training

Jump Trading · Chicago, IL, US

1 months ago

Research Engineer, Pre-Training

Jump Trading · Chicago, IL, US

a month ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Research Engineer, Pre-Training role.

Rezi rewrites your resume against Jump Trading's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Research Engineer, Pre-Training posting at Jump Trading — free, in seconds.

About the Role

We are seeking an exceptional Pre-Training Engineer to join our team, focusing on developing massive-scale foundation models that transform market prediction. You will own and drive the entire training stack, from infrastructure to data pipelines and custom kernels, pushing the boundaries of pre-training at scale.

Responsibilities

  • Build fault-tolerant infrastructure that scales across thousands of GPUs and TPUs with near-linear performance.
  • Engineer data pipelines that stream terabytes per second as models train on petabytes of data.
  • Design custom kernels that unlock 10x efficiency gains.
  • Co-design novel architectures with researchers.
  • Pioneer cutting-edge approaches to mixed-precision training and model parallelism.
  • Other duties as assigned or needed.

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.
  • 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).
  • Ability to balance ambitious research goals with practical engineering constraints.
  • Strong problem-solving skills, results orientation, and excellent collaborative communication.
  • Reliable and predictable availability.

Skills

  • CUDA kernel development
  • Triton/Pallas/CuTe DSLs
  • PyTorch/JAX internals
  • XLA optimization
  • Hardware acceleration (FPGA/ASIC)
  • Reinforcement learning
  • Post-training techniques
  • Fine-tuning techniques
  • Financial markets knowledge
  • Trading knowledge

Location

  • Remote

Work Type

  • Full-time

Experience Level

  • MS or PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier lab.

Education Level

  • MS or PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related quantitative field, or equivalent industry experience at a frontier lab.

Salary/Compensations

  • $300,000—$350,000 USD

Benefits

  • 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

About the Company

  • Jump Trading Group is committed to world-class research, empowering exceptional talents in Mathematics, Physics, and Computer Science to seek scientific boundaries and apply cutting-edge research to global financial markets.
  • Our culture is unique, characterized by constant innovation, fearlessness, creativity, intellectual honesty, and a relentless competitive streak.
  • We believe in winning together, unlocking unique individual talent by incentivizing collaboration and mutual respect.
  • Research outcomes drive superior risk-adjusted returns, and we design, develop, and deploy technologies that change our world.
  • We fund start-ups across industries and partner with leading global research organizations and universities to solve problems.