Graphcore - Senior Machine Learning Engineer (Large Systems)

Bristol, UK - Hybrid+ Equity2w ago

In Office Senior EMEA Insurance Cloud Computing Machine Learning Engineer JAX Reporting C++Python MLOps

Requirements

• Essential: • 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. • Desirable: • Experience in one or more of: • MLOps for Kubernetes-based clusters • Building production systems with large language models • Efficient computing based on low-precision arithmetic. • Experience writing C++/Triton/CUDA kernels for performance optimisation of ML models. • Experience in distributed training or inference of ML models across 64+ accelerators. • Familiarity with HPC systems and networking including Infiniband, NVLink, RoCE technologies, and cloud computing platforms. • Have contributed to open-source projects or published research papers in relevant fields. • Keen to present, publish and deliver talks in the AI community. • Peace of mind protection: Income protection and life assurance to provide financial security for you and your loved ones • Peace of mind protection: • Electric Vehicle Scheme: Choose an electric vehicle and lease it through salary sacrifice • Electric Vehicle Scheme: • Flexible benefits: Tailor your benefits package with a choice of additional options, including private medical insurance and dental cover

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.

Benefits

• Flexible working: Balance your work and personal life with greater flexibility • Flexible working: • Generous leave: Take time to rest, recharge and enjoy life outside of work • Generous leave: • Retirement planning support: Up to 5% matched pension • Retirement planning support: • Phantom equity: Share in Graphcore’s success • Phantom equity: • Workplace experience: Enjoy thoughtfully designed office spaces for collaboration, with free food and drinks to support your day • Optional benefits: Dental cover, health cash plan, private medical insurance, cycle to work scheme, give as you earn • We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.