Campus AI Research Engineer (Intern)

Job title: Campus AI Research Engineer (Intern)

About the Role Jump Trading Group is committed to world class research and seeks to recruit talented individuals to collaborate across research, trading and engineering to build state-of-the-art AI/ML systems for quantitative finance. As a Campus AI Research Engineer (Intern), you will contribute to applying advanced ML techniques, helping to scale and productionize models in a fast-paced trading environment.

This internship offers the opportunity to work with researchers, quants, and engineers to design flexible AI/ML frameworks, optimize training pipelines on HPC resources, and deploy models where latency matters.

What You'll 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.

What We're Looking For

• 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.
• INTERNATIONAL STUDENTS are encouraged to apply. We accept students eligible for CPT/OPT and we sponsor work visas for full-time positions.