AI Research Engineer
Core
Designing and implementing AI/ML experiments to advance hardware-aware algorithms and efficient training/inference for next-gen AI hardware.
Role type
AI Research Engineer (Hardware-Aware ML)
Builds
Scalable AI research implementations and publications for Graphcore's AI compute stack.
Domain
AI/ML, Semiconductor Hardware, Deep Learning
Deliverable
production ML models
Required skills
Python (PyTorch/JAX), Deep Learning fundamentals, C++/CUDA/Triton, Hardware-aware optimization, Distributed training/inference, Experiment design & reporting
Preferred skills
Mathematical modeling (calculus/probability/linear algebra), Conference publication experience
Technologies
PyTorch, JAX, C++, CUDA, Triton, XLA, ATen
Responsibilities
Generate AI/ML ideas, design experiments, implement them & evaluate results; Prepare, submit & present work to AI conferences; Provide technical insight via actionable reports; Collaborate with silicon/software engineers to define and test next-gen AI hardware.
Seniority
Mid-Senior, hands-on IC