AI Accelerator, Software Engineer- Graph Optimization/Compilers

Ampere; Santa Clara (CA); On-site; USD 159,000 - 239,000; Full time

Benefits offered by this job

Health insurance

401K retirement plan

Unlimited flextime

Job summary

Ampere is seeking a Software Engineer for its AI Accelerator team to optimize deep learning graphs and maximize performance on Ampere hardware. You will work across the software stack from model frameworks to kernels, contributing to graph optimizations and runtime improvements.

You will collaborate with compiler, runtime, and hardware teams, applying CUDA, ROCm, OpenCL, SYCL, or Triton expertise to improve transformer and LLM workloads while ensuring energy-efficient, scalable inference.

Qualifications

• Bachelor’s degree in a technical field and 3–5 years of relevant experience, or a Master’s with 3+ years.
• Strong foundations in algorithms, data structures, and systems programming.
• Experience with Python and C/C++ in real-world projects or research.
• Familiarity with deep learning concepts, transformer models, and GPU/accelerator architectures.

Responsibilities

• Optimize deep learning computational graphs for performance, throughput, and energy efficiency on Ampere accelerators.
• Develop graph-level optimizations such as fusion, memory planning, and quantization.
• Analyze end-to-end performance across frameworks, runtimes, and hardware; build profiling infrastructure.
• Collaborate with compiler, runtime, kernel, architecture, and hardware teams on co-design.

Skills

Python

C/C++

Graph algorithms

Systems programming

Profiling

Education

Bachelor's degree in Computer Science / Computer Engineering or related field

Tools

CUDA

ROCm

OpenCL

SYCL

Triton

Job description

Invent the future with us.

Ampere is a semiconductor design company for a new era, leading the future of computing with an innovative approach to CPU design focused on high-performance, energy efficient AI compute.

Description

As a pioneer in the new frontier of energy efficient high-performance computing, Ampere is part of the Softbank Group of companies driving sustainable computing for AI, Cloud, and edge applications.

Join us at Ampere and work alongside a passionate and growing team -

About The Role

As a Software Engineer on Ampere’s AI Accelerator team, you will optimize deep learning computational graphs to maximize the performance, efficiency, and scalability of Ampere’s AI accelerator hardware. You will work across the software stack, from model frameworks and inference-serving systems to graph optimization, compiler infrastructure, runtimes, and compute kernels.

What You’ll Achieve

• Optimize computational graphs for performance, throughput, latency, memory efficiency, and power efficiency on Ampere AI accelerators
• Enable and optimize models, frameworks, and inference platforms, including PyTorch, Llama.cpp, vLLM, and SGLang
• Develop graph-level optimizations such as operator fusion, pattern matching, redundancy elimination, constant folding, layout optimization, memory planning, quantization, and accelerator offload
• Optimize transformer and LLM workloads, including dynamic shapes, attention mechanisms, KV-cache management, and mixed-precision execution
• Analyze end-to-end performance across frameworks, compilers, runtimes, kernels, and hardware
• Build profiling, benchmarking, validation, and performance-regression infrastructure
• Identify bottlenecks using traces, compiler diagnostics, microbenchmarks, and hardware performance data
• Collaborate with compiler, runtime, kernel, architecture, hardware, and applications teams on hardware/software co-design
• Contribute to architecture, design reviews, code reviews, documentation, and engineering best practices

About You

• Bachelor’s degree in Computer Science, Computer Engineering, Mathematics, or a related technical field & 5 years of relevant experience; or a Master’s degree with & 3 years of relevant experience
• Strong foundations in algorithms, data structures, graph algorithms, computational complexity, and systems programming
• Proficiency in Python and C/C++, demonstrated through internships, research, coursework, open-source contributions, or personal projects
• Strong ability to reason about execution dependencies, memory movement, numerical correctness, and hardware execution behavior
• Experience diagnosing performance issues through profiling, benchmarking, tracing, or hardware-level analysis is a plus
• Familiarity with deep learning concepts, neural-network architectures, tensor operations, numerical precision, quantization, and memory layout
• Experience with CUDA, ROCm, OpenCL, SYCL, Triton, GPU programming, NPU programming, or other accelerator architectures is a plus
• Familiarity with transformer models, LLM inference, attention mechanisms, KV-cache optimization, speculative decoding, mixed-precision execution, or sparsity is a plus
• Demonstrated exceptional problem-solving ability—IOI medal, ACM ICPC medal, Codeforces Grandmaster, USACO Platinum, or equivalent achievement in research or production engineering is a strong plus
• Strong analytical and debugging skills, with the ability to investigate ambiguous technical problems and deliver robust solutions
• Fast learner who can quickly understand new architectures, frameworks, compilers, and workloads
• Experience using AI-assisted development tools to accelerate implementation, testing, debugging, and code review while maintaining technical ownership and code quality

What We’ll Offer

At Ampere we believe in taking care of our employees and providing a competitive total rewards package that includes base pay, cash long-term incentive, and comprehensive benefits. The full base pay range for this role is between $159,000 and $239,000. Our benefits include health, wellness, and financial programs that support employees through every stage of life.

Benefit Highlights Include

• Premium medical insurance, dental insurance, vision insurance, as well as income protection and a 401K retirement plan, so that you can feel secure in your health and financial future.
• Unlimited Flextime and 10+ paid holidays so that you can embrace a healthy work-life balance.
• A variety of healthy snacks, energizing espresso, and refreshing drinks to keep you fueled and focused throughout the day.

And there is much more than compensation and benefits. At Ampere, we foster an inclusive culture that empowers our employees to do more and grow more. We are excited to share more about our career opportunities with you through the interview process. Our benefits include health, wellness, and financial programs that support employees through every stage of life.

Ampere is an inclusive and equal opportunity employer and welcomes applicants from all backgrounds. All qualified applicants will receive consideration for employment without regard to race, color, national origin, citizenship, religion, age, veteran and/or military status, sex, sexual orientation, gender, gender identity, gender expression, physical or mental disability, or any other basis protected by federal, state or local law.