Principal Machine Learning Engineer

About the position

AI Platform Core Components drive Red Hat's contribution to key open source AI communities. We work on PyTorch, vLLM, Triton and their many dependent and related projects. Our mission is to support and foster these communities, as well as productise them as part of RHEL AI and OpenShift AI. The PyTorch team is looking for a Software Engineer to contribute to PyTorch upstream and PyTorch integration into the RHEL AI and OpenShift AI products.

Responsibilities

• Contribute to the PyTorch upstream community
• Contribute to the integration of PyTorch into the RHEL AI and OpenShift AI products
• Work closely with teams within AI Engineering, where collaboration opportunities present themselves
• Adhere to architectural and design principles that help improve overall development within the group
• Learn the architectural concepts that are key to the development of our product set

Requirements

• Experience contributing to Open-Source projects
• Experience using and contributing to AI related projects
• Existing PyTorch contributions a strong plus
• Experience with test-base development and agile/scrum methodologies

Benefits

• Comprehensive medical, dental, and vision coverage
• Flexible Spending Account - healthcare and dependent care
• Health Savings Account - high deductible medical plan
• Retirement 401(k) with employer match
• Paid time off and holidays
• Paid parental leave plans for all new parents
• Leave benefits including disability, paid family medical leave, and paid military leave
• Additional benefits including employee stock purchase plan, family planning reimbursement, tuition reimbursement, transportation expense account, employee assistance program, and more!