Graduate Machine Learning Engineer - Central Technology
Core
Develop hardware-aware optimization algorithms and internal tooling to enable state-of-the-art AI models to run efficiently on Arm computing platforms (CPU/GPU/NPU), focusing on performance, power, and area (PPA) constraints.
Role type
Graduate Machine Learning Engineer (Hardware/Algorithm)
Builds
Optimized ML/AI workloads and internal exploration tooling for Arm IP development
Domain
Semiconductor / AI Computing / Arm Architecture
Deliverable
production ML models
Required skills
Python, Deep Learning (PyTorch/TensorFlow), Hardware-aware optimization, Compute kernel optimization (CUDA/Triton), ML/AI model training and deployment, Software systems design
Preferred skills
Industry or academic ML/AI project experience, Hardware-aware performance optimization, Knowledge of the full ML stack from model to hardware
Technologies
Arm, Python, PyTorch, Tensorflow, CUDA, Triton, CPU, GPU, NPU
Responsibilities
Analyze complex AI models on Arm computing platforms, Develop hardware-aware optimization algorithms, Collect and analyze ML/AI use cases to drive workload support direction, Develop internal tooling for algorithmic and architecture exploration
Seniority
Graduate (Entry-level)
