Software Engineer, AI Training Infrastructure

1 Month ago • 3 Years + • Devops • $175,000 PA - $220,000 PA

Job Summary

Job Description

Fireworks AI is building the future of generative AI infrastructure, offering a platform with high-quality models and fast, scalable inference. The Training Infrastructure Engineer will design, build, and optimize the infrastructure for large-scale model training operations. This role involves collaborating with AI researchers and engineers to create robust training pipelines, optimize distributed training workloads, and ensure reliable model development. Responsibilities include designing scalable infrastructure, developing distributed training pipelines for LLMs and multimodal models, optimizing training performance across multiple GPUs, nodes, and data centers, implementing monitoring and debugging tools, architecting data storage solutions, automating infrastructure provisioning, and troubleshooting performance issues in distributed training environments.
Must have:
  • Bachelor's degree in Computer Science or related field
  • 3+ years of experience with distributed systems
  • Experience with ML infrastructure
  • Experience with PyTorch
  • Proficiency in cloud platforms (AWS, GCP, Azure)
  • Experience with containerization
  • Experience with orchestration (Kubernetes, Docker)
  • Knowledge of distributed training techniques
Good to have:
  • Master's or PhD in Computer Science
  • Experience training large language models
  • Experience with multimodal AI systems
  • Experience with ML workflow orchestration tools
  • Background in optimizing high-performance computing
  • Familiarity with ML DevOps practices
  • Contributions to open-source ML projects
Perks:
  • Meaningful equity in a fast-growing startup
  • Competitive salary
  • Comprehensive benefits package

Job Details

About Us:

Here at Fireworks, we’re building the future of generative AI infrastructure. Fireworks offers the generative AI platform with the highest-quality models and the fastest, most scalable inference. We’ve been independently benchmarked to have the fastest LLM inference and have been getting great traction with innovative research projects, like our own function calling and multi-modal models. Fireworks is funded by top investors, like Benchmark and Sequoia, and we’re an ambitious, fun team composed primarily of veterans from Pytorch and Google Vertex AI.

The Role: 

As a Training Infrastructure Engineer, you'll design, build, and optimize the infrastructure that powers our large-scale model training operations. Your work will be essential to developing high-performance AI training infrastructure. You'll collaborate with AI researchers and engineers to create robust training pipelines, optimize distributed training workloads, and ensure reliable model development.

Key Responsibilities:

  • Design and implement scalable infrastructure for large-scale model training workloads
  • Develop and maintain distributed training pipelines for LLMs and multimodal models
  • Optimize training performance across multiple GPUs, nodes, and data centers
  • Implement monitoring, logging, and debugging tools for training operations
  • Architect and maintain data storage solutions for large-scale training datasets
  • Automate infrastructure provisioning, scaling, and orchestration for model training
  • Collaborate with researchers to implement and optimize training methodologies
  • Analyze and improve efficiency, scalability, and cost-effectiveness of training systems
  • Troubleshoot complex performance issues in distributed training environments

Minimum Qualifications:

  • Bachelor's degree in Computer Science, Computer Engineering, or related field, or equivalent practical experience
  • 3+ years of experience with distributed systems and ML infrastructure
  • Experience with PyTorch
  • Proficiency in cloud platforms (AWS, GCP, Azure)
  • Experience with containerization, orchestration (Kubernetes, Docker)
  • Knowledge of distributed training techniques (data parallelism, model parallelism, FSDP)

Preferred Qualifications:

  • Master's or PhD in Computer Science or related field
  • Experience training large language models or multimodal AI systems
  • Experience with ML workflow orchestration tools
  • Background in optimizing high-performance distributed computing systems
  • Familiarity with ML DevOps practices
  • Contributions to open-source ML infrastructure or related projects

Total compensation for this role also includes meaningful equity in a fast-growing startup, along with a competitive salary and comprehensive benefits package. Base salary is determined by a range of factors including individual qualifications, experience, skills, interview performance, market data, and work location. The listed salary range is intended as a guideline and may be adjusted.

Base Pay Range (Plus Equity)

$175,000 - $220,000 USD

Why Fireworks AI?

  • Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.
  • Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.
  • Ownership & Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.
  • Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.

Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Similar Jobs

deel. - QA Automation Engineer

deel.

Bosnia And Herzegovina (Remote)
2 Weeks ago
AeroSpike - Senior Manager of Technical Support

AeroSpike

Mountain View, California, United States (On-Site)
2 Months ago
bytedance - Software Engineer, Global Payment Privacy & Security

bytedance

San Jose, California, United States (On-Site)
4 Months ago
bytedance - Senior Software Development Engineer, Virtual Network

bytedance

San Jose, California, United States (On-Site)
6 Months ago
Rackspace Technology - Operational Lead – Day-to-Day Operations

Rackspace Technology

Riyadh, Riyadh Province, Saudi Arabia (On-Site)
1 Month ago
eBay - Software Architect, SEO Platforms

eBay

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
Sailpoint - Principal SRE (Site Reliability Engineer)

Sailpoint

United States (Remote)
3 Months ago
Qualcomm - Senior Engineer - Android Platform Engineer

Qualcomm

Hyderabad, Telangana, India (On-Site)
3 Months ago
CyberArk - Senior Software Engineer, Golang, Cloud Native

CyberArk

Santa Clara, California, United States (Hybrid)
3 Months ago
Britive - Software Engineer (Cloud)

Britive

Bengaluru, Karnataka, India (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

bytedance - Cloud Solutions Technical Account Manager

bytedance

Singapore (On-Site)
4 Months ago
Philips - Consumer Marketing Manager

Philips

Stockholm, Stockholm County, Sweden (Hybrid)
2 Weeks ago
Rackspace Technology - Software Engineer IV

Rackspace Technology

India (Remote)
4 Months ago
Roblox - Supply Chain Manager - Server Hardware

Roblox

San Mateo, California, United States (On-Site)
1 Month ago
Anthology  Inc  - Associate Software Engineer II

Anthology Inc

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Playstation - IT Coordinator

Playstation

Santa Monica, California, United States (On-Site)
1 Month ago
broadcom - Maintenance Technician

broadcom

Breinigsville, Pennsylvania, United States (On-Site)
4 Weeks ago
Square - Account Management Coordinator

Square

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)
1 Month ago
Blinkhealth - Supervisor, Pharmacy Operations (Claims and Patient Outreach)

Blinkhealth

Pittsburgh, Pennsylvania, United States (On-Site)
3 Months ago
Scout - Senior Specialist, Product Support & Safety

Scout

Columbia, South Carolina, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Redwood City, California, United States

Starkflow - AI Product Engineer

Starkflow

San Francisco, California, United States (On-Site)
4 Months ago
Exploding Kittens - Senior Staff Accountant

Exploding Kittens

Los Angeles, California, United States (On-Site)
2 Months ago
Axon - Senior Accountant, Revenue Policy & Enablement

Axon

Seattle, Washington, United States (Hybrid)
1 Month ago
GoMotive - Sales Development Intern

GoMotive

Austin, Texas, United States (On-Site)
1 Month ago
lifechruh - Life.Church Central Internship

lifechruh

Edmond, Oklahoma, United States (On-Site)
10 Months ago
Sierra - Senior Commercial Counsel

Sierra

San Francisco, California, United States (On-Site)
4 Weeks ago
extreme network - Credit Analyst

extreme network

Raleigh, North Carolina, United States (Remote)
1 Month ago
Guardian - Assistant General Counsel, Regulatory Affairs

Guardian

New York, United States (Hybrid)
3 Weeks ago
WebFX - Social Media Specialist Account Manager (Open to Remote or On-Site)

WebFX

Harrisburg, Pennsylvania, United States (Remote)
9 Months ago
quience - Manager Growth Marketing, Paid Social (Pinterest)

quience

United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Devops Jobs

Mistral AI - Applied AI Engineer, Senior/Staff Devops/SRE

Mistral AI

Singapore (On-Site)
3 Months ago
Nightfall AI - Senior ML Platform Backend Engineer

Nightfall AI

San Francisco, California, United States (Hybrid)
2 Months ago
Capgemini - Automation Engineer

Capgemini

Bengaluru, Karnataka, India (On-Site)
2 Months ago
playrix  - Senior C++ Software Engineer (Build System)

playrix

Ireland (Remote)
8 Months ago
Cadence - Software Architect

Cadence

Shanghai, China (On-Site)
3 Months ago
Palo Alto Networks - Principal Consultant, Cloud Security, Proactive Services (Unit 42)

Palo Alto Networks

United States (Remote)
1 Month ago
Axon - Solutions Architect II

Axon

Scottsdale, Arizona, United States (Hybrid)
3 Months ago
Synamedia - Software Engineer (Node JS, GoLang, AWS)

Synamedia

Bengaluru, Karnataka, India (On-Site)
11 Months ago
Ness - ECS Platform Engineer – C++ / Mission-Critical Systems

Ness

Timișoara, Timiș, Romania (On-Site)
1 Month ago
Attio - Site Reliability Engineer

Attio

London, England, United Kingdom (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (Hybrid)

New York, United States (Hybrid)

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (Remote)

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (Hybrid)

Redwood City, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Fireworks AI

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug