PyTorch Development Lead

Location: Raleigh, NC  [Remote]
Category: Technology
Employment Type: Direct Hire
Job ID: 12541
Date Added: 04/10/2024

Apply Now

Fill out the form below to submit your information for this opportunity. Please upload your resume as a doc, pdf, rtf or txt file. Your information will be processed as soon as possible.

US Applications only
* Required field.

About the job
Our client is looking for a lead developer in the PyTorch community to join us as a Principal Software Engineer in our growing AI engineering team.  In this role, you will lead the design and development of the PyTorch framework and the AI/ML components.  This AI/ML engineering effort is focused on developing, improving and integrating the PyTorch framework, as well as the related support libraries and tools necessary to give developers, researchers, data scientists, and our internal teams  the environment required to develop AI-based models, applications, tools, etc.  You will work alongside our GPU and Hardware Accelerator teams, who will be integrating hardware support for the AI components that this AI/ML team is building.  We’re open to candidates in the US and other countries.
What you will do

  • Design and build the RHEL AI/ML infrastructure
  • Actively develop the PyTorch framework in collaboration with the upstream community
  • Work closely with key stakeholders who are using the frameworks, libraries and tools that you develop
  • Optimize the AI infrastructure to better perform and integrate with our layered products
  • Be a leader in the team defining AI platforms that will be used by the AI/ML industry
What you will bring
  • Deep internal development experience in the PyTorch framework
  • Must have contributed to the Pytorch community
  • Open Source Development experience with the INTERNALS of Pytorch and related tools, utilitites and libraries
  • Fluent in Python, C/C++, and Linux scripting tools
  • Passion for growing the AI/ML infrastructure
The following are considered a plus
  • Master’s degree or PhD in computer science or related field is a big plus
  • Experience with AI/ML model development using Pytorch or other AI/ML platforms like Tensorflow
  • Experience with LLVM or MLIR compiler infrastructure project
  • Experience in operating system integration and tooling; RHEL experience is a big plus
  • Experience with container-related technologies like Podman, OpenShift or Kubernetes
  • Knowledge of Git version control system
  • Passion for developing open source solutions