Member of Technical Staff, AI Training InfrastructureActive$200K–$350K

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

  • 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

What they're looking for

  • Collaborate with researchers to implement and optimize training methodologies
  • Analyze and improve efficiency, scalability, and cost-effectiveness of training systems
  • Troubleshoot complex performance issues in distributed training environments
  • Bachelor's degree in Computer Science, Computer Engineering, or related: field, or equivalent practical experience