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

Red Hat · 4 hours ago

Principal Machine Learning Engineer, Distributed vLLM Inference

Red Hat is the world’s leading provider of enterprise open source software solutions, using a community-powered approach to deliver high-performing Linux, cloud, container, and Kubernetes technologies. As a Principal Machine Learning Engineer, you will develop and maintain distributed inference infrastructure and collaborate with the team to tackle challenges in scalable inference systems and cloud-native deployments.

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

PythonGoLangKubernetesRustC++Layer 7 networkingAPI gatewaysCommunicationMentoringContinuous learning

Required

Strong proficiency in Python, GoLang and at least one of the following: Rust, or C++
Experience with cloud-native Kubernetes service mesh technologies/stacks such as Istio, Cilium, Envoy (WASM filters), and CNI
A solid understanding of Layer 7 networking, HTTP/2, gRPC, and the fundamentals of API gateways and reverse proxies
Working knowledge of high-performance networking protocols and technologies including UCX, RoCE, InfiniBand, and RDMA is a plus
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

Preferred

Experience with the Kubernetes ecosystem, including core concepts, custom APIs, operators, and the Gateway API inference extension for GenAI workloads
Experience with GPU performance benchmarking and profiling tools like NVIDIA Nsight or distributed tracing libraries/techniques like OpenTelemetry
Ph.D. in an ML-related domain is a significant advantage

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
Mark Little
CTO JBoss
linkedin
Company data provided by crunchbase