Framework Engineer - PyTorch

AI compute hardware and software company developing next-generation artificial intelligence compute systems, backed by SoftBank Group.
Bristol, UK
Machine Learning
Mid-Level Software Engineer
In-Person
AI

Description For Framework Engineer - PyTorch

Graphcore, a pioneering company in AI compute hardware and software backed by SoftBank Group, is seeking a Framework Engineer to join their PyTorch team. This role offers an exciting opportunity to shape the future of AI technology by working on cutting-edge machine learning frameworks.

As a Framework Engineer, you'll be integral to ensuring Graphcore's hardware maintains first-class support in popular ML frameworks, particularly PyTorch. Working within a SCRUM team, you'll tackle challenges in performance optimization, feature development, and code maintenance while collaborating closely with various teams across the organization.

The position combines technical expertise in C++ and Python with the opportunity to work on high-impact projects in the machine learning industry. You'll be part of a team that values continuous improvement and personal development, following agile working practices.

The ideal candidate brings commercial experience in C++ application development and Python programming, with preferred experience in ML frameworks and AI/ML knowledge. This role offers growth opportunities through mentoring junior team members and contributing to open-source development.

Graphcore provides an inclusive work environment with comprehensive benefits, including flexible working arrangements, competitive healthcare coverage, and various office perks. Located in Bristol, the company offers a collaborative atmosphere where innovation and personal growth are encouraged.

Last updated 9 days ago

Responsibilities For Framework Engineer - PyTorch

  • Design and implement new features & maintain and support existing features
  • Develop and maintain unit tests, component tests and integration tests
  • Optimize the software stack for better performance
  • Manage code quality, complexity and technical debt
  • Contribute to documentation, including user manuals and tutorials
  • Conduct code reviews
  • Resolve regressions, performance issues and software defects
  • Coach and mentor team members
  • Contribute to collaborative team culture
  • Work 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 (preferred)
  • Knowledge of AI/ML (preferred)
  • Experience in computationally intensive engineering (preferred)
  • Understanding of computing architectures (preferred)
  • Experience in profiling / optimizing high performance code (preferred)

Benefits For Framework Engineer - PyTorch

Medical Insurance
Dental Insurance
Vision Insurance
Parental Leave
Mental Health Assistance
  • Flexible working
  • Generous annual leave policy
  • Private medical insurance and health cash plan
  • Dental plan
  • Pension matched up to 5%
  • Life assurance
  • Income protection
  • Parental leave policy
  • Employee assistance programme
  • Office perks including healthy food and snacks
  • 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 optimization and hardware development

Machine Learning Engineer

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

Crypto Data Scientist / Machine Learning Engineer

Remote Machine Learning Engineer position at Token Metrics, developing AI-driven crypto investment solutions with Python and Java.

Machine Learning Engineer - LLM

Machine Learning Engineer position at CloudWalk, focusing on LLM applications in fintech, developing conversational AI solutions for financial management.

Machine Learning Engineer

Machine Learning Engineer position at Viggle AI, focusing on developing and optimizing AI models for video generation platform, requiring 3+ years of ML experience.