PyTorch Integration Engineer for Accelerated ML

PyTorch Integration Engineer for Accelerated ML

Full-Time No working from home possible
G

Graphcore is seeking a Software Engineer to help make Graphcore hardware feel native inside ML frameworks. You will design, implement, and optimize features that connect Graphcore accelerators with PyTorch, Triton and related technologies.

Join a SCRUM team in Frameworks, contribute to upstream PyTorch, and support a hybrid Python/C++ extension. You will solve real engineering problems, ship tooling, and improve performance and developer experience for ML researchers.

#J-18808-Ljbffr

PyTorch Integration Engineer for Accelerated ML employer: graphcore

Graphcore is an exceptional employer, offering a vibrant work culture that fosters continuous learning and innovation in the heart of London. With a commitment to inclusivity and employee well-being, we provide flexible working arrangements, comprehensive benefits including private medical insurance and generous parental leave, and opportunities for professional growth within a cutting-edge AI environment. Join us to be part of a transformative journey in AI technology, where your contributions will directly impact the future of intelligent computing.

G

Contact Details:

graphcore Recruitment Team