Machine Learning Intern

Job Description

About Central Technology – Machine Learning (CT-ML)

Central Technology – Machine Learning (CT-ML) is dedicated to enabling all ML Compute on ARM. They believe that advancing innovative ML technology requires a collaborative approach across hardware, software, and algorithms. The CT-ML Algorithms team acts as the Center of Excellence for ML content analysis, algorithm development, and tools, driving this unified effort forward.

Machine Learning Intern Job Description

As a Machine Learning Intern at CT-ML, you will be working on developing algorithms and optimization techniques to drive PPA (Performance, Power, Area) on current and future ARM platforms. This involves:

• Developing a deep understanding of ML workloads.
• Conducting ML workload analysis and serving as the ML Workload expert to support all phases of ARM IP development.
• Devising algorithms and prototyping on current platforms and supporting future architecture exploration.
• Developing internal tooling capabilities to support algorithmic and architecture exploration.

Specific examples of your work could include:

• Developing domain expertise in ML models and hardware-aware optimization techniques to support architecture development.
• Conducting “Proof of Concept” exploration to prototype algorithms and optimized models in support of architecture development.
• Developing and prototyping algorithms to automate deployment of HW-aware optimized network optimization processes on ARM processors.

Required Qualifications:

• Graduate student in Computer Engineering, Electrical Engineering, Computer Science or other related technical fields.
• 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.
• Willingness to learn and train large deep learning models on GPU-based systems.

Qualities that will help your application stand out:

• 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.
• Adaptability to the fast paced ML industry and willingness to learn new technology in a multifaceted environment.

What Arm offers to interns:

• Working on new projects with leaders in the field.
• Regular feedback and development opportunities.
• Social activities to connect with your peers.
• An end of internship celebration.
• The opportunity to be considered for future Graduate positions (subject to performance).
• A competitive salary and rewards package.
• Unparalleled learning and networking opportunities from the best in the industry.

Please note: Internships require enrollment in a higher education degree and returning to your course after the internship. If you are graduating on or before August 2025, you are not eligible for an Intern role, but you may be eligible for graduate roles.