Campus AI Research Engineer – Deep Learning (Full-Time)

Job title: Campus AI Research Engineer – Deep Learning (Full-Time)

Jump Trading Group is seeking a Campus AI Research Engineer – Deep Learning to join our research-driven team. The role focuses on applying state-of-the-art machine learning techniques to complex domains in finance, from concept to production, while collaborating with researchers, quants, and engineers.

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 ML.
• Optimize training pipelines to make the best use of our HPC resources.
• Integrate 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 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

• Strong publication record at ICML, ICLR, AAAI, NeurIPS, UAI, KDD, or equivalent and/or contributions to open-source AI research.
• Strong general ML background with exposure to modern deep learning techniques and/or language modeling architectures (e.g., transformers, SSMs).
• Solid development skills in Python and/or C++.
• Familiarity with ML libraries/frameworks such as PyTorch, JAX, and/or TensorFlow.
• Intellectual curiosity, versatility, and originality combined with a pragmatic outlook.
• Ability to thrive in a collaborative, team-oriented environment.
• Ability to reason through quantitative problems and communicate effectively with trading researchers.
• Reliable and predictable availability.

Nice to Have

• Experience with HPC and distributed large model training.
• Experience with GPU performance optimization (CUDA or ROCm).
• Experience with end-to-end model development.
• Strong opinions on best practices in ML research, tooling, and/or infrastructure.
• International students are encouraged to apply. We accept CPT/OPT and sponsor work visas for full-time positions.

Compensation & Benefits

• The estimated base salary for this role is $300,000 per year.
• We sponsor work visas for full-time positions; international students are encouraged to apply (CPT/OPT).