Staff Software Engineer / Tech Lead, Anyscale

• Multiple years of experience building, scaling, and maintaining complex software systems in production
• Proven experience leading or mentoring engineering teams in a technical capacity
• Expertise in machine learning frameworks (e.g., PyTorch, TensorFlow, XGBoost)
• Hands-on experience with distributed systems and designing fault-tolerant infrastructure
• Excellent communication and collaboration skills

Desirable

• Anyscale is looking for a staff software engineer to lead the Model Training Infrastructure team
• The Model Training Infrastructure team leads the development and optimization of Ray’s distributed training libraries, focusing on enabling large-scale ML workloads
• The team owns and maintains widely adopted open source libraries like Ray Train for distributed model training and Ray Tune for distributed hyperparameter tuning
• Thinking deeply about delightful, programmatic interfaces for machine learning engineers to scale model training
• Build and rethink distributed training architectures to scale seamlessly from laptop to the cloud
• Implement and innovate on distributed training algorithms like elastic training to improve model training performance
• Working with and leading a robust open source community around the Ray project
• Engage directly with ML infrastructure teams around the world to iterate and build the best training infrastructure
• Advocate and share your work broadly with the ML community through talks, tutorials, and blog posts
• N the day-to-day basis, you will drive the technical direction of the team, mentor engineers, and deliver high-impact projects
• You’ll shape the vision for what training infrastructure looks like for enterprises around the world and remain hands-on with the code and product development