| Job Title | Principal Machine Learning Engineer |
|---|---|
| Category | Engineering |
| Location | Bengaluru |
| Company | Red Hat |
| Description | 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 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. 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. |
| Salary | Not Disclosed |
| Last Date | 2026-04-18 00:00:00 |
| Apply Link | Click Here |
© EllipseJob is Proudly Owned by