Senior / Staff Software Engineer (AI Platforms)

Job Title: Senior / Staff Software Engineer (AI Platforms) Company: Arm Job Category: Python Developer Roles & Responsibilities: We are an AI\-native engineering team. We actively use AI\-assisted development tools and agentic workflows throughout the software development lifecycle, while maintaining high standards for correctness, testing, and software quality. Responsibilities -------------------- * Design and implement distributed infrastructure for AI inference, including model serving, request routing, workload scheduling, and resource management. * Build reliable, high\-performance software, solving problems involving concurrency, networking, scalability, data consistency, and failure recovery. * Own software capabilities end to end, from evaluating design options and writing code to testing, deployment, debugging, and ongoing improvements. * Investigate system bottlenecks through profiling, benchmarking, and diagnostics. Improve latency, throughput, resource efficiency, and reliability. * Use and extend AI\-assisted development tools and agentic workflows to improve engineering productivity, with appropriate testing, verification, and code review. * Collaborate with engineers across Arm on architecture, interfaces, and technical decisions. Contribute to code reviews, engineering practices, and the development of a growing team. * 5\+ years of software engineering experience building complex software systems, with demonstrated technical depth, ownership, and impact. * Strong programming skills in C\+\+, Go, Rust, Python, or comparable languages, with solid foundations in algorithms, data structures, and software design. * Strong understanding of distributed systems, including concurrency, networking, service interfaces, state management, and fault tolerance. * Experience developing and debugging software on Linux, cloud infrastructure, or container\-based environments, with an understanding of system performance and reliability. * Hands\-on experience with AI\-assisted development tools, including evaluating, debugging, and validating AI\-generated code. * Strong problem\-solving skills, sound technical judgment, and the ability to work independently through unfamiliar or ambiguous engineering problems. * We are considering candidates at both Senior and Staff levels. Senior engineers should demonstrate strong technical execution and ownership of complex components. Staff engineers should also demonstrate the ability to lead significant technical designs, make architectural tradeoffs, and influence engineering decisions across teams. * Experience with AI inference infrastructure, LLM serving, distributed compute, or ML platforms, including batching, caching, or model execution. * Experience with Kubernetes, distributed scheduling, resource orchestration, or infrastructure that manages workloads across compute environments. * Experience with performance profiling, benchmarking, distributed tracing, and diagnosing complex system behavior. * Experience building developer tools, automated testing infrastructure, CI/CD systems, or agentic development workflows. * Prior AI inference experience is valuable but not required for engineers with strong distributed systems backgrounds. You’ll have opportunities to build Arm’s next\-generation AI inference infrastructure from the ground up, take ownership of significant platform capabilities, and collaborate with engineers across Arm. The role offers scope to influence technical design, deepen your expertise in distributed systems and AI\-native development, and grow your technical impact while tackling complex challenges in performance, scalability and reliability. 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. Work Location: KA, India Availability: Full time

Required Skills and Experience ---------------------------------- Nice\-to\-Have Skills and Experience ---------------------------------------- At Arm, we embrace the benefits AI can bring, including in recruitment. You may use AI to support your preparation, such as improving the clarity of written responses, practicing interview questions, or learning more about our industry. However, your application and interview responses must reflect your own thinking, experience, and abilities. We want to understand your authentic perspective, creativity, and approach to problem solving. As such, you may not use AI to complete assigned tasks as part of any assessments. If you plan to use AI\-supported or assistive technologies as part of a reasonable adjustment, please contact accommodations@arm.com so we can support you. Key Skills: Python, Data Structures, Algorithms, Linux Salary: Not disclosed