AI/ML Systems Engineer Intern - HPC & Low-Latency

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