Senior AI Engineer

Job Overview:
The PE AI Transformation team is redefining how AI powers engineering at Arm! We partner with engineering teams across Arm to deliver real-world impact through applied AI-boosting productivity, fueling innovation, and making powerful AI tools part of every engineer's workflow. Guided by our vision-engineering redefined through smarter workflows, faster execution, and greater creativity-we're building systems that let technology amplify human potential.
As an AI Engineer, you'll be hands-on in designing, developing, and deploying AI capabilities that reshape how engineers work. You'll learn from experienced AI architects and senior engineers, gaining exposure to sophisticated system design and scaling patterns, while supplying your own technical depth to deliver robust, high-quality solutions. You'll also mentor junior peers and share practical insights gained through your work.
Responsibilities:

• Deliver end-to-end AI solutions, from proof-of-concept to production, that improve developer workflows and engineering productivity.
• Embed with partner teams to find opportunities, translate needs into AI solutions, and deliver tangible, measurable results.
• Implement and optimize LLM-based systems, including retrieval-augmented generation (RAG), evaluation, and guardrails.
• Build and refine agentic AI components such as planning, memory, and tool orchestration.
• Write production-ready, testable Python code, maintaining high standards for quality, security, and performance.
• Collaborate with and learn from senior engineers and architects, applying mentorship feedback to continuously improve design, code quality, and scalability awareness.
• Supply to shared components and documentation, helping establish reusable patterns and frameworks for future projects.
• Provide guidance to less-experienced team members, reviewing code and sharing knowledge through demos or pair programming.
• Stay ahead of emerging AI tools and frameworks, evaluating their potential to accelerate Arm's engineering transformation.

Required Skills and Experience:

• Experience in AI, ML, or software engineering, with validated delivery of production-grade solutions.
• Strong Python programming skills
• Experience deploying AI or ML models into production systems.
• Experience with LLM systems, including prompting, RAG, evaluation, and orchestration.
• Exposure to agentic AI frameworks (e.g., LangChain, LlamaIndex, or custom orchestration stacks).
• Familiarity with enterprise AI APIs (e.g., OpenAI, Anthropic, Azure OpenAI, or similar).
• Experience working with retrieval systems, prompt engineering, and cloud platforms (AWS, Azure, or GCP - AWS preferred).
• A learning-oriented mindset-open to feedback, eager to understand architectural trade-offs, and committed to refining your craft.
• Strong communication and collaboration skills, with the ability to partner effectively across disciplines.

Nice to have skills and experience:

• Experience in semiconductor or large-scale software environments.
• Familiarity with MLOps, data pipelines, or AI infrastructure.
• Contributions to open-source or technical communities.
• Experience using AI to improve developer workflows, code generation, or data visualization.

In Return:
In return, you'll join a pioneering team where AI transforms how engineers create and collaborate. You'll work alongside world-class experts, learning from and supplying to real-world AI systems that enhance productivity and creativity across Arm. Expect opportunities to grow your technical depth, shape innovative solutions, and see your work deliver tangible impact-within an inclusive, forward-thinking culture that values curiosity, teamwork, and continuous learning.
Our 10x mindset guides how we engineer, collaborate, and grow. Understand what it means and how to reflect 10x in your work: https://careers.arm.com/en/10x-mindset
#LI-ZA
Salary Range:
$156,500-$211,700 per year
We value people as individuals and our dedication is to reward people competitively and equitably for the work they do and the skills and experience they bring to Arm. Salary is only one component of Arm's offering. The total reward package will be shared with candidates during the recruitment and selection process.
Accommodations at Arm
At Arm, we want to build extraordinary teams. If you need an adjustment or an accommodation during the recruitment process, please email [email protected] . To note, by sending us the requested information, you consent to its use by Arm to arrange for appropriate accommodations. All accommodation or adjustment requests will be treated with confidentiality, and information concerning these requests will only be disclosed as necessary to provide the accommodation. Although this is not an exhaustive list, examples of support include breaks between interviews, having documents read aloud, or office accessibility. Please email us about anything we can do to accommodate you during the recruitment process.
Hybrid Working at Arm
Arm's approach to hybrid working is designed to create a working environment that supports both high performance and personal wellbeing. We believe in bringing people together face to face to enable us to work at pace, whilst recognizing the value of flexibility. Within that framework, we empower groups/teams to determine their own hybrid working patterns, depending on the work and the team's needs. Details of what this means for each role will be shared upon application. In some cases, the flexibility we can offer is limited by local legal, regulatory, tax, or other considerations, and where this is the case, we will collaborate with you to find the best solution. Please talk to us to find out more about what this could look like for you.
Equal Opportunities at Arm
Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. We are a diverse organization of dedicated and innovative individuals, and don't discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.