Deep Learning Performance Engineer

Anyscale commercializes Ray, a popular open-source project for scalable machine learning, making distributed computing accessible to developers.
$170,112 - $237,000
Machine Learning
Senior Software Engineer
In-Person
101 - 500 Employees
5+ years of experience
AI · Enterprise SaaS

Description For Deep Learning Performance Engineer

Anyscale, backed by Andreessen Horowitz, NEA, and Addition with $250+ million in funding, is revolutionizing distributed computing through Ray, their open-source project. As a Deep Learning Performance Engineer, you'll be at the forefront of optimizing cutting-edge ML models, working with companies like OpenAI, Uber, and Spotify. This role is crucial for maintaining Anyscale's market-leading performance in AI infrastructure.

The position demands expertise in GPU/CUDA programming, deep learning frameworks, and system optimizations. You'll collaborate with product and research teams, implementing state-of-the-art practices in LLM engines. The role offers exciting opportunities to work with advanced technologies like vLLM and TensorRT-LLM.

The company provides comprehensive benefits including competitive equity, healthcare coverage, and various stipends for wellness and education. Located in San Francisco, you'll join a team dedicated to democratizing distributed computing and making it accessible to developers of all skill levels. This is an excellent opportunity for those passionate about performance engineering in AI and distributed systems.

Additional valued skills include ML Systems knowledge, experience with deep learning model training, and contributions to frameworks like PyTorch or TensorFlow. Experience with Ray or deep learning compilers would be advantageous. The role offers competitive compensation ranging from $170,112 to $237,000, reflecting Anyscale's data-driven and transparent approach to compensation.

Last updated 3 months ago

Responsibilities For Deep Learning Performance Engineer

  • Iterate quickly with product teams to ship optimizations to Anyscale platform, Anyscale Endpoints, and open source offerings
  • Work closely with research teams on LLM engines like vLLM, TensorRT-LLM
  • Follow and implement state-of-the-art practices from open source and research community

Requirements For Deep Learning Performance Engineer

Python
  • Prior experience working on GPUs / CUDA
  • Solid understanding of operating systems and/or networking fundamentals
  • Familiarity with deep learning and deep learning frameworks (e.g. PyTorch)

Benefits For Deep Learning Performance Engineer

Equity
Medical Insurance
401k
Education Budget
Parental Leave
Commuter Benefits
  • Stock Options
  • Healthcare plans covered 99% by Anyscale
  • 401k Retirement Plan
  • Wellness stipend
  • Education stipend
  • Paid Parental Leave
  • Flexible Time Off
  • Commute reimbursement
  • 100% of in-office meals covered

Interested in this job?

Jobs Related To Anyscale Deep Learning Performance Engineer

Technical Curriculum Developer

Technical Curriculum Developer position at Anyscale, focusing on creating educational content for Ray and AI platform users.

AI/ML Engineer - Large Language Models

AI/ML Engineer role at Anyscale, focusing on expanding open source LLM usability for enterprise applications.

Senior Software Developer, AI/Machine Learning, Applied AI

Senior Software Developer role at Google focusing on AI/ML development, requiring 5 years of experience in software development and ML expertise.

Lead Senior Engineering Analyst, AI Safety

Lead Senior Engineering Analyst position at Google focusing on AI safety and trust, combining data analysis, machine learning, and security expertise.

Senior Software Engineer, Machine Learning, Core

Senior Software Engineer position at Google's Core team, focusing on Machine Learning and AI development for flagship products, offering competitive compensation and opportunities for technical leadership.