Software Engineer, AI Training Infrastructure Job at Fireworks Ai, Redwood City, CA

YVh1eHVKMk5EYzJvdWp5Y2Q2T042QktJSlE9PQ==
  • Fireworks Ai
  • Redwood City, CA

Job Description

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

Compensation is determined by various factors including individual qualifications, experience, skills, interview performance, market data, and work location. The listed salary range for this role is a guideline and may be modified.

Redwood City Pay Range

$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.

Job Tags

Full time,

Similar Jobs

GQR Healthcare

Travel Wound Care Registered Nurse Job at GQR Healthcare

 ...Job Description GQR Healthcare is seeking a travel nurse RN Wound Care for a travel nursing job in Los Angeles, California. Job Description & Requirements ~ Specialty: Wound Care ~ Discipline: RN ~ Start Date: 02/04/2026~ Duration: 13 weeks ~40 hours... 

Slimelab llc

Slime Instructor Job at Slimelab llc

Job Description Job Description Job Overview We are seeking a passionate and dedicated Instructor to join our slime team. The ideal candidate will possess a strong commitment to fostering a positive learning environment and will be responsible for delivering high...

Bold MK

Entry Level Communications Agent Job at Bold MK

 ...Description Location: Austin, TX Job Type: Full-Time About Us: We are seeking a motivated and enthusiastic Entry Level Communications Agent to join our dynamic team. This role serves as a foundational position in our organization, aimed at fostering... 

Joon Loloi

Quality Control Manager, Home Furnishings Job at Joon Loloi

 ...Loloi, and we are committed to maintaining a safe and trustworthy hiring experience for all candidates. We are seeking a Quality Control Manager, Home Furnishings in Cartersville, GA. This Quality Control Manager will be responsible for the quality standards of... 

EY-Parthenon

EY Parthenon - Strategy and Execution (Junior and Senior Consultant) Job at EY-Parthenon

 ...(min GPA of 3.2 or equivalent). Fresh graduate or 1-2 years (Junior Consultant) and 3-5...  ...other types of transaction engagements. Experience related to mergers, acquisitions, and...  ...English. Relevant degrees and certifications (MBA, CPA, CMA, CFA, PMP) are a plus....