15 days ago
Lund, SwedenEntry Level
Responsibilities
- Analyze current and future ML/AI workloads on Arm computing platforms with a focus on performance, power, and area.
- Track ML/AI technology trends and analyze use cases to guide workload support across Arm IP development.
- Develop hardware-aware optimization algorithms for AI models running on Arm CPUs, GPUs, and NPUs.
- Build internal tooling to support algorithmic and architecture exploration.
- Develop understanding of the machine learning stack from models through hardware implementation.
Requirements
- Hold, or be on track to hold by the programme start date, a Master’s or PhD in Computer Engineering, Electrical Engineering, Computer Science, or a related technical field.
- Have experience developing and working with software systems, including Python, and knowledge of hardware and software design.
- Have experience training and deploying ML/AI models using deep learning libraries such as PyTorch or TensorFlow.
- Demonstrate strong technical communication skills.
- ML/AI project experience in industry or academia, compute-kernel optimization with CUDA or Triton, and hardware-aware model performance optimization are desirable.
Benefits
- Relocation support is available for domestic moves within the country and regional moves within Europe.
- Visa sponsorship support is available for candidates who require it.
- Hybrid working arrangements are determined by the team, with role-specific details provided during the application process.
- Graduate development includes Grad-teach-Grads workshops, guest speaker sessions, social events, and participation in the Graduate Committee.
- Attendance at Arm’s annual Global Graduate Conference.
- Formal mid-year and year-end progression reviews and opportunities to progress within the salary range.
Categories
About Arm
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.