Machine Learning Engineer- Inference Optimization | Experienced Hire

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

Job summary

Susquehanna International is seeking a Machine Learning Engineer focused on low-latency inference optimization to help build, tune, and productionize high-performance model serving systems. The role sits at the intersection of ML, systems engineering, and GPU performance, addressing latency, throughput, reliability, and hardware efficiency in production environments.

You will work closely with quantitative researchers and engineers to identify bottlenecks, evaluate frameworks, and shape

Qualifications

• Experience deploying ML inference in production.
• Proficient in Python and at least one systems language (C/C++, Rust, or Go).
• Strong understanding of ML frameworks (e.g., PyTorch) and performance profiling.

Responsibilities

• Design, build, and optimize low-latency inference systems for production ML workloads.
• Profile inference pipelines across execution, batching, memory, networking, and I/O.
• Evaluate and tune inference runtime systems and frameworks.
• Improve latency, throughput, and GPU utilization in production workloads.
• Build tooling for benchmarking and profiling model variants and deployments.
• Debug performance issues related to GPU memory, kernel behavior, and data movement.

Skills

Python

C/C++

Java

Rust/Go

PyTorch

CUDA

Kubernetes

Tools

PyTorch

CUDA

Triton

Kubernetes

Job description

Overview We are looking for a Machine Learning Engineer focused on low-latency inference optimization to help build, tune, and productionize high-performance model serving systems. This role sits at the intersection of machine learning, systems engineering, and GPU performance. You will work on inference workloads where latency, throughput, reliability, and hardware efficiency all matter, and where a deep understanding of modern inference runtimes can meaningfully improve production outcomes.

Overview We are looking for a Machine Learning Engineer focused on low-latency inference optimization to help build, tune, and productionize high-performance model serving systems. This role sits at the intersection of machine learning, systems engineering, and GPU performance. You will work on inference workloads where latency, throughput, reliability, and hardware efficiency all matter, and where a deep understanding of modern inference runtimes can meaningfully improve production outcomes.

We will work closely with quantitative researchers and engineers to understand model structure, identify inference bottlenecks, and turn research ideas into efficient production systems. The work may involve other types of models, but focuses on transformer-style architectures, and structured inference workloads. You will evaluate and tune frameworks and related serving or compilation systems, while also reasoning about GPU execution, memory layout, batching strategies, precision tradeoffs, and end-to-end latency.

What you'll do

• Design, build, and optimize low-latency inference systems for production machine learning workloads.
• Profile model inference pipelines across model execution, runtime configuration, batching, memory movement, serialization, networking, and I/O.
• Evaluate, integrate, and tune inference runtime systems.
• Improve latency, throughput, GPU utilization, for production inference workloads.
• Build and support benchmarking and profiling tools to compare model variants, hardware targets, runtime configurations, and deployment strategies.
• Debug performance issues involving GPU memory, compute saturation, kernel behavior, CPU/GPU coordination, data movement, and serving-layer overhead.
• Help shape model and system design choices so that research models are efficient to deploy under real latency constraints.
• Where necessary, collaborate with lower-level systems or GPU specialists on custom operators, kernel-level optimization, or hardware-specific performance work.

What 'We're Looking For

• Experience deploying, optimizing, or operating machine learning inference workloads in production or production-like environments.
• Programming experience in Python, Java, C# etc. and at least one systems language such as C, C++, Rust, or Go
• Solid understanding of modern ML frameworks such as PyTorch, including model execution, export, tracing, compilation, and performance profiling.
• Ability to reason about latency, throughput, batching, memory use, GPU utilization, and reliability under real workloads.
• Strong practical judgment around tradeoffs between model quality, latency, throughput, implementation complexity, and maintainability.

Preferred Qualifications

• Experience optimizing inference for latency-sensitive or high-throughput applications.
• Experience with model optimization techniques such as quantization, pruning, distillation, operator fusion, graph lowering, custom operators, or model compilation.
• Exposure to CUDA, Triton language, ROCm, PTX, CuTe, CUTLASS, FlashInfer, or similar low-level GPU programming tools.
• Experience running inference workloads on Kubernetes or GPU clusters, including scheduling, autoscaling, observability, and resource management.
• Background in mathematics, physics, computer science, engineering, statistics, quantitative finance, or another technical field.
• Demonstrated ability to improve real-world inference performance beyond a baseline framework implementation.