Avatar for Scale AI
Scale AI
Actively Hiring
Accelerate the development of AI applications
  • B2B
  • Scale Stage
    Rapidly increasing operations
  • Top Investors
    This company has received a significant amount of investment from top investors
  • +2

AI Infrastructure Engineer, Model Serving Platform

Posted: 1 year ago
Job Location
Visa Sponsorship

Not Available

RelocationAllowed

About the job

As a Software Engineer on the ML Infrastructure team, you will design and build platforms for scalable, reliable, and efficient serving of LLMs. Our platform powers cutting-edge research and production systems, supporting both internal and external use cases across various environments.

The ideal candidate combines strong ML fundamentals with deep expertise in backend system design. You’ll work in a highly collaborative environment, bridging research and engineering to deliver seamless experiences to our customers and accelerate innovation across the company.

You will:

  • Build and maintain fault-tolerant, high-performance systems for serving LLMs workloads at scale.
  • Build an internal platform to empower LLM capability discovery.
  • Collaborate with researchers and engineers to integrate and optimize models for production and research use cases.
  • Conduct architecture and design reviews to uphold best practices in system design and scalability.
  • Develop monitoring and observability solutions to ensure system health and performance.
  • Lead projects end-to-end, from requirements gathering to implementation, in a cross-functional environment.

Ideally you'd have:

  • 4+ years of experience building large-scale, high-performance backend systems.
  • Strong programming skills in one or more languages (e.g., Python, Go, Rust, C++).
  • Experience with LLM serving and routing fundamentals (e.g. rate limiting, token streaming, load balancing, budgets, etc.)
  • Experience with LLM capabilities and concepts such as reasoning, tool calling, prompt templates, etc.
  • Experience with containers and orchestration tools (e.g., Docker, Kubernetes).
  • Familiarity with cloud infrastructure (AWS, GCP) and infrastructure as code (e.g., Terraform).
  • Proven ability to solve complex problems and work independently in fast-moving environments.

Nice to haves:

  • Experience with modern LLM serving frameworks such as vLLM, SGLang, TensorRT-LLM, or text-generation-inference.

About the company

Scale AI company logo

Scale AI

Actively Hiring
Accelerate the development of AI applications1001-5000 Employees
  • B2B
  • Scale Stage
    Rapidly increasing operations
  • Top Investors
    This company has received a significant amount of investment from top investors
  • YC Funded
    Startup funded by Y Combinator
  • Valuation $1B+
    This company has a valuation of $1B or more
Learn more about Scale AI image

Funding

AMOUNT RAISED
$277.6M
FUNDED OVER
5 rounds
Rounds
D
$155000000
Series D - Dec 2020+4

Perks

Healthcare benefits
HEALTH, DENTAL & VISION COVERAGE
Choose the benefit plan that suits you and your family's needs.
EASY TO USE 401(K) VIA GUIDELINE
Plan and invest for the future with a 401(K) via Guideline.
Parental leave
Equity benefits
Work from home policy
Vacation policy
GENEROUS PAID TIME OFF
Enjoy time to travel or plan a staycation - whatever you need to relax and recharge.
Company meals
FREE LUNCH AND DINNER
Stay fueled for the workday. Lunch and dinner are on us.
Commuter benefits
Company events
REGULAR SOCIAL ACTIVITIES
From karaoke to off-sites, we take time to celebrate the wins and connect with teammates.
FLEXIBLE WORK HOURS
Flexible hours allow you to work when you are most productive.

Similar Jobs

impakter.com company logo
impakter.com
Empower your sustainable lifestyle - Take action everyday
Nextdoor company logo
Nextdoor
Nextdoor is the private social network for your neighborhood
Instrumentl company logo
Instrumentl
The best platform for nonprofits looking to grow revenue (YC S16)
Coast company logo
Coast
Coast is re-imagining B2B payments, beginning with fleet and fuel
Kick Health company logo
Kick Health
The Online Performance Medicine Clinic for Energizing Sleep and Confident Presentations
Archesys company logo
Archesys
Improving the government services that impact everyday lives