Machine Learning Engineer, Images

📋 Description

• Evaluate image generation and identity preservation papers/models.
• Develop and deploy image generation and image analysis pipelines.
• Monitor and fix production issues that affect users.
• Fine-tune/optimize models to improve character consistency, prompt response, and latency.
• Design and run experiments to benchmark model performance and metrics.
• Collaborate with cross-functional teams to ship ML features.

🎯 Requirements

• Interest in AI image generation.
• Deep ML foundation in image synthesis.
• 5+ years software engineering experience, preferably in services.
• 2+ years building production ML models in industry/research.
• Strong Python programming and deploying Python services.
• Familiar with Stable Diffusion, DiT, ViT, TensorFlow, PyTorch, Diffusers, CUDA.
• Experience building scalable ML infra on Baseten, GCP, AWS or Azure.
• Strong teamwork and cross-functional collaboration.

🎁 Benefits

• Competitive salary and generous equity
• Medical, dental, and vision insurance – premiums covered
• 42 days PTO: 15 PTO, 10 sick, 15 holidays, 2 floating
• Generous parental leave and fertility support
• 401(k) retirement savings plan
• Lifestyle spending account – $500/month
• Complimentary lunch and snacks for in-office employees
• One Medical membership and more