About this role
Senior Software Engineer - Model Training at Baseten. Location: San Francisco or New York. Role: Design distributed training infrastructure, Implement scalable pipelines, Optimize training performance Requirements: 4+ years of software engineering in ML infrastructure; distributed systems experience; strong GPU/accelerator optimization; production systems design and operation; excellent communication. Category: Software Development Seniority: Senior Level Tools: PyTorch, Transformers, Lightning, TRL, Kubernetes, Ray, Slurm, AWS, GCP, Azure Commitment: Full Time Workplace: Onsite Languages: English