Campus AI Research Engineer - Deep Learning (intern)

Jump Trading Jump Trading · Quant · Chicago, New York City · Front Office

Research Engineer Intern at Jump Trading focused on applying state-of-the-art ML techniques to financial markets. The role involves building flexible frameworks, optimizing training pipelines, integrating ML models into production systems with latency constraints, and developing observable, performant, and flexible large-scale ML systems. Requires strong ML background, publication record, and programming skills in Python/C++/CUDA.

What you'd actually do

  1. Apply state-of-the-art techniques to complex and challenging domains.
  2. Work closely with researchers and quants to build flexible and reusable frameworks for financial ML.
  3. Optimize training pipelines to make the best use of our HPC resources.
  4. Integrate ML models into production systems where latency matters.
  5. Build large-scale ML systems that are observable, performant, and flexible. Help improve productivity by reducing the iteration cycle time on research.

Skills

Required

  • Python
  • C++
  • CUDA
  • PyTorch
  • JAX
  • TensorFlow
  • ICML
  • ICLR
  • AAAI
  • NeurIPS
  • UAI
  • KDD

Nice to have

  • HPC
  • distributed large model training
  • GPU performance optimization
  • CUDA
  • ROCm
  • end-to-end model development

What the JD emphasized

  • state-of-the-art capabilities
  • implementing an open-ended research project from concept to production
  • continuously improving model design, tools, and infrastructure
  • latency matters
  • large-scale ML systems
  • reduce the iteration cycle time on research
  • Strong publication record at ICML, ICLR, AAAI, NeurIPS, UAI, KDD, or equivalent and/or contributions to open-source AI research
  • modern deep learning techniques
  • language modeling architectures
  • end-to-end model development

Other signals

  • Implement an open-ended research project from concept to production
  • Continuously improving model design, tools, and infrastructure
  • Integrate ML models into production systems where latency matters
  • Build large-scale ML systems that are observable, performant, and flexible
  • Reduce the iteration cycle time on research