Principal Machine Learning Engineer

About The Job

Join the AI Platform Core Components organization within AI Engineering at Red Hat, a place for individuals passionate about Open Source and Machine Learning to expand the reach of Red Hat's AI offerings for customers and the wider community.

We are seeking a skilled PyTorch Machine Learning Engineer focused on enhancing, extending, and contributing upstream to PyTorch on Red Hat platforms. Your primary role will involve deep engagement with PyTorch core development, optimizing performance on contemporary hardware, and fostering collaboration within the upstream community.

What You Will Do

• Spearhead the design, implementation, and maintenance of features within PyTorch core, encompassing operations, kernels, and tooling, using Python and C++.
• Conduct thorough profiling and optimization of PyTorch execution across CPU and GPU/accelerators, including Intel, AMD, and NVIDIA CUDA.
• Develop comprehensive test suites, benchmarks, and minimal examples to rigorously validate correctness and performance.
• Diagnose and resolve issues spanning the entire stack, from PyTorch and associated libraries to hardware and drivers, contributing fixes upstream.
• Collaborate effectively with upstream PyTorch maintainers and internal Red Hat teams, authoring clear documentation and design proposals.
• Actively contribute to the PyTorch upstream community.

What You Will Bring

• 2 to 6 years of professional experience in ML systems development.
• Demonstrated experience in contributing to Open Source projects.
• Proficiency in both C++ and Python programming languages.
• Hands-on experience with PyTorch, including internal workings, custom operations, or advanced usage scenarios.
• A solid understanding of fundamental algorithms, data structures, and principles of performance-oriented coding.
• Comfort and familiarity with Linux environments, Git for version control, and modern software development workflows.

The Following Are Considered a Plus

• Familiarity with numerical computing techniques, vectorization, and low-level performance profiling tools.
• Prior contributions to PyTorch or other significant ML/AI open-source projects.
• Experience with CUDA, ROCm/AMD GPUs, or Intel GPU/oneAPI technologies.

About Red Hat

Red Hat is globally recognized as a leader in enterprise open source software solutions, leveraging a community-driven approach to deliver cutting-edge Linux, cloud, container, and Kubernetes technologies. With associates in over 40 countries, Red Hat offers flexible work arrangements, including in-office, office-flex, and fully remote options, depending on role requirements. We encourage all associates to bring forward their best ideas, fostering an open and inclusive environment that drives innovation. Our leadership in open source is a direct result of our commitment to an inclusive culture where creative, passionate individuals can solve complex problems and make a real impact.

Inclusion at Red Hat

Red Hat's culture is deeply rooted in open source principles of transparency, collaboration, and inclusion, ensuring that the most innovative ideas can emerge from any individual or team. This environment empowers people from diverse backgrounds, perspectives, and experiences to collaborate, challenge conventional thinking, and drive innovation. We are committed to fostering a culture where everyone has equal opportunity and access, and where all voices are not only heard but also celebrated. We invite you to join our celebration, and we enthusiastically welcome and encourage applicants from all the diverse backgrounds that enrich our global community.

Equal Opportunity Policy (EEO)

Red Hat is a proud equal opportunity workplace and affirmative action employer. We evaluate applications for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, citizenship, age, veteran status, genetic information, physical or mental disability, medical condition, marital status, or any other characteristic protected by law. Red Hat does not accept unsolicited resumes or CVs from recruitment agencies and is not liable for fees related to them unless a written contract is in place. Red Hat supports individuals with disabilities and provides reasonable accommodations for job applicants. For assistance with the online application, please email [HIDDEN TEXT]. General inquiries will not receive a response.