Member of Technical Staff, LLM Infrastructure
Remote • San Mateo • FullTime
Posted 10mo ago
Job Location
San Mateo
Tech Stack
Remote Work Policy
Fully remote
Employment Type
FullTime
Categories
LLM Engineer
About the job
As a Software Engineer on the AI Infrastructure team, you will help design the core systems that power Fireworks AI’s generative AI platform. You will build infrastructure and tools that ensure the reliability, performance, quality, and availability of our AI system. Your 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, product, and performance teams to deliver infrastructure that bridges the gap between our customers and the ultra-performant proprietary Fireworks inference engine.
Responsibilities
- Contribute to the design and development of scalable backend infrastructure for distributed training, inference, and data pipelines.
- Build and maintain core backend services including LLM CI/CD pipelines, control planes, and model serving systems.
- Support performance optimization, cost efficiency, and reliability improvements across compute, storage, and networking layers.
- Build frameworks and safeguards to ensure high model quality.
- Collaborate with performance, training, and product teams to translate research and product needs into infrastructure solutions.
- Participate in code reviews, technical discussions, and CI/CD processes.
Requirements
- 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).
- Basic understanding of LLM knowledge (e.g., context length, disaggregated prefill, KV cache memory estimation).
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.