Senior Principal Machine Learning Engineer, Distributed vLLM Inference with Kubernetes jobs in United States
cer-icon
Apply on Employer Site
company-logo

Red Hat · 2 days ago

Senior Principal Machine Learning Engineer, Distributed vLLM Inference with Kubernetes

Red Hat is the world’s leading provider of enterprise open source software solutions, and they are seeking a Senior Principal Machine Learning Engineer to join their Inference team. The role focuses on developing distributed inference infrastructure and optimizing AI deployments using Kubernetes.

Enterprise SoftwareInsurTechLinuxOpen SourceOperating SystemsSoftware
check
Culture & Values
check
H1B Sponsor Likelynote

Responsibilities

Develop and maintain distributed inference infrastructure leveraging Kubernetes APIs, operators, and the Gateway Inference Extension API for scalable LLM deployments
Create system components in Go and/or Rust to integrate with the vLLM project and manage distributed inference workloads
Design and implement KV cache-aware routing and scoring algorithms to optimize memory utilization and request distribution in large-scale inference deployments
Enhance the resource utilization, fault tolerance, and stability of the inference stack
Contribute to the design, development, and testing of various inference optimization algorithms
Actively participate in technical design discussions and propose innovative solutions to complex challenges
Provide timely and constructive code reviews
Mentor and guide fellow engineers, fostering a culture of continuous learning and innovation

Qualification

PythonGoLangKubernetesCloud-native technologiesLayer 7 networkingGPU performance benchmarkingBachelor'sMaster's degreeCommunication

Required

Strong proficiency in Python and at least one systems programming language (GoLang, Rust, or C++), with GoLang being highly preferred
A solid understanding of Layer 7 networking, HTTP/2, gRPC, and the fundamentals of API gateways and reverse proxies
Excellent communication skills, capable of interacting effectively with both technical and non-technical team members
A Bachelor's or Master's degree in computer science, computer engineering, or a related field. A Ph.D. in an ML-related domain is a significant advantage

Preferred

Experience with the Kubernetes ecosystem, including core concepts, custom APIs, operators, and the Gateway API inference extension for GenAI workloads (a plus)
Working knowledge of high-performance networking protocols and technologies including UCX, RoCE, InfiniBand, and RDMA is a plus
Experience with GPU performance benchmarking and profiling tools like NVIDIA Nsight or distributed tracing libraries/techniques like OpenTelemetry is a plus

Benefits

Comprehensive medical, dental, and vision coverage
Flexible Spending Account - healthcare and dependent care
Health Savings Account - high deductible medical plan
Retirement 401(k) with employer match
Paid time off and holidays
Paid parental leave plans for all new parents
Leave benefits including disability, paid family medical leave, and paid military leave
Additional benefits including employee stock purchase plan, family planning reimbursement, tuition reimbursement, transportation expense account, employee assistance program, and more!

Company

Red Hat is a software company that offers enterprise open-source software solutions. It is a sub-organization of IBM.

H1B Sponsorship

Red Hat has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (159)
2024 (148)
2023 (156)
2022 (181)
2021 (154)
2020 (106)

Funding

Current Stage
Public Company
Total Funding
unknown
2018-10-28Acquired
1999-08-20IPO
1999-03-09Corporate Round

Leadership Team

leader-logo
Chris Wright
Chief Technology Officer and Senior Vice President Global Engineering
linkedin
leader-logo
Andrew Brown
Senior Vice President and Chief Revenue Officer
linkedin
Company data provided by crunchbase