Arm

Staff Engineer - AI Developer Tools and Platforms

Arm
Apply
27 days ago
Shanghai, ChinaStaff+

Responsibilities

  • Design, build, and maintain developer-facing APIs, libraries, tools, and platforms.
  • Provide technical leadership for AI developer tooling and help shape architecture, technical direction, and product roadmaps.
  • Improve developer workflows and experiences while optimizing performance, power, reliability, observability, and reproducibility.
  • Collaborate with engineering, product, UX, and developer teams to understand user needs and align technical decisions with product and ecosystem goals.
  • Drive software quality through testing, code review, maintainable implementations, and contributions to upstream and open-source projects where appropriate.

Requirements

  • Experience with Linux, containerisation, and programming languages such as Go, Python, Rust, or TypeScript.
  • Proven ability to deliver maintainable software through automated testing, continuous integration, effective engineering processes, and customer feedback.
  • Ability to navigate unfamiliar domains, identify improvement opportunities, and translate them into practical solutions.
  • Passion for building high-quality developer tools and intuitive AI workflows.
  • Experience with developer-facing CLIs, SDKs, APIs, or web tools is beneficial.
  • Knowledge of Arm architectures is beneficial.
  • Understanding of neural-network execution and inference runtimes, including computational graphs, operators, memory management, backend partitioning, and heterogeneous hardware execution, is beneficial.
  • Experience with hardware-aware model optimization, performance analysis, quantization, graph optimization, operator fusion, precision reduction, profiling, and debugging is beneficial.
  • Experience with PyTorch, ONNX, ONNX Runtime, ExecuTorch, TensorFlow, LiteRT, or OpenVINO is beneficial.

Benefits

  • Inclusive and highly collaborative working environment with opportunities to learn across products and technology stacks.
  • Opportunity to work on greenfield software products that ship with emerging Arm hardware on day one.
  • Hybrid working patterns determined by the team, with role-specific details shared during application.
  • If skilled worker sponsorship is required, Arm will cover employer sponsorship costs; individual visa application fees and the Immigration Health Surcharge remain the candidate’s responsibility.

Categories

Arm

About Arm

5,001-10,000 employees

Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.

Contact me