Arm

Machine Learning Software Engineer

Arm
Apply
2 months ago
Lund, SwedenMid Level / Senior

Responsibilities

  • Develop and optimize C++ and Python software for machine learning workloads.
  • Work on ML compilation, PyTorch integration, runtimes, drivers, performance analysis, and embedded software.
  • Analyze model execution to improve software performance and efficiency on Arm compute platforms.
  • Contribute to technical design and take ownership of larger engineering challenges.
  • Support the development of other engineers and collaborate with ML software and hardware engineers.

Requirements

  • Professional C++ development experience with complex software systems.
  • Experience with machine learning frameworks or tooling such as PyTorch.
  • Python experience for software development, ML tooling, testing, or automation.
  • Experience debugging, analyzing performance, and solving complex software problems.
  • Experience with collaborative software development, source control, code review, testing, and continuous integration.
  • Knowledge of ML compilers, PyTorch internals, runtimes, drivers, embedded software, performance optimization, or hardware acceleration is valuable but not required across every area.

Benefits

  • Continuous learning and career development opportunities.
  • Collaborative working environment and comprehensive employee benefits.
  • Hybrid working patterns determined by the team, with role-specific details provided during the application process.
  • Recruitment adjustments and accommodations are available upon request.

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