Graduate Machine Learning Engineer - Central Technology

About the Position

The mission of the Central Technology Machine Learning (CTML) team is to pioneer and implement the technology roadmap that paves the way for the future of AI computing on Arm architecture. We believe that innovative AI technology necessitates joint development across hardware, software, and ML algorithms. The CTML Content and Algorithm team plays a key role within the Centre of Excellence for Machine Learning. We actively support this joint development initiative, conducting in-depth analyses of complex AI models on Arm computing platforms and devising algorithms that will make AI run better on Arm.

What You Could Be Doing as a Graduate

Develop in-depth understanding of current and future ML/AI workload on Arm compute platforms with a focus on their PPA (performance, power, and area).

• Understand and follow the latest technology trends in ML/AI, collecting and analyzing these use cases to drive the direction of ML/AI workload support in all phases of Arm IP development.
• Develop hardware-aware optimization algorithms that can make state-of-the-art AI models run on Arm platforms including CPU/GPU/NPUs.
• Develop internal tooling capabilities to support algorithmic and architecture exploration.

Requirements

• Hold (or are on track to hold by the programme start date) a Master’s or PhD in Computer Engineering, Electrical Engineering, Computer Science, or other related technical fields.
• Experience in developing and working with software systems in programming languages like Python, with good knowledge of Hardware and Software design.
• Experience in training and deploying ML/AI models using deep learning libraries such as PyTorch or TensorFlow.
• Strong technical communication skills.

Nice to Have

• Demonstrated ML/AI projects experience in industry or academia.
• Experience with optimizing compute kernels (e.g., CUDA, Triton).
• Hardware-aware performance optimization of ML/AI models.
• Passion and adaptability to the fast-moving AI industry and willingness to learn new technology in a dynamic environment.
• Good knowledge of the whole machine learning stack from model to hardware implementation.

Additional Information

We encourage early applications as we review them on a first come/first served basis. Please provide an academic transcript as part of your application.

Please note that relocation support is available for this role for domestic moves within the country and regional moves within Europe. Visa sponsorship support is also available for candidates who require it.

We Offer

You’ll receive a starting salary of SEK 545,000 (Bachelor level) or SEK 555,000 (Master level), alongside a comprehensive rewards package. As you develop in role, you’ll have clear opportunities to progress within the salary range at Arm. With mid-year and year-end reviews, you’ll have two formal progression windows each year, recognizing your growth.

You can also attend our “Grad-teach-Grads” workshops, guest speaker series, and social events to expand your knowledge. If you’d like to get more involved, you can join the Graduate Committee to further develop your planning, networking, and coordination skills.

You’ll be invited to our Global Graduate Conference (GGC) - an annual event where graduates hear from executive leaders, guest speakers, and build lasting connections.

With many of our leaders and technical specialists having originally joined as a Graduate, will you be next?

About the Company

At Arm, we believe progress happens when people are empowered to think bigger and push beyond what seems possible. Our 10x mindset is about curiosity, ambition, and creating impact that will be used by millions. We learn fast, prioritize collaboration, and turn bold ideas into real technology. We look for people who are inspired by this way of working and want to grow in an environment where bold ideas are welcomed. Discover how the 10x mindset shapes life at Arm: 10x Mindset

Equal Opportunities

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.

Salary Range

478,700 kr - 647,700 kr per year