Staff Machine Learning Engineer, Computer Vision

Pinterest's Visual Modeling team is seeking a Staff Machine Learning Engineer to develop large-scale visual encoders and multimodal representations. The role involves prototyping state-of-the-art models, training on billion-scale datasets, and contributing to generative AI projects. Ideal candidates have experience in computer vision, multimodal learning, and large-scale model training with PyTorch.

• Prototype state-of-the-art visual encoders for recommender systems and visual language models.
• Experiment with billion-scale datasets and large-scale GPU computing.
• Build flexible visual reasoning tools like composed image retrieval and instruction-tuned models.
• Mentor junior researchers and contribute to research publications.

• Experience building and training computer vision models.
• Strong preference for multimodal representations and visual language modeling.
• Track record of research contributions or shipping ML models to production.
• Hands-on experience with PyTorch and large-scale model training.
• M.S. or PhD in Machine Learning or related field.

• Publications at top ML conferences.
• Experience using AI coding assistants like Cursor, Copilot, or Codex.