Principal Machine Learning Engineer
Posted by Red Hat India Private Limited • Bengaluru, Karnataka, India
About the Role
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...
Ready to Apply?
Submit your application today and take the next step in your career journey with Red Hat India Private Limited.
Apply Now