Software Engineer, AI Infrastructure

7 days ago


San Mateo, California, United States Fireworks AI Full time

About Us:
At Fireworks, we're building the future of generative AI infrastructure. Our platform delivers the highest-quality models with the fastest and most scalable inference in the industry. We've been independently benchmarked as the leader in LLM inference speed and are driving cutting-edge innovation through projects like our own function calling and multimodal models. Fireworks is a Series C company valued at $4 billion and backed by top investors including Benchmark, Sequoia, Lightspeed, Index, and Evantic. We're an ambitious, collaborative team of builders, founded by veterans of Meta PyTorch and Google Vertex AI.

The Role:

As a Software Engineer on our AI Infrastructure team, you will help design the core systems that power Fireworks AI's generative AI platform. You will help build infrastructure and tools that ensure the reliability, performance, quality, and availability of our AI system.

Our mission is to make Fireworks AI the most reliable and user friendly generative AI platform in the world. You will partner closely with our cloud infrastructure team, product team, and performance team to deliver infrastructure that bridges the gap between our customer and the ultra-performant proprietary Fireworks inference engine.

Key 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

Minimum 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 machine learning 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)

Preferred Qualifications:

  • 5+ years of experience in software engineering, with a focus on infrastructure or machine learning systems
  • Experience with open source inference engine like vLLM, Sglang, or TRT-LLM
  • Contributions to open-source infrastructure or ML projects
  • Experience in building large scale ML/MLOps infrastructure

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.



  • San Mateo, California, United States Fireworks AI Full time

    About Us:At Fireworks, we're building the future of generative AI infrastructure. Our platform delivers the highest-quality models with the fastest and most scalable inference in the industry. We've been independently benchmarked as the leader in LLM inference speed and are driving cutting-edge innovation through projects like our own function calling and...


  • San Mateo, California, United States Fireworks AI Full time

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

  • Applied AI Engineer

    1 week ago


    San Mateo, California, United States GC AI Full time

    About companyGC AI is currently an AI platform for in-house lawyers to 10x themselves. Using the latest LLMs, we are building AI agents that can use company knowledge, leverage external sources, and connect with internal tools to provide better and faster legal advice to the rest of the company. Thousands of weekly active users tell us they "cannot imagine...

  • Applied AI Engineer

    2 weeks ago


    San Mateo, California, United States GC AI Full time

    About CompanyGC AI is currently an AI platform for in-house lawyers to 10x themselves. Using the latest LLMs, we are building AI agents that can use company knowledge, leverage external sources, and connect with internal tools to provide better and faster legal advice to the rest of the company. Thousands of weekly active users tell us they "cannot imagine...


  • San Francisco, California, United States Rethink recruit Full time

     About RunloopRunloop is pioneering the next generation of AI-powered software engineering. Our platform empowers developers to build, scale, and optimize AI coding solutions—accelerating how software gets built, maintained, and evolved.We're a tight-knit, senior team of former Google and Stripe engineers, deeply focused on the hard problems of...


  • San Mateo, California, United States Guidewire Software Full time $169,000 - $253,000

    SummaryAre you a result-oriented, polyglot programmer using 12-factor app methodology to craft your Cloud products? If so, we would like to meet youGuidewire provides outstanding software for the second-largest financial services industry in the world: insurance. We deliver the core applications that Property and Casualty (P/C) insurers use to build...


  • San Mateo, California, United States Guidewire Software Full time

    SummaryReporting to VP, Infrastructure the Director, Cloud Infrastructure will own and drive the next evolution of core infrastructure services delivering innovative Network, Security and cloud services. You will encourage speed and agility through cloud adoption and automation. You will partner with Real Estate and Workplace Services to commission/refresh...


  • San Francisco, California, United States HireOTS Full time

    A stealth-stage AI infrastructure company is building a self-healing system for software that automates defect resolution and development. The platform is used by engineering and support teams to:Autonomously debug problems in production softwareFix issues directly in the codebasePrevent recurring issues through intelligent root-cause automationThe company...


  • San Diego, California, United States Shield AI Full time $156,000 - $235,000

    Founded in 2015, Shield AI is a venture-backed deep-tech company with the mission of protecting service members and civilians with intelligent systems. Its products include the V-BAT and X-BAT aircraft, Hivemind Enterprise, and the Hivemind Vision product lines. With nine offices and facilities across the U.S., Europe, the Middle East, and the Asia-Pacific,...


  • San Francisco, California, United States Eventual Full time

    About EventualEvery breakthrough AI application, from foundation models to autonomous vehicles, relies on processing massive volumes of images, video, and complex data. But today's data platforms (like Databricks and Snowflake) are built on top of tools made for spreadsheet-like analytics, not the petabytes of multimodal data that power AI. As a result,...