Campus AI Research Engineer (Intern)

Jump Trading; New York (NY); Hybrid; USD 255,000 - 345,000; Full time

Job summary

Jump Trading Group seeks world-class engineers to collaborate with our research, trading, and engineering teams to build state-of-the-art ML systems for quantitative finance. You will optimize training pipelines on HPC clusters, develop low-latency inference, and push AI/ML from concept to production using C/C++, Python, CUDA.

Based in the United States, this fast-paced role offers impactful projects across research and production systems.

Qualifications

• Proven ability for building ML systems in production environments.
• Strong in Python and C++, with GPU programming experience.
• Experience with modern ML frameworks (PyTorch, TensorFlow).
• Experience scaling ML training on large HPC clusters.

Responsibilities

• Apply state-of-the-art techniques to complex domains.
• Collaborate with researchers and quants to build reusable ML frameworks.
• Optimize training pipelines to utilize HPC resources efficiently.
• Integrate AI/ML models into latency-sensitive production systems.
• Work across C/C++, Python, CUDA and other GPU languages.
• Build scalable, observable AI/ML systems with high performance.
• Other duties as assigned or needed.

Skills

Python

C++

CUDA

PyTorch

JAX

TensorFlow

GPU programming

Distributed systems

Job description

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

Our trading teams are each comprised of a dynamic group of traders, quantitative researchers, and engineers who work together to examine the global markets, seeking to understand the complexities of various traded products and exchanges. They leverage their impeccable statistical analysis and data mining skills, using the results of their research to make forecasts and develop profitable predictive trading models.

We are seeking world-class engineers to collaborate with our research,trading,andengineering teams to buildstate-of-theartML systems that solve some of the most complex problems in quantitative finance. Whetheroptimizingtraining pipelines on high-performance computing clusters, developing low-latency inference systems, or pushing the boundaries of AI research from concept to production,you'llhave the opportunity to work on impactful projects in a fast-paced, collaborative environment. If you are driven by technical challenges, eager to work with large-scale systems, and passionate about advancingAI/ML capabilities, we want to meet you.

WhatYou'llDo:

• Applystate-of-the-arttechniques to complex and challenging domains.
• Work closely with researchers and quants to build flexible and reusable frameworks for financialAI/ML.
• Optimizetraining pipelines to make the best use of our HPC resources.
• IntegrateAI/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.
• Buildlarge-scaleAI/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.

SkillsYou'llNeed:

This role covers a widerangeof potential projects and skills. Wedon'texpect everyone to have all of these, but for the applicable areas we are looking for deep technicalexpertise.

• Creative thinkers who are driven, self-motivated, and eager to solve challenging problems
• Proficiencyin Python and/or C++
• ProficiencyinPyTorch, JAX,TensorFlow, and/orsimilarframeworks
• Ability to thrive in a collaborative, team-oriented environment
• Expertisein GPU oracceleratorprogramming (CUDA, Triton, SYCL,ROCm, orequivalent)
• Experience buildingAI/ML systems at scale (hundreds of TBs of training data,low-latencyorhigh-throughputinference 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/OPTand we sponsor work visas for full-time positions.

The estimated base salary for this role (annualized) is $300,000 per year.

We invest in what's next. Then we build it.