AI Accelerator Software Principal Engineer- Framework Integration

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. 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-we’dlove to have you apply! About The Role As an AI Accelerator Software Principal Engineer – Framework Integration, you will lead the design and delivery of high-performance, low-latency deep learning inference solutions. You’ll help advance Ampere’s AI software stack by integrating and optimizing popular deep learning frameworks so models can run efficiently across data center and edgeenvironments—meeting the performance and efficiency requirements of next-generation AI workloads. You will operate at the intersection ofsoftware engineering, performance engineering, and hardware-aware optimization, contributing to the full stack from model execution to accelerator-ready kernel performance. What You’ll Achieve: Framework integration for accelerator backends Integrate and optimize deep learning frameworks—such asPyTorch,ONNX, andllama.cpp—for theAmpere deep learning acceleratorbackend, enabling efficient and correct execution across a wide set of model types. End-to-end deep learning performance acceleration Go deep into the full software/hardware execution stack, including:inference serving and orchestrationframework integration layerscompiler and graph/runtime supportruntime libraries and user-mode execution pathscompute kernel developmentprofiling, benchmarking, and performance tuningModel enablement with quality and speed Improve bothperformance and accuracyfor models using popular frameworks, and ensure compatibility with serving ecosystems such asvLLMandSGLang—helping deliver production-ready inference behavior. Hardware/software co-design and optimization Partner with hardware and platform teams to co-optimize AI execution for better outcomes:increased throughputreduced latencyimproved scalabilitybetter resource utilization (compute/memory/IO)higher sustained performance under realistic workloadsBuild state-of-the-art AI software components Contribute to the development ofsoftware 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 intoAmpere’s cloud-native processor platforms and accelerators. About You: Education & 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 yearsCore framework experience: Strong experience building with or integratingAI frameworkssuch asPyTorch,llama.cpp, andONNX. Linux + accelerator/runtime expertise (preferred): Experience with developinguser-mode drivers, runtime libraries, or low-level integration forGPUs or deep learning acceleratorsinLinuxis a plus. Strong systems programming & performance skills:Expert inPythonandC/C++Strong background inperformance profiling and tuning(latency/throughput, memory behavior, kernel efficiency)Deep ML understanding: Solid understanding ofAI/ML conceptsincluding neural networks and data processing frameworks. Experience with modern deep model architectures such asTransformersandDiffusionmodels is preferred. Modern AI tooling fluency (preferred):Fluent with modern AI programming tools such asCodexorClaude 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. 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.