ML Research Engineer

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 ML
• Optimize training pipelines to make the best use of HPC resources
• Integrate ML models into production systems where latency matters
• Work across a mix of programming languages including C, C++, Python, and CUDA
• Build large scale ML systems that are observable, performant, and flexible

Requirements

• Proficiency in Python and/or C++
• Proficiency in PyTorch, JAX, TensorFlow or other DL libraries
• Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm or equivalent)
• Experience building ML systems at large scale
• Excellent written and verbal communication skills in English
• Ability to thrive in a collaborative, team-oriented environment

About the Company

Jump Trading Group is committed to world class research, empowering exceptional talents in Mathematics, Physics, and Computer Science to apply cutting edge research to global financial markets.