Graphcore is seeking a Software Engineer to join the PyTorch team. You will help build software that connects Graphcore accelerators with state-of-the-art ML frameworks, designing, implementing, and optimizing features across PyTorch, Triton and related technologies.
You will contribute to upstream PyTorch, support hybrid Python and C++ extensions, and help integrate compiler backend capabilities with torch.compile.
#J-18808-Ljbffr
PyTorch & C++ Integration Engineer for AI Accelerators 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.