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 22 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

Software Development Engineer

Software Development Engineer role at Amazon's FinTech team, building AI/ML applications for financial risk management and treasury operations.

SDE II - Machine Learning, Amazon

Machine Learning Software Development Engineer role at Amazon's advertising division, building scalable ML systems and pipelines.

Software Dev Engineer II, Amazon

Software Dev Engineer II position at Amazon focusing on developing LLMs for e-commerce, offering competitive compensation and opportunity to work on cutting-edge AI technology.