Principal Machine Learning Engineer
This job is in your area. Enjoy a short commute and work close to home.
Job Description
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, li...