The Central Technology - Machine Learning (CT-ML) team at Arm aims to develop technology that enables all Machine Learning (ML) Compute on ARM platforms. This involves advancing ML technology through a collaborative approach across hardware, software, and algorithms, with the CT-ML Algorithms team focusing on ML content analysis, algorithm development, and tools.
Requirements
- Deep knowledge in machine learning, deep learning, and neural network design, optimization, and compression techniques.
- Overall high-level knowledge of computer architecture, systems, and HW-SW co-design.
- Ability to develop and work with large software systems in programming languages like Python.
- Knowledge of innovative deep learning libraries such as Tensorflow, and Pytorch.
- Willing to learn and train large deep learning models on GPU-based systems.
- Experience with ML model design, optimization, and HW-SW co-development methodology.
- ML Model Optimization techniques targeting PPA (Performance, Power, and Area) of neural networks on ARM compute platforms.
Responsibilities
- Develop in depth understanding of ML workload, and develop algorithms and optimization techniques to drive PPA (Performance, Power, Area) on current and future Arm platforms.
- Conduct ML workload analysis and serve as the ML Workload experts to support all phases of ARM IP development.
- Devise algorithm and prototyping on current platforms and to support future architecture exploration.
- Develop internal tooling capabilities to support algorithmic and architecture exploration
Other
- Are currently enrolled and studying towards an Electrical or Computer Engineering Degree (Masters or PhD students welcome). Candidates with alternative degrees will also be considered if they have relevant experience.
- Adaptability to the fast paced ML industry and willingness to learn new technology in a multifaceted environment.
- Arm Internships require you to be enrolled in a higher education degree and be returning to your course after your internship/placement.
- If you are graduating on or before August 2026 you will not be eligible for an Intern role, but you will be eligible for our graduate roles.
- This is a non-exempt hourly role which will be paid at an hourly rate based on the number of hours worked. Pay rates vary by location and educational level.