Campus AI Research Engineer (Intern)

Summary

Jump Trading Group is committed to world-class research in quantitative finance, seeking exceptional talents in Mathematics, Physics, and Computer Science. The Campus AI Research Engineer (Intern) will collaborate with research and engineering teams to build advanced ML systems, optimizing training pipelines and integrating AI/ML models into production systems.

Responsibilities

• 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

Skills

• 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
• Creative thinkers who are driven, self-motivated, and eager to solve challenging problems

Qualifications

Must Haves

• 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

Nice to Haves

• Creative thinkers who are driven, self-motivated, and eager to solve challenging problems

Benefits

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