Software Engineer, AI Infrastructure Job at Fireworks Ai, New York, NY

dTBsU1R5WEJRNkgxbGF0b2tZQ2tDUHhxOGc9PQ==
  • Fireworks Ai
  • New York, NY

Job Description

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.

Job Tags

Full time,

Similar Jobs

Lotus Sales

2026 Summer Sales Intern - Housing Provided + Paid Job at Lotus Sales

 ...Job Description Job Description Relocate for the summer, live with your team, get trained, and earn strong income all in one internship. What Youll Do Door-to-door sales with a team Present services + answer questions Sign up new customers Full-time... 

Nuvance Health

Gastroenterology Physician Assistant/ Nurse Practitioner Job at Nuvance Health

 ...Gastroenterology Physician Assistant/ Nurse Practitioner at Nuvance Health summary: The Gastroenterology Physician Assistant or Nurse Practitioner provides specialized care for patients with digestive health issues including irritable bowel disease, working both in... 

FillerUp Express LLC

Cargo/Sprinter Van Delivery Driver Job at FillerUp Express LLC

 ...Cargo/Sprinter Van Delivery Drivers Location: Franklin Park, IL Start Date: As soon as possible Employment Type: 1099 Independent Contractor We are seeking reliable and dependable Cargo and Sprinter Van Drivers to join our delivery team as independent contractors... 

Sevita

Caregiver Job at Sevita

 ...in all areas of required trainings. \n ~ Shall be at least 18 years of age. \n \n \n \n \n Sevita is a leading provider of home and community-based specialized health care. We believe that everyone deserves to live a full, more independent life. We provide people... 

NetWorth Realty USA

Fix and Flip Real Estate Agent Job at NetWorth Realty USA

 ...a national sales and acquisition company specializing in distressed real estate. Its not quite HGTVbut its every bit as thrilling (just without the 30-minute wrap-up). Whether you have been an agent for a year or two, or are brand new to real estate, weve got your back...