Senior AI Training Infrastructure Engineer

Fireworks

San Francisco (CA)

In loco

USD 150.000 - 230.000

Tempo pieno

14 giorni+

Aumenta le possibilità di ricevere una risposta da questo datore di lavoro

Il tuo curriculum e una lettera di presentazione personalizzati, che corrispondono esattamente a ciò che sta cercando.

Supera i filtri ATS

Descrizione del lavoro

Fireworks seeks a Training Infrastructure Engineer to design, build, and optimize the infrastructure powering our large-scale model training operations. You’ll collaborate with AI researchers to create robust training pipelines, optimize distributed workloads, and ensure reliable model development across multiple GPUs and data centers.

You will shape scalable storage, automate provisioning, and implement monitoring to improve efficiency and cost-effectiveness.

Competenze

Bachelor's degree in Computer Science, Computer Engineering, or related field, 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, orchestration (Kubernetes, Docker).
Knowledge of distributed training techniques (data parallelism, model parallelism, FSDP).

Mansioni

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.
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.