Software Engineer, LLM InfrastructureActive$175K–$220K

Fireworks AI
Apply now

The opportunity

Fireworks is the platform for specialized intelligence, enabling companies to build, train, and serve AI models tailored to their own data, workflows, and products. Founded by the team behind PyTorch and backed by AMD, Atreides, Benchmark Capital, Index Ventures, Lightspeed,…

What you'll do

  • 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

What they're looking for

  • Bachelor’s degree in Computer Science, Engineering, or a related technical: field (or equivalent practical experience)
  • 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.).