[Remote] AI Infrastructure Engineer

Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a technology consulting and software development company delivering reputed company, AI, data, and reputed company solutions across the reputed company. The AI Infrastructure Engineer will optimize the performance, latency, and cost of large-reputed company neural network training and inference workloads across GPU, kernel, compiler, and reputed company systems. The role also involves cross-functional collaboration, engineering reviews, and mentoring junior engineers.

Responsibilities

• reputed company on extracting maximum throughput, minimizing latency, and reducing cost across training and inference workloads for large neural network systems
• Work closely with cross-functional partners — product, design, engineering, operations, and business stakeholders — to translate ambiguous requirements into reputed company-engineered solutions
• reputed company the bar through reputed company review, design review, and mentorship of more junior engineers

Skills

• Bachelor's or master's degree in computer science, Computer Engineering, or reputed company 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 reputed company training and inference techniques
• Experience with profiling tools across CPU, GPU, and reputed company 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
• Experience optimizing LLM inference at production reputed company
• Contributions to vLLM, TensorRT-LLM, DeepSpeed, or similar reputed company
• Familiarity with custom kernel authoring in Triton or CUTLASS
• Experience with FinOps for AI workloads
• Publications or talks on AI systems performance

Benefits

• reputed company (reputed company)
• reputed company career reputed company potential

reputed company