AI Infrastructure Engineer

Fireworks

San Mateo

On-site

PHP 1,800,000 - 3,000,000

Full time

14 days+
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Job summary

Fireworks AI is seeking a Software Engineer on our AI Infrastructure team to design the core systems powering our generative AI platform. You will help build infrastructure and tools that ensure the reliability, performance, quality, and availability of our AI system.

You''ll partner with the cloud infrastructure, product, and performance teams to deliver scalable infrastructure that bridges the gap between customers and our ultra-performant Fireworks inference engine.

Qualifications

  • Bachelor’s degree in Computer Science, Engineering, or a related technical field (or equivalent practical experience).
  • 3 years of experience in software engineering, with a focus on infrastructure or ML systems.
  • Strong programming skills in Python, Go, or a similar language.
  • Proven experience in ML infrastructure and tooling (e.g., PyTorch, MLflow, Vertex AI, SageMaker, Kubernetes, etc.).
  • Basic understanding of LLM knowledge (e.g., context length, disaggregated prefill, KV cache memory estimation, etc)

Responsibilities

  • Contribute to the design and development of scalable backend infrastructure that supports distributed training, inference, and data pipelines
  • Build and maintain core backend services such as LLM CI/CD pipeline, control plane, and model serving systems
  • Support performance optimization, cost efficiency, and reliability improvements across compute, storage, and networking layers
  • Building frameworks and safeguards to ensure Fireworks AI has the best model quality in the industry
  • Collaborate with performance, training, and product teams to translate research and product needs into infrastructure solutions
  • Participate in code reviews, technical discussions, and continuous integration and deployment processes

Skills

Python
Go
ML infrastructure
Kubernetes
PyTorch
LLM knowledge
Open-source ML engines

Education

Bachelor’s degree in Computer Science, Engineering, or related field

Tools

MLflow
Vertex AI
SageMaker

Job description

Fireworks AI is seeking a Software Engineer on our AI Infrastructure team to design the core systems powering our generative AI platform. You will help build infrastructure and tools that ensure the reliability, performance, quality, and availability of our AI system.

You''ll partner with the cloud infrastructure, product, and performance teams to deliver scalable infrastructure that bridges the gap between customers and our ultra-performant Fireworks inference engine.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior Backend Engineer, AI Platform & Orchestration
Senior Backend Engineer, AI Platform & Orchestration

Fireworks • San Mateo

On-site
PHP 900,000 - 1,300,000
Senior Reliability Engineer, AI Infrastructure
Senior Reliability Engineer, AI Infrastructure

Fireworks • San Mateo

On-site
PHP 900,000 - 1,300,000
Member of Technical Staff, LLM Infrastructure
Member of Technical Staff, LLM Infrastructure

Fireworks • San Mateo

On-site
PHP 1,800,000 - 3,000,000
Member of Technical Staff, Software Engineer
Member of Technical Staff, Software Engineer

Fireworks • San Mateo

On-site
PHP 900,000 - 1,300,000
AI Infra & Workflow Orchestration Engineer
AI Infra & Workflow Orchestration Engineer

Fireworks • San Mateo

On-site
PHP 250,000 - 350,000
Applied ML Engineer - Build & Deploy Customer AI Apps
Applied ML Engineer - Build & Deploy Customer AI Apps

Fireworks • San Mateo

On-site
PHP 1,200,000 - 1,800,000
Member of Technical Staff - Reliability Engineering
Member of Technical Staff - Reliability Engineering

Fireworks • San Mateo

On-site
PHP 900,000 - 1,300,000
IT DevOps Engineer
IT DevOps Engineer

Fireworks • San Mateo

On-site
PHP 250,000 - 350,000
Product Manager, Platform
Product Manager, Platform

Fireworks • San Mateo

On-site
PHP 900,000 - 1,500,000
Applied Machine Learning Engineer
Applied Machine Learning Engineer

Fireworks • San Mateo

On-site
PHP 1,200,000 - 1,800,000