Distributed LLM Inference Engineer

Anyscale; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Stock Options

Healthcare plans

401k Retirement Plan

Education & Wellbeing Stipend

Paid Parental Leave

Fertility Benefits

Paid Time Off

Commute reimbursement

100% of in-office meals covered

Job summary

Anyscale is seeking a Distributed LLM Inference Engineer in San Francisco, California. This pivotal role involves pushing the boundaries of performance for ML inference at scale. You'll work closely with product teams to deliver end-to-end solutions while leveraging open-source technologies and contributing to community projects. Candidates should have a solid understanding of distributed systems and familiarity with deep learning frameworks, ideally with experience in PyTorch and Ray. Anyscale offers competitive compensation and extensive benefits, including healthcare coverage and stock options.

Qualifications

• 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 and ML inference challenges.

Responsibilities

• Iterate with product teams to ship end-to-end solutions for inference.
• Work across the stack integrating Ray Data and LLM engine for optimizations.
• Contribute improvements to open source software like vLLM.

Skills

ML inference at large scale

Deep learning frameworks

Distributed systems knowledge

Tools

PyTorch

Ray

CUDA

Job description

About Anyscale

At Anyscale, were on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. Were commercializing Ray, a popular open-source project that is creating an ecosystem of libraries for scalable machine learning. Companies like OpenAI, Uber, Spotify, Instacart, Cruise, and many more have Ray in their tech stacks to accelerate the progress of AI applications out into the real world.

With Anyscale, were building the best place to run Ray, so that any developer or data scientist can scale an ML application from their laptop to the cluster without needing to be a distributed systems expert.

Proud to be backed by Andreessen Horowitz, NEA, and Addition with $250+ million raised to date.

About the role

As a Distributed LLM Inference Engineer, you will help with 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 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

We''d love to hear from you if you have

• 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

Bonus points!

• 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

Compensation

At Anyscale, we take a market-based approach to compensation. We are data-driven, transparent, and consistent. As the market data changes over time, the target salary for this role may be adjusted.

This role is also eligible to participate in Anyscale\'s Equity and Benefits offerings, including the following:

• 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

Anyscale Inc. is an Equal Opportunity Employer. Candidates are evaluated without regard to age, race, color, religion, sex, disability, national origin, sexual orientation, veteran status, or any other characteristic protected by federal or state law.

Anyscale Inc. is an E-Verify company and you may review the Notice of E-Verify Participation and the Right to Work posters in English and Spanish.