Ampere Computing

AI Accelerator Software Principal Engineer - NPU Full-Stack Integration

Ampere Computing$182K — $273K *
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • BS in Computer Science, Engineering, or related field and 8 years of relevant experience; or MS with 6 years; or PhD with 3 years.
  • Strong understanding of AOT (Ahead-Of-Time) compilation in frameworks like PyTorch.
  • Experience with Linux and developing user-mode drivers for deep learning accelerators is a plus.
  • Proficient in Python and C/C++, with strong performance profiling and tuning skills.
  • Solid understanding of AI/ML concepts and experience with modern architectures like Transformers.

Responsibilities

  • Lead the design of low-latency deep learning inference solutions.
  • Enhance model performance and accuracy in edge devices using popular frameworks.
  • Collaborate with hardware teams for AI execution optimization.
  • Develop reusable software components and libraries for AI accelerators.
  • Integrate AI capabilities into cloud-native platforms through cross-functional teamwork.

Benefits

  • Premium medical, dental, and vision insurance with income protection and a 401K plan.
  • Unlimited Flextime and over 10 paid holidays for work-life balance.
  • Access to a variety of healthy snacks and beverages to enhance wellbeing.
Full Job Description
About the Role

As an AI Accelerator Software Principal Engineer - NPU Full-Stack Integration, you will lead the design and delivery of high-performance, low-latency deep learning inference solutions on the Arm® Ethos™-U85. You'll help advance Ampere's AI software stack by enabling models with performance and efficiency requirements

You will operate at the intersection of software engineering, performance engineering, and hardware-aware optimization, contributing to the full stack from model execution to accelerator-ready kernel performance.

What You'll Achieve:

  • End-to-end deep learning performance acceleration
    Go deep into the full software/hardware execution stack, including:
    • framework integration layers
    • compiler and graph/runtime support
    • runtime libraries and user-mode execution paths
    • compute kernel development
    • profiling, benchmarking, and performance tuning
  • Model enablement with quality and speed
    Improve both performance and accuracy for models using popular frameworks, helping deliver production-ready inference behavior in edge devices.
  • Hardware/software co-design and optimization
    Partner with hardware and platform teams to co-optimize AI execution for better outcomes:
    • increased throughput
    • reduced latency
    • improved scalability
    • better resource utilization (compute/memory/IO)
    • higher sustained performance under realistic workloads
  • Build state-of-the-art AI software components
    Contribute to the development of software and hardware AI co-processors/accelerators, delivering reusable libraries, optimized execution paths, and robust integration with existing tooling.
  • Cross-functional collaboration
    Work closely with cross-functional teams (compiler/runtime, kernels, platform, and product engineering) to integrate AI capabilities into Ampere's cloud-native processor platforms and accelerators.

About You:

  • Education and Experience: BS Computer Science, Computer Engineering, Electrical Engineering, or Software Engineering or related technical field & 8 years of related experience; or MS degree & 6 years; or PhD & 3 years.
  • Understands AOT (Ahead-Of-Time) compilation path in popular frameworks like PyTorch and deployment path like execuTorch in edge environment
  • Linux + accelerator/runtime expertise (preferred): Experience with developing user-mode drivers, runtime libraries, or low-level integration for GPUs or deep learning accelerators in Linux is a plus.
  • Strong systems programming & performance skills:
    • Expert in Python and C/C++
    • Strong background in performance profiling and tuning (latency/throughput, memory behavior, kernel efficiency)
  • Deep ML understanding: Solid understanding of AI/ML concepts including neural networks and data processing frameworks. Experience with modern deep model architectures such as Transformers and Diffusion models is preferred.
  • Modern AI tooling fluency (preferred): Fluent with modern AI programming tools such as Codex or Claude Code, and comfortable accelerating development workflows.
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 $182,000 and $273,000, except in the San Francisco Bay Area where the range is between $195,000 and $292,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.

#LI-Hybrid

#LI-Hybrid #LI-DR

#LI-Hybrid

About Ampere Computing

Ampere Computing is a semiconductor company that designs and manufactures high-performance processors for cloud and edge computing. The company's processors are based on the Arm architecture and are optimized for power efficiency and performance. Ampere Computing was founded in 2017 by former Intel president Renee James and is headquartered in Santa Clara, California.
Learn more about Ampere Computing
Size
200 employees
Industry
Founded
2017

Similar Jobs

More Jobs at Ampere Computing

More Consumer Technology Jobs

Find similar AI Accelerator Software Principal Engineer - NPU Full-Stack Integration jobs: