Member of Technical Staff, AI Training Infrastructure

Fireworks AI
San Mateo, New York
Workplace: HybridFull timeUSD 200,000 - 350,000 annuallyFunction: Education & TrainingExperience: 3+ yearsEducation: bachelorsSkills: []

Design, build, and optimize infrastructure for large-scale AI model training. Develop distributed training pipelines for LLMs and multimodal models, optimize performance across GPUs, nodes, and data centers, and ensure reliable model development. Implement monitoring, logging, and debugging for training operations, architect data storage for training datasets, automate provisioning and orchestration, and collaborate with researchers to improve training methodologies, efficiency, scalability, and cost.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Fireworks AI
Fireworks AI
1 week ago

Member of Technical Staff, AI Training Infrastructure

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 4 hours agoStatus: Live
Reposted: similar role first listed 1 year ago

Job Summary

Design, build, and optimize infrastructure for large-scale AI model training. Develop distributed training pipelines for LLMs and multimodal models, optimize performance across GPUs, nodes, and data centers, and ensure reliable model development. Implement monitoring, logging, and debugging for training operations, architect data storage for training datasets, automate provisioning and orchestration, and collaborate with researchers to improve training methodologies, efficiency, scalability, and cost.
Location: San Mateo, New York
Workplace: Hybrid
Employment Type: Full time
Job Function: Education & Training
Seniority: Mid level

Key Responsibilities

  • •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.
  • •Automate infrastructure provisioning, scaling, and orchestration for model training.

Pay and Benefits

Salary: USD 200,000 - 350,000 annually
Equity and Bonus:Equity

Key Requirements

  • •Bachelor's degree in Computer Science/Computer Engineering or equivalent practical experience.
  • •3+ years of experience with distributed systems and ML infrastructure.
  • •Experience with PyTorch.
  • •Proficiency in cloud platforms (AWS, GCP, Azure).
  • •Experience with containerization and orchestration (Kubernetes, Docker).
Experience:3+ yearsML infrastructureDistributed systemsLLMsMultimodal AI
Education:Bachelor's in Computer Science, Computer Engineering, or related field
Tech Stack:PyTorchAWSGCPAzureKubernetesDockerFSDPData parallelismModel parallelismDistributed training

Company Brief

Fireworks AI
Develops AI-driven tools to generate and optimize visual marketing content for brands and creators, automating production of short-form videos and multimedia assets for social platforms to improve engagement and scale creative workflows.
Industry: SaaS
Website