Machine Learning Systems Engineer: Distributed Training

Susquehanna International Group; Bala Cynwyd (PA); On-site; USD 140,000 - 210,000; Full time

Job summary

Susquehanna International Group is seeking a Machine Learning Systems Engineer to strengthen the performance and scalability of its distributed training infrastructure. You will collaborate with researchers to streamline large-scale training runs and help researchers make the most of compute resources.

Join us to build tooling for efficient training, profile and optimize training runs, and push the boundaries of high-performance ML systems in a collaborative, research-driven environment.

Qualifications

• Experience with large-scale ML training pipelines and distributed training frameworks.
• Strong Python software engineering skills.
• Ability to profile, benchmark, and optimize system-level performance.
• Interest in high-performance distributed systems.

Responsibilities

• Collaborate with researchers to enable efficient models and architectures.
• Apply techniques to improve hardware efficiency of training runs.
• Create tooling to distribute training jobs effectively.
• Profile and optimize training runs.

Skills

Distributed training

Python

Profiling & benchmarking

System optimization

Job description

Overview

We’re looking for a Machine Learning Systems Engineer to strengthen the performance and scalability of our distributed training infrastructure. In this role, you’ll work closely with researchers to streamline the development and execution of large-scale training runs, helping them make the most of our compute resources. You’ll contribute to building tools that make distributed training more efficient and accessible, while continuously refining system performance through careful analysis and optimization. This position is a great fit for someone who enjoys working at the intersection of distributed systems and machine learning, values high-performance code, and has an interest in supporting innovative machine learning efforts.

What You’ll Do

• Collaborate with researchers to enable them to develop systems-efficient models and architectures
• Apply the latest techniques to our internal training runs to achieve impressive hardware efficiency for our training runs
• Create tooling to help researchers distribute their training jobs more effectively
• Profile and optimize our training runs

What we’re looking for

• Experience with large-scale ML training pipelines and distributed training frameworks
• Strong software engineering skills in python
• Passion for diving deep into systems implementations and understanding fundamentals to improve their performance and maintainability
• Experience improving resource efficiency across distributed computing environments by leveraging profiling, benchmarking, and implementing system-level optimizations

Why Join Us?

Susquehanna is a global quantitative trading firm that combines deep research, cutting-edge technology, and a collaborative culture. We build most of our systems from the ground up, and innovation is at the core of everything we do. As a Machine Learning Systems Engineer, you’ll play a critical role in shaping the future of AI at Susquehanna - enabling research at scale, accelerating experimentation, and helping unlock new opportunities across the firm.

LI-Onsite