USD per year
Principal Machine Learning Engineer
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 Hat’s 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.
- Contribute to the PyTorch upstream community
What you will bring :
- 2-6 years of experience in ML systems.
- Experience contributing to Open Source projects
- 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
The following are considered a Plus :
- Familiarity with numerical computing, vectorization, and low-level performance profiling tools.
- Prior contributions to PyTorch or other ML/AI open-source projects.
- Experience with CUDA, ROCm/AMD GPUs, or Intel GPU/oneAPI.
Red Hat is the leading provider of enterprise open source software solutions. Enterprises around the world trust our broad portfolio of hybrid cloud infrastructure, application services, cloud-native application development, automation, and artificial intelligence solutions to deliver IT services on any infrastructure quickly and cost effectively. More than 90% of companies in the U.S. Fortune 500 continue to rely on Red Hat. Building enterprise-ready solutions with open source: We believe using an open development model helps create more secure, stable, and innovative technologies. By collaborating with open source communities, we’re developing software that pushes the boundaries of technological ability. Red Hat is an open hybrid cloud technology leader, delivering a consistent, comprehensive foundation for transformative IT and artificial intelligence (AI) applications in the enterprise. As a trusted adviser to the Fortune 500, Red Hat offers cloud, developer, Linux, automation, and application platform technologies, as well as award-winning services.
View Company Profile