The Central Technology ML (CTML) team at Arm is looking to pioneer and implement the technology roadmap for machine learning computing on Arm architecture, requiring joint development across hardware, software, and ML algorithms.
Requirements
- Deep knowledge in machine learning, deep learning, neural network design, and optimization
- Proficient in computer architecture, basic knowledge of HW and SW design
- Developing and working with large software systems in programming languages like Python
- Knowledge of cutting-edge deep learning libraries such as TensorFlow and PyTorch
- Training large deep learning models on powerful 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 an in-depth understanding of current and future ML workloads on ARM compute platforms with a focus on their PPA (performance, power, and area)
- Architecture exploration necessitates the joint development of hardware, software, and algorithms
- Develop hardware-aware optimization algorithms that can make state-of-the-art networks on Arm platforms, including CPU/GPU/NPUs
- Algorithm innovation and prototyping to drive and validate architectural features
- Conducting in-depth analyses of complex neural networks on ARM's computing platforms
- Devising algorithms that will elevate us to the forefront of the future
- Supply to this joint development initiative
Other
- 3+ years working in a ML Algorithm Development and/or Optimization environment
- A PhD or Master’s degree in Computer Engineering, Electrical Engineering, Computer Science, or other related technical fields
- Take daily responsibility to make the Global Arm community thrive
- Be non-traditional not egocentric
- Recognize the power in saying ‘I don’t know’
