Hiring: Member of Technical Staff, AI Training Infrastructure Company: Fireworks AI Location: San Mateo, CA Job Posted Time: 2026-09-11 10:29:34 Target Skills & Keywords : AWS, Azure, Docker, GCP, Kubernetes, LLM, PyTorch About the job Experience: •3+ years of experience with distributed systems and ML infrastructure Required Skills: •Design and implement scalable infrastructure for large-scale model training workloads •Develop and maintain distributed training pipelines for LLMs and multimodal models •Optimize training performance across multiple GPUs, nodes, and data centers •Implement monitoring, logging, and debugging tools for training operations •Architect and maintain data storage solutions for large-scale training datasets •Automate infrastructure provisioning, scaling, and orchestration for model training •Partner cross-functionally with researchers to implement and optimize training methodologies •Analyze and improve efficiency, scalability, and cost-effectiveness of training systems Qualifications: •Bachelor's degree in Computer Science, Computer Engineering, or related field, or equivalent practical experience •Proficiency in cloud platforms (AWS, GCP, Azure) •Knowledge of distributed training techniques (data parallelism, model parallelism, FSDP) •Master's or PhD in Computer Science or related field •Background in optimizing high-performance distributed computing systems •Operational familiarity with ML DevOps practices •Contributions to open-source ML infrastructure or related projects •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. Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!