Distributed LLM Inference Engineer

Anyscale • Palo Alto, CA

4d • Hybrid

About The Position

As a Distributed LLM Inference Engineer, you will help systems and optimizations that push the boundaries of performance for inference at large scale. This is an incredibly critical role to Anyscale as it allows us to achieve a market leading position for AI infrastructure.
As part of this role, you will
Iterate very quickly with product teams to ship the end to end solutions for Batch and Online inference at high scale which will be used by open-source Ray users and customers of Anyscale
Work across the stack integrating Ray Data and LLM engine providing optimizations achieving low cost solutions for large scale ML inference
Integrate with Open source software like vLLM, work closely with the community to adopt these techniques in Anyscale solutions, and also contribute improvements to open source
Follow the latest state-of-the-art in the open source and the research community, implementing and extending best practices

Requirements

• Familiarity with running ML inference at large scale with high throughput and low latency
• Familiarity with deep learning and deep learning frameworks (e.g. PyTorch)
• Solid understanding of distributed systems, ML inference challenges

Nice To Haves

• ML Systems knowledge
• Experience using Ray
• Work closely with community on LLM engines like vLLM, TensorRT-LLM
• Contributions to deep learning frameworks (PyTorch, TensorFlow)
• Contributions to deep learning compilers (Triton, TVM, MLIR)
• Prior experience working on GPUs / CUDA

Responsibilities

• Iterate very quickly with product teams to ship the end to end solutions for Batch and Online inference at high scale which will be used by open-source Ray users and customers of Anyscale
• Work across the stack integrating Ray Data and LLM engine providing optimizations achieving low cost solutions for large scale ML inference
• Integrate with Open source software like vLLM, work closely with the community to adopt these techniques in Anyscale solutions, and also contribute improvements to open source
• Follow the latest state-of-the-art in the open source and the research community, implementing and extending best practices

Benefits

• Stock Options
• Healthcare plans, with premiums covered by Anyscale at 99% for both employees and dependents
• 401k Retirement Plan
• Education & Wellbeing Stipend
• Paid Parental Leave
• Fertility Benefits
• Paid Time Off
• Commute reimbursement
• 100% of in-office meals covered

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed