AI Performance Engineer

About the role

Location: 100% Remote (U.S.). Position Type: Full-time, Direct W2. Sponsorship: U.S. Citizens, Green Card Holders, EAD Holders, and H-1B transfer candidates are encouraged to apply; we are unable to sponsor new H-1B visa petitions for this position.

Responsibilities

• Focus on extracting maximum throughput, minimizing latency, and reducing cost across training and inference workloads for large neural network systems.
• Work across the full stack from low‑level kernel optimization to distributed system tuning, requiring deep understanding of GPU architecture, model parallelism, memory management, and compiler‑level optimization.
• Collaborate with cross‑functional partners (product, design, engineering, operations, business) to translate ambiguous requirements into well‑engineered solutions.
• Raise the bar through code review, design review, and mentorship of junior engineers.

Required Qualifications

• Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related field.
• Six or more years of experience in performance engineering, ML systems, or HPC.
• Strong proficiency in Python and C++.
• Hands‑on experience optimizing deep learning workloads on modern GPUs.
• Deep understanding of distributed training and inference techniques.
• Experience with profiling tools across CPU, GPU, and distributed systems.
• Familiarity with model compression techniques and their accuracy implications.
• Strong grasp of memory hierarchies, communication primitives, and parallelism strategies.
• Excellent measurement, debugging, and analytical reasoning skills.
• Strong communication and collaboration skills.

Preferred Qualifications

• Experience optimizing LLM inference at production scale.
• Contributions to vLLM, TensorRT‑LLM, DeepSpeed, or similar projects.
• Familiarity with custom kernel authoring in Triton or CUTLASS.
• Experience with FinOps for AI workloads.
• Publications or talks on AI systems performance.

Pay

Salary Range: $75,000–$100,000 Annually.