Jobs

Interviews
Job Alerts
Tools

Upskill and Grow with AI

Mock Interview Practice interviews in realistic simulations

Coding Practice Improve your coding skills with challenges

Certification Earn certifications to validate your skills

AI Learning Get trained with AI expert sessions

Career Path AI insights for smarter career decisions

AI Job Match Score AI-Powered Job Match Against Your Resume and Optimize Your Resume

Career Tools and Resources

Resume Builder Build Professional Resume with Ease

ATS Friendliness Check Check Resume Friendliness for Applicant Tracking Systems

Auto Apply Apply to hundreds of jobs on any platform effortlessly

Co-Pilot (Chrome Extension) Your AI Assistant for Seamless Browsing Efficiency

Interview Questions Streamline interviews with ready-to-use questions

Salaries Discover market-driven salary insights across skillsets and geographies

Companies Explore leading companies actively hiring talent
For Employers

Home
>
Jobs in bengaluru
>
Red Hat
>
Principal Machine Learning Engineer - PyTorch

Principal Machine Learning Engineer - PyTorch

Red Hat

5 - 9 years

15 - 19 Lacs

bengaluru

Posted:3 days ago| Platform:

Apply

Skills Required

machine learning kubernetes algorithms python medical c performance profiling ai cloud open source git recruitment linux design pytorch software solutions ai platform data structures red hat nvidia ml

Work Mode

Work from Office

Job Type

Full Time

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.

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.

More Jobs at Red Hat

Senior Consultant

Mumbai

3 - 7 yrs

INR 11 - 16 Lacs

Senior Software Engineer

Bengaluru

3 - 5 yrs

INR 6 - 8 Lacs

Software Engineer

Pune

2 - 5 yrs

INR 8 - 12 Lacs

Account Solution Architect

Mumbai

6 - 10 yrs

INR 8 - 12 Lacs

Corporate Account Executive

Mumbai, Pune, Bengaluru

1 - 4 yrs

INR 3 - 6 Lacs

Mock Interview

Practice Video Interview with JobPe AI

Start Machine Learning Interview

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

Enhance Your Python Skills

Practice Python coding challenges to boost your skills

Start Practicing Python Now

Red Hat

Software Development

Raleigh NC

Login to

Please Verify Your Phone or Email

Confirm Action

Principal Machine Learning Engineer - PyTorch