Software Engineer – AI Inference Engine

Job description preview

From FriendliAI’s original posting

ABOUT THE JOB We are seeking a highly technical Inference Engine Engineer to optimize the performance and efficiency of our core inference engine. In this role, you will focus on designing, implementing, and optimizing GPU kernels and supporting infrastructure for next-generation generative and agentic AI workloads. Your work will directly power the most latency-critical and compute-intensive systems deployed by our customers. We are looking for an exceptional engineer with a strong foundation in GPU programming and compiler infrastructure. The ideal candidate enjoys pushing performance boundaries and has experience supporting production-scale machine learning applications. KEY RESPONSIBILITIES - Design and optimize custom GPU kernels for AI (e.g., transformer and diffusion) workloads - Contribute to the development of FriendliAI’s kernel compiler, memory planner, runtime, and other core components. - Collaborate with cloud and infrastructure engineers to ensure end-to-end inference performance - Analyze performance bottlenecks across the software and hardware stack, and implement targeted optimizations - Drive support for new model architectures and tensor compute patterns -…