Red Hat

Principal Machine Learning Engineer - PyTorch

Red Hat Bengaluru, India
Full-time Posted 8 months ago

. What you should bring . 26 years of experience in ML systems

Hands-on experience with PyTorch (internals,custom ops,or advanced usage)

Experience with CUDA,ROCm / AMD GPUs,or Intel GPU / oneAPI

Tags & Focus Areas

Machine Learning Kubernetes Algorithms Python Medical C Performance Profiling Ai Pytorch

About Red Hat

Write robust Python and C++, working on vLLM systems, high performance machine learning primitives, performance analysis and modeling, and numerical methods Contribute to the design, development, and testing of various inference optimization algorithms Participate in technical design discussions and provide innovative solutions to complex problems Give thoughtful and prompt code reviews. Proactively utilizing AI-assisted development tools for code generation, auto-completion, and intellige...

Industry Machine Learning
HQ Bengaluru, India