5 days ago
Responsibilities
- Implement latest machine learning models and optimise them for performance and accuracy, scaling to 1000s of accelerators.
- Test and evaluate new internal software releases, provide feedback to software engineering teams, make necessary code fixes, and conduct code reviews.
- Benchmark models and key ML techniques to identify performance bottlenecks and improve model efficiency.
- Design and conduct experiments on novel AI methods, implement them 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 deep learning from model training to optimisation and evaluation.
- Experience in distributed training or inference of ML models across 64+ accelerators.
- Capable of designing, executing and reporting from ML experiments.
- Developed deep understanding of performance bottlenecks and how to overcome them.
- Ability to move quickly in a dynamic environment.
- Enjoy cross-functional work collaborating with other teams.
- Strong communicator - able to explain complex technical concepts to different audiences.
Benefits
- Flexible working arrangements.
- Generous annual leave policy.
- Private medical insurance and health cash plan.
- Dental plan and pension matched up to 5%.
- Life assurance and income protection.
- Generous parental leave policy and employee assistance programme.
- Healthy food and snacks at the office and a barista bar.