Campus AI Research Engineer Intern

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

Job summary

Jump Trading Group is a leading quantitative research and trading firm recognized for world-class research in mathematics, physics, and computer science. The team designs, develops, and deploys state-of-the-art AI/ML systems to solve complex problems in global financial markets.

The role involves applying cutting-edge techniques, building scalable AI/ML frameworks with researchers and quants, and deploying low-latency inference systems on HPC clusters.

Qualifications

• Strong Python/C++ programming skills in a research or production setting.
• Experience with ML frameworks (PyTorch, TensorFlow, JAX) and HPC environments.
• Ability to work with researchers and engineers to deploy scalable AI/ML 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.
• Other duties as assigned or needed.

Skills

Python

C++

PyTorch

JAX

TensorFlow

GPU programming

Distributed systems

English communication

Team collaboration

CUDA

Tools

CUDA

Triton

SYCL

ROCm

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, and engineering teams to build state-of-the-art ML systems that solve some of the most complex problems in quantitative finance. Whether optimizing training pipelines on high-performance computing clusters, developing low-latency inference systems, or pushing the boundaries of AI research from concept to production, you'll have 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 advancing AI/ML capabilities, we want to meet you.

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.

Skills You’ll Need:

This role covers a wide range of potential projects and skills. We don't expect everyone to have all of these, but for the applicable areas we are looking for deep technical expertise.

• 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.

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