Fireworks AI · 1 day ago
Software Engineer, AI Infrastructure
United States
Full-time
Remote
Mid, Senior Level
3+ years expFireworks AI is building the future of generative AI infrastructure. As a Software Engineer on the AI Infrastructure team, you will design and develop core systems that ensure the reliability and performance of the generative AI platform, collaborating with various teams to deliver high-quality infrastructure solutions.
Artificial Intelligence (AI)Data ManagementSaaSSoftware
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
Qualification
Required
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
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
Benefits
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.
Company
Fireworks AI
Fireworks AI is an advanced platform that enables users to build, tune, and scale AI applications using open-source models
51-200 employees
H1B Sponsorship
Fireworks AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (11)
2024 (3)
2023 (1)
Funding
Current Stage
Growth StageTotal Funding
$327MKey Investors
Sequoia CapitalBenchmark
2025-10-28Series C· $230M
2025-10-28Secondary Market· $20M
2024-07-07Series B· $52M
Leadership Team
Lin Qiao
CEO and cofounder
Aishwarya Srinivasan
Head of AI Developer Relations
Recent News
2025-11-19
Tech Startups - Tech News, Tech Trends & Startup Funding
2025-11-15
Tech Startups - Tech News, Tech Trends & Startup Funding
2025-11-15
Company data provided by crunchbase