Campus AI Research Engineer (Full-Time)

The role

You will build state-of-the-art ML systems to solve complex problems in quantitative finance. This involves optimizing training pipelines, developing low-latency inference systems, and pushing AI research from concept to production.

What you will do

• Apply state-of-the-art techniques to complex and challenging domains.
• Work closely with researchers and quants to build flexible and reusable frameworks for financial AI/ML.
• Optimize training pipelines to make the best use of our HPC resources.
• Integrate AI/ML models into production systems where latency matters.
• Work across a mix of programming languages: C / C++ / Python / CUDA and other low-level GPU languages.
• Build large-scale AI/ML systems that are observable, performant, and flexible.
• Help improve productivity by reducing the iteration cycle time on research.
• Other duties as assigned or needed.

Required qualifications

• Creative thinkers who are driven, self-motivated, and eager to solve challenging problems
• Proficiency in Python and/or C++
• Proficiency in PyTorch, JAX, TensorFlow, and/or similar frameworks
• Ability to thrive in a collaborative, team-oriented environment
• Expertise in GPU or accelerator programming (CUDA, Triton, SYCL, ROCm, or equivalent)
• Experience building AI/ML systems at scale (hundreds of TBs of training data, low-latency or high-throughput inference requirements)
• Excellent written and verbal communication skills in English
• Reliable and predictable availability

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

Before you apply

• We accept students eligible for CPT/OPT
• We sponsor work visas for full-time positions

About the company

Jump Trading Group is committed to world class research. They design, develop, and deploy technologies that change our world, fund start-ups, and partner with research organizations and universities.