Campus AI Research Engineer — Visa Sponsorship

Trading Interview; New York (NY); On-site; USD 270,000 - 330,000; Full time

Benefits offered by this job

Discretionary bonus eligibility

Medical, dental, and vision insurance

HSA, FSA, and Dependent Care options

Employer Paid Group Term Life and AD&D

Voluntary Life & AD&D insurance

Paid vacation plus paid holidays

Retirement plan with employer match

Paid parental leave

Wellness Programs

Job summary

Jump Trading Group seeks world-class engineers to collaborate with research, trading, and engineering teams to build state-of-the-art ML systems for quantitative finance. You will optimize training pipelines on HPC clusters, integrate AI/ML models into production where latency matters, and work across C/C++, Python, and CUDA.

This role offers exposure to large-scale, impactful projects in a fast-paced environment.

Qualifications

• Proficiency in Python and/or C++.
• Experience with PyTorch, JAX, or TensorFlow.
• GPU or accelerator programming expertise (CUDA, Triton, SYCL, ROCm).
• Experience building AI/ML systems at scale (hundreds of TBs of training data, low-latency or high-throughput inference).
• Excellent written and verbal communication skills in English.

Responsibilities

• Apply state-of-the-art techniques to complex domains.
• Work with researchers and quants to build flexible AI/ML frameworks.
• Optimize training pipelines on HPC resources.
• Integrate AI/ML models into production systems where latency matters.
• Work across languages: C/C++, Python, CUDA.
• Build scalable AI/ML systems that are observable and performant.
• Other duties as assigned.

Skills

Python

C++

PyTorch

JAX

TensorFlow

GPU programming

Tools

CUDA

Triton

SYCL

ROCm

Job description

Jump Trading Group seeks world-class engineers to collaborate with research, trading, and engineering teams to build state-of-the-art ML systems for quantitative finance. You will optimize training pipelines on HPC clusters, integrate AI/ML models into production where latency matters, and work across C/C++, Python, and CUDA.

This role offers exposure to large-scale, impactful projects in a fast-paced environment.