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

Red Hat · 8 hours ago

Principal Machine Learning Engineer, AI Inference

Red Hat is the world’s leading provider of enterprise open source software solutions, and they are seeking a Principal Machine Learning Engineer focused on vLLM. In this role, you will collaborate with the team to tackle challenges in model performance and efficiency, contributing to the design and development of inference optimization algorithms.

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

Responsibilities

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
Mentor and guide other engineers and foster a culture of continuous learning and innovation

Qualification

PythonC++GPU performance optimizationTensor math librariesParallel processingDistributed computingHigh-performance networkingTechnical design discussionsCommunication skillsMentoring

Required

Extensive experience in writing high performance code for GPUs and deep knowledge of GPU hardware
Strong understanding of computer architecture, parallel processing, and distributed computing concepts
Experience with tensor math libraries such as PyTorch
Deep understanding and experience in GPU performance optimizations
Experience optimizing kernels for deep neural networks
Experience with high-performance networking protocols and technologies including UCX, RoCE, InfiniBand, and RDM
Strong communications skills with both technical and non-technical team members
BS, or MS in computer science or computer engineering or a related field. A PhD in a ML related domain is considered 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
Mark Little
CTO JBoss
linkedin
Company data provided by crunchbase