Staff Machine Learning Engineer - ML Frameworks

Join Adobe's Firefly team as a Staff Machine Learning Engineer focused on ML frameworks. You will design, develop, and maintain robust AI/ML infrastructure solutions for large-scale generative AI models, leveraging Kubernetes and Python on AWS. This role involves improving distributed training frameworks, optimizing resource utilization, and collaborating with researchers to drive innovation in machine learning infrastructure.

• Design and maintain AI/ML infrastructure for large-scale AI models
• Implement and improve distributed training frameworks using GPUs
• Enhance orchestration, scheduling, and experimentation for model training
• Collaborate with data scientists and ML researchers
• Drive innovation in ML infrastructure practices
• Requires PhD or Master's in CS and 5+ years of industry experience