Campus ML Engineer (Intern)
We are seeking world-class engineers to collaborate with our research, trading, engineering teams to build state-of-the-art ML systems that solve some of the most complex problems in quantitative finance. Whether optimising training pipelines on high-performance computing clusters, developing low-latency inference systems, or pushing the boundaries of AI research from concept to production, you'll have the opportunity to work on impactful projects in a fast-paced, collaborative environment. If you are driven by technical challenges, eager to work with large-scale systems, and passionate about advancing 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 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.
Skills you'll need:
This role covers a wide gamut 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 or other DL library.
- Ability to thrive in a collaborative, team-oriented environment
- Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm or equivalent)
- Experience building ML systems at large 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 required
INTERNATIONAL STUDENTS are encouraged to apply. We accept students eligible for CPT/OPT and we sponsor work visas for full-time positions.