Machine Learning Engineer, vLLM Inference - Tool Calling and Structured Output jobs in United States
cer-icon
Apply on Employer Site
company-logo

Red Hat · 2 hours ago

Machine Learning Engineer, vLLM Inference - Tool Calling and Structured Output

Red Hat is the world’s leading provider of enterprise open source software solutions, and they are seeking a Machine Learning Engineer focused on vLLM. In this role, you will collaborate with the team to innovate and tackle challenges in model performance and efficiency while building and maintaining subsystems for vLLM.

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

Responsibilities

Write robust Python and Pydantic, working on vLLM systems, high performance machine learning primitives, performance analysis and modeling, and numerical methods
Contribute to the design, development, and testing of function calling, tool calling parser, and structured output subsystems in vLLM
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

PythonPydanticLLM Inference ConceptsOpenAI Chat Completions APITensor math librariesEfficient parsing techniquesJinja2 chat templatesBeam SearchGreedy DecodingCommunication skills

Required

Strong experience in Python and Pydantic
Strong understanding of LLM Inference Core Concepts, such as logits processing (ie. Logit Generation -> Sampling -> Decoding loop)
Deep familiarity with the OpenAI Chat Completions API specification
Deep familiarity with libraries like Outlines, XGrammar, Guidance, or Llama.cpp grammars
Strong communication skills with both technical and non-technical team members
BS, or MS in computer science or computer engineering, mathematics, or a related field

Preferred

Proficiency with efficient parsing techniques (e.g., incremental parsing) is a strong plus
Proficiency with Jinja2 chat templates
Familiarity with Beam Search and Greedy Decoding in the context of constraints
Familiarity with LLM inference metrics and tradeoffs
Experience with tensor math libraries such as PyTorch is a strong plus
PhD in an 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 (128)
2024 (149)
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