Graduate Machine Learning Engineer - Central Technology
Core
Develop hardware-aware optimization algorithms and internal tooling to improve performance, power, and area (PPA) of AI models on Arm CPU/GPU/NPU platforms.
Role type
Graduate Machine Learning Engineer (Hardware/Algorithm)
Builds
Hardware-aware optimization algorithms and internal 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 workload analysis
Preferred skills
Experience optimizing compute kernels, Hardware-aware performance optimization, Model-to-hardware implementation knowledge
Technologies
Arm, CPU, GPU, NPU, CUDA, Triton, PyTorch, TensorFlow, Python
Responsibilities
Analyze ML workloads on Arm compute platforms, Develop hardware-aware optimization algorithms, Build internal tooling for algorithmic and architecture exploration
Seniority
Graduate, Early Career
