Principal Machine Learning Engineer

Red Hat

Bangalore, India
Fully remote
Experience with pytorch
Strong skills in c++ and python
2-6 years of experience in ml systems
The AI Platform Core Components organization is looking for individuals passionate about Open Source and Machine Learning

Job Summary

  • The AI Platform Core Components organization is looking for individuals passionate about Open Source and Machine Learning.
  • You will design, implement, and maintain features in PyTorch core, contributing to its performance on modern hardware.
  • Red Hat fosters an open and inclusive environment where creative ideas are encouraged and celebrated.

Matching Summary

The AI Platform Core Components organization is looking for individuals passionate about Open Source and Machine Learning.

Skills & Requirements

Must-have

  • Experience with PyTorch
  • Strong skills in C++ and Python
  • 2-6 years of experience in ML systems

Nice-to-have

  • Familiarity with numerical computing
  • Prior contributions to PyTorch
  • Experience with CUDA or AMD GPUs

Key Requirements

  • Hands-on experience with PyTorch internals
  • Solid understanding of algorithms and data structures
  • Comfortable working in Linux and Git

Work Rights

Not specified

Tailored Resume

Cover Letter