Description
As a Senior Machine Learning Engineer in the Applied AI team at Graphcore, 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 to the success of projects, and play a critical role in identifying opportunities to innovate and differentiate Graphcore's technology.
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 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.
Requirements:
- Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field.
- Proficiency in deep learning frameworks like PyTorch/JAX.
- Strong Python or C++ software development skills.
- Expertise in hardware-accelerated deep learning from model training to optimisation and evaluation.
- Capable of designing, executing and reporting from ML experiments.
- Well-developed understanding of performance bottlenecks and how to overcome them.
- Ability to move quickly in a fast-moving field.
- Enjoy cross-functional work collaborating with other teams.
- Strong communicator - able to explain complex technical concepts to different audiences.
Benefits:
- Flexible working
- Generous leave
- Retirement planning support: up to 5% matched pension
- Phantom equity
- Workplace experience: thoughtfully designed office spaces with free food and drinks
- Peace of mind protection: income protection and life assurance
- Electric Vehicle Scheme
- Flexible benefits: tailor your benefits package with a choice of additional options