AI/ML Engineer - Large Language Models
About Anyscale:
At Anyscale, we're on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. We’re commercializing Ray, a popular open-source project that's 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, we’re 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.
Anyscale is based in San Francisco, CA. Employees are required to come in office 3x a week.
About the role
We are seeking exceptional AI/ML Engineers to join our hybrid research and product development team. In this role, you will collaborate closely with cross-functional teams to expand the usability of open source LLMs within Anyscale’s ecosystem for enterprise applications.
You will have the opportunity to experiment directly with models, build features for fine-tuning and inference, and work hand-in-hand with customers to push the boundaries of their LLM use cases on our AI platform.
As part of this role, you will
Design, develop, and deploy cutting-edge AI/ML solutions for enterprise applications, making it easy and self-sufficient for customers to adopt AI in their applicationsContribute to the development of features and enhancements for fine-tuning and inference of LLMs, ensuring a clear value proposition, elegant API design, and intuitive user experienceCollaborate with customers to understand their needs, help them grow their use cases on our platform, and enable them to achieve their desired outcomesConduct experiments and research to push the practical boundaries of open source LLMs within Anyscale’s ecosystemContinuously improve the performance, scalability, and usability of our AI platform, adopting a full-stack approach to deliver value for LLM platform featuresCreate top-notch technical documentation, including design docs for internal discussions, engineering docs and tutorials for customers, and compelling data-driven case studies for marketing purposesDrive influence through meetings, mentorship, collaboration, and other non-coding activities Qualification
Master’s degree with at least 3+ years of experience, or Ph.D. in a relevant fieldStrong technical skills, with a keen eye for API design and a deep understanding of LLM architecture, training, and inferenceHad hands on experience on at least one of the following topics:Training: LLM Pretraining, LLM fine-tuning, RLHF, distillation, parameter-efficient methods like LoRA, quantization, etc.Inference: Building inference engine / server, Writing optimized kernels, inference + training co-optimizations (speculative decoding), etc.Have experience with observability, monitoring, debugging, and measuring success for LLM training / inference systemsExcellent technical writing skills, with the ability to create clear, concise, and engaging documentation for various audiencesProactive and self-driven, with a strong sense of ownership over the systems you work onAdaptable to the ever-changing startup environment, embracing pivots and new challengesStrong problem-solving skills and the ability to thrive in a collaborative environment Compensation
At Anyscale, we take a market-based approach to compensation. We are data-driven, transparent, and consistent. The target salary for this role is $170,112 ~ $237,000. 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, Stock OptionsHealthcare plans, with premiums covered by Anyscale at 99%401k Retirement PlanWellness stipendEducation stipendPaid Parental LeaveFlexible Time OffCommute Reimbursement100% 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.
Apply for this job →