AI Research Engineer

Jump Trading Group is dedicated to world-class research. We empower exceptional talents in Mathematics, Physics, and Computer Science to push scientific boundaries and apply cutting-edge research to global financial markets. Our culture is unique, driven by constant innovation, fearlessness, creativity, intellectual honesty, and a relentless competitive streak. We believe in winning together and unlocking individual talent through collaboration and mutual respect. Research outcomes at Jump 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.

Our trading teams are dynamic groups of traders, quantitative researchers, and engineers who examine global markets to understand the complexities of various traded products and exchanges. They leverage impeccable statistical analysis and data mining skills, using research results 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 complex problems in quantitative finance. You will have the opportunity to work on impactful projects in a fast-paced, collaborative environment, whether optimizing training pipelines on high-performance computing clusters, developing low-latency inference systems, or pushing AI research from concept to production. If you are driven by technical challenges 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.