PyTorch Development Lead

Posting Location : Location US-Remote
Posting date 5 days ago(4/22/2024 1:18 PM)
Job ID
101304
Workday ID
R-037016
Category
Software Engineering

About the job

The Red Hat Engineering team 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 for Red Hat Enterprise Linux (RHEL).  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 (e.g., Red Hat OpenShift AI) 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 within Red Hat 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 Red Hat’s 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

 

 

The salary range for this position is $163,240- $281.370. Actual offer will be based on your qualifications.

 

 

#LI-LS2

#Li-REMOTE

Interested in this job?

Sorry the Share function is not working properly at this moment. Please refresh the page and try again later.
Share on your newsfeed