Senior Performance Engineer

Please Note: To provide the best candidate experience amidst our high application volumes, each candidate is limited to 10 applications across all open jobs within a 6-month period.  Advancing the World’s Technology Together Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you’ll have an opportunity to be part of a global leader whose innovative designs are pushing the boundaries of what’s possible and powering the future.  We believe innovation and growth are driven by an inclusive culture and a diverse workforce. We’re dedicated to empowering people to be their true selves. Together, we’re building a better tomorrow for our employees, customers, partners, and communities. The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads while minimizing energy consumption and maximizing performance. To achieve this goal, we collaborate closely with both hardware and software engineers to identify and address the unique challenges posed by AI/ML workloads and to explore new computing abstractions that can provide a better balance between the hardware and software components of our systems. Additionally, we continuously conduct research and development in emerging technologies and trends across memory, computing, interconnect, and AI/ML, ensuring that our platforms are always equipped to handle the most demanding workloads of the future. By working together as a dedicated and passionate team, we aim to revolutionize the way AI/ML applications are deployed and executed, ultimately contributing to the advancement of AGI in an affordable and sustainable manner. Join us in our passion to shape the future of computing! This role is offered by the STG group within the AGI Lab as part of DSRA. We are a systems research and engineering team working at the intersection of large language models, accelerator hardware, and high-performance software. Our mission is to design, prototype, and optimize next-generation AI systems through tight hardware–software co-design. Our team works hands-on with cutting-edge accelerator hardware, advanced memory systems, and large-scale distributed AI infrastructure. We develop and optimize the software stack required to maximize performance, efficiency, and scalability for modern and emerging LLM workloads. We are seeking a Senior LLM Systems Performance Engineer to build representative AI environments, characterize emerging workloads, and drive performance analysis for next-generation AI platforms. In this role, you will set up and operate realistic LLM serving and agentic AI environments, collect workload traces and performance data, and develop methodologies to characterize workload behavior. You will analyze system bottlenecks across compute, memory, communication, and scheduling resources, and evaluate how emerging workloads interact with AI accelerator architectures and system infrastructure. The ideal candidate combines hands-on experience building large-scale AI systems with strong performance engineering skills and a solid understanding of AI accelerator architecture. You should be comfortable working across the full stack—from application frameworks and serving systems to runtime software, networking, memory systems, and accelerator hardware. You will work closely with hardware architects, systems engineers, and software researchers to understand the performance implications of emerging workloads such as agentic AI, long-context reasoning, disaggregated inference, and Mixture-of-Experts models. Your analysis will help shape future hardware–software co-design decisions and guide the development of next-generation AI infrastructure. Location: Daily onsite presence at our San Jose, CA office / U.S. headquarters in alignment with our Flexible Work policy. What You’ll Do Build and operate representative AI environments, including agentic workflows, distributed inference systems, disaggregated serving architectures, and MoE deployments. Collect workload traces, telemetry, and performance data from real-world AI applications; characterize workload behavior, develop representative benchmarks, and identify performance bottlenecks across compute, memory, communication, and scheduling resources. Evaluate AI systems across the full hardware and software stack, and analyze the impact of runtime, memory hierarchy, interconnect, and accelerator architecture on application performance. Collaborate with hardware and software teams to drive performance analysis, architecture exploration, and hardware–software co-design for next-generation AI platforms. What You Bring MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field. B.S with 5+ years of experience in performance engineering, AI systems, distributed systems, high-performance computing, or a related area. MS in Computer/Electrical Engineering or Computer Science with 3+ years of relevant working experience or PhD and 0+ years of relevant working experience preferred. Strong understanding of LLM inference and training systems. Strong understanding of NVIDIA GPU architecture and performance characteristics, including compute, memory hierarchy, communication, and system-level bottlenecks. Hands-on experience profiling and optimizing AI workloads on NVIDIA GPU platforms using tools such as Nsight Systems, Nsight Compute, and related performance analysis frameworks. Experience analyzing performance of large-scale distributed AI workloads. Proficiency in Python and C++. Experience with one or more modern AI frameworks or serving systems, such as PyTorch, vLLM, SGLan