Framework Engineer - PyTorch

Graphcore develops hardware for machine learning frameworks and AI applications.
Bristol, UK
$60,000 - $90,000
Machine Learning
Mid-Level Software Engineer
In-Person
501 - 1,000 Employees
3+ years of experience
AI

Description For Framework Engineer - PyTorch

Graphcore is seeking a Framework Engineer to join their PyTorch team, focusing on machine learning framework development. The role involves working with cutting-edge ML hardware integration, ensuring Graphcore's hardware has optimal support in frameworks like PyTorch. You'll be part of a SCRUM team, collaborating with various engineering teams to deliver high-performance ML solutions. The position offers an opportunity to work on performance optimizations, code maintenance, and framework development while mentoring junior team members. The company emphasizes continuous improvement and personal development, following agile practices. This role is perfect for engineers passionate about ML frameworks, performance optimization, and working with open-source communities. The position offers comprehensive benefits and a collaborative work environment in Bristol.

Last updated a day ago

Responsibilities For Framework Engineer - PyTorch

  • Designing and implementing new features & maintaining and supporting existing features
  • Developing and maintaining unit tests, component tests and integration tests
  • Optimising the software stack for performance
  • Managing code quality, code complexity and technical debt
  • Contributing to documentation, including user manuals and tutorials
  • Conducting code reviews
  • Resolving regressions, performance issues and software defects
  • Coaching and mentoring other team members
  • Working with upstream open source development teams

Requirements For Framework Engineer - PyTorch

Python
  • Experience of developing performant C++ applications in a commercial setting
  • Experience of Python development
  • Experience using and/or development of ML Frameworks such as PyTorch (desired)
  • Demonstrable knowledge of AI/ML (desired)
  • Experience in a computationally intensive engineering field (desired)
  • Understanding of computing architectures (desired)
  • Experience of maintaining and supporting a complex code base (desired)
  • Experience in profiling / optimising high performance code (desired)

Benefits For Framework Engineer - PyTorch

Medical Insurance
Dental Insurance
Parental Leave
Mental Health Assistance
  • Flexible working
  • Generous annual leave policy
  • Private medical insurance
  • Health cash plan
  • Dental plan
  • Pension (matched up to 5%)
  • Life assurance
  • Income protection
  • Generous parental leave policy
  • Employee assistance programme
  • Office snacks and barista bar

Interested in this job?

Jobs Related To Graphcore Framework Engineer - PyTorch

Machine Learning Engineer

Machine Learning Engineer position at Graphcore, focusing on AI model implementation and optimization for specialized hardware.

Machine Learning Engineer

Machine Learning Engineer position at Graphcore, focusing on AI model implementation and optimization for specialized hardware.

Framework Engineer - PyTorch

Framework Engineer position at Graphcore, focusing on PyTorch development and optimization for AI compute systems.

Machine Learning Engineer

Machine Learning Engineer position at Graphcore, focusing on AI model optimization and hardware development

Software Development Engineer, Frontier AI & Robotics

Software Development Engineer role at Amazon's Frontier AI & Robotics team, focusing on optimizing large-scale transformer models for robotics applications.