Senior Machine Learning Engineer

About the role

As a Senior Machine Learning Engineer in the Applied AI team, you will contribute to advancing AI technology by developing and optimising new and existing AI models for specialised hardware. You will work on large scale systems where performance is critical, gaining visibility of the entire pipeline from novel accelerator hardware to state-of-the-art AI applications.

Responsibilities

• Implement and train state-of-the-art machine learning models and optimise for performance, accuracy and scalability across systems comprising 1000s of accelerators.
• Benchmark and profile ML models to identify performance bottlenecks.
• Develop deep understanding across the software stack in order to optimise kernel implementations.
• Test and evaluate new internal software releases, provide feedback to software engineering teams, make necessary code fixes, and conduct code reviews.
• Design and conduct experiments on novel AI methods and evaluate results.
• Collaborate with Research, Software, and Product teams to define, build, and test Graphcore’s next generation of AI hardware.
• Engage with AI community and keep in touch with the latest developments in AI.

Required skills

• Machine Learning
• PyTorch
• JAX
• Python
• C++
• Deep learning
• Hardware-accelerated deep learning
• Performance optimization

Nice to have

• MLOps
• Kubernetes
• Large language models
• Low-precision arithmetic

Qualifications

• Bachelor's degree in Machine Learning, Computer Science, Maths, Data Science, or related field
• Master's degree in Machine Learning, Computer Science, Maths, Data Science, or related field
• PhD in Machine Learning, Computer Science, Maths, Data Science, or related field
• Equivalent practical experience