Principal Machine Learning Engineer PyTorch

About the Job :

The AI Platform Core Components organization part of AI Engineering is looking for individuals with a passion for Open Source and Machine Learning enthusiasts helping grow the impact of Red Hats AI offerings for our customers and the community.

We are looking for a PyTorch Machine Learning Engineer to help improve extend and upstream PyTorch on Red Hat platforms. You will work primarily on PyTorch contributing to PyTorch core improving performance on modern hardware and collaborating with the upstream community.

What you will do

• Design implement and maintain features in PyTorch core (Python and C) including ops kernels and tooling.

• Profile and optimize PyTorch execution on CPU and GPU/accelerators (Intel AMD NVIDIA CUDA).

• Build tests benchmarks and minimal examples to validate correctness and performance.

• Debug issues across the stack (PyTorch libraries hardware drivers) and contribute fixes upstream.

• Collaborate with upstream PyTorch maintainers and internal teams; write clear docs and design notes.

What you should bring

• 26 years of experience in ML systems.

• Strong skills in C and Python.

• Hands-on experience with PyTorch (internals custom ops or advanced usage).

• Solid understanding of algorithms data structures and performance-oriented coding.

• Comfortable working in Linux Git and modern development workflows.

Nice to have

• Experience with CUDA ROCm/AMD GPUs or Intel GPU/oneAPI.

• Prior contributions to PyTorch or other ML/AI open-source projects.

• Familiarity with numerical computing vectorization and low-level performance profiling tools.

About Red Hat

Red Hat is the worlds leading provider of enterprise open source software solutions using a community-powered approach to deliver high-performing Linux cloud container and Kubernetes technologies. Spread across 40 countries our associates work flexibly across work environments from in-office to office-flex to fully remote depending on the requirements of their role. Red Hatters are encouraged to bring their best ideas no matter their title or tenure. Were a leader in open source because of our open and inclusive environment. We hire creative passionate people ready to contribute their ideas help solve complex problems and make an impact.

Inclusion at Red Hat
Red Hats culture is built on the open source principles of transparency collaboration and inclusion where the best ideas can come from anywhere and anyone. When this is realized it empowers people from different backgrounds perspectives and experiences to come together to share ideas challenge the status quo and drive innovation. Our aspiration is that everyone experiences this culture with equal opportunity and access and that all voices are not only heard but also celebrated. We hope you will join our celebration and we welcome and encourage applicants from all the beautiful dimensions that compose our global village.

Equal Opportunity Policy (EEO)
Red Hat is proud to be an equal opportunity workplace and an affirmative action employer. We review applications for employment without regard to their 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 basis prohibited by law.

Red Hat does not seek or accept unsolicited resumes or CVs from recruitment agencies. We are not responsible for and will not pay any fees commissions or any other payment related to unsolicited resumes or CVs except as required in a written contract between Red Hat and the recruitment agency or party requesting payment of a fee.

Red Hat supports individuals with disabilities and provides reasonable accommodations to job applicants. If you need assistance completing our online job application email . General inquiries such as those regarding the status of a job application will not receive a reply.