ML Performance Engineer: GPU/CUDA at Scale

Selby Jennings; Chicago (IL); On-site; USD 140,000 - 210,000; Full time

Job summary

Selby Jennings in Chicago seeks a Performance Engineer to develop scalable ML infrastructure for training and inference at scale. You will focus on low-level optimization and high-performance data pipelines, enabling rapid experimentation and model deployment.

You should bring strong CUDA/C++ and Python skills, plus hands-on experience with PyTorch, TensorFlow, or JAX. Join a team at the intersection of finance and technology, driving innovation.

Qualifications

• BS or MS in CS or related field; strong fundamentals in algorithms and systems.
• Proven experience in GPU-accelerated ML workloads and optimization.
• Experience with deep learning frameworks and performance profiling.

Responsibilities

• Build scalable pipelines for deep learning training and inference.
• Optimize and extend ML frameworks for performance and functionality.
• Collaborate with researchers to accelerate experimentation and deployment.

Skills

CUDA

C++

Python

PyTorch

TensorFlow

JAX

Education

B.S. or M.S. in Computer Science

Job description

Selby Jennings in Chicago seeks a Performance Engineer to develop scalable ML infrastructure for training and inference at scale. You will focus on low-level optimization and high-performance data pipelines, enabling rapid experimentation and model deployment.

You should bring strong CUDA/C++ and Python skills, plus hands-on experience with PyTorch, TensorFlow, or JAX. Join a team at the intersection of finance and technology, driving innovation.