Senior AI Performance Engineer – GPU & DL

Jobtailor; California (MO); On-site; USD 180,000 - 280,000; Full time

Job summary

Jobtailor is seeking a senior AI performance engineer to analyze end-to-end performance of large-scale AI workloads across compute, network, storage, and software stacks. You will design rigorous performance studies and establish baselines, diagnose regressions, and quantify bottlenecks.

Collaborating with DL engineers, platform teams, and GPU architects, you will define benchmarks and communicate findings to influence system design and optimizations across the stack.

Qualifications

• BS or higher degree in computer science, computer engineering, or a related field (or equivalent experience).
• 12+ years of experience with strong programming skills in C++ and Python, able to build reliable analysis and automation workflows.
• Solid foundation in operating systems, computer architecture, and distributed systems.
• Experience with performance engineering, benchmarking, profiling, and optimization of complex software or systems.
• Ability to communicate technical findings, prioritize high-impact work, and align across teams.
• Experience analyzing large-scale AI clusters or distributed training/inference workloads.
• Experience with CUDA, GPU computing systems, and GPU performance analysis.
• Hands-on experience with deep learning frameworks such as PyTorch or JAX/XLA.
• Deep understanding of system-level performance analysis, workload characterization, and optimization.

Responsibilities

• Analyze end-to-end performance of large-scale AI workloads across compute, network, storage, and software stacks.
• Design and execute rigorous performance studies to establish baselines, diagnose regressions, and quantify bottlenecks.
• Define performance and efficiency evaluation methodologies, benchmarks, and success metrics for AI workloads.
• Use profiling, observability, and data analysis to turn performance measurements into actionable optimization plans.
• Partner with deep learning engineers, platform teams, and GPU architects to validate and deliver performance improvements.
• Communicate performance findings, trade-offs, and recommendations clearly to influence system and software design decisions.

Skills

C++ programming

Python programming

Performance engineering

Profiling

CUDA

GPU performance analysis

Distributed systems

System-level performance analysis

Data analysis

Automation workflows

PyTorch

JAX/XLA

Education

Bachelor's degree in CS/CE or related field

Tools

PyTorch

JAX/XLA

GPU Computing Systems

Job description

Jobtailor is seeking a senior AI performance engineer to analyze end-to-end performance of large-scale AI workloads across compute, network, storage, and software stacks. You will design rigorous performance studies and establish baselines, diagnose regressions, and quantify bottlenecks.

Collaborating with DL engineers, platform teams, and GPU architects, you will define benchmarks and communicate findings to influence system design and optimizations across the stack.