Note: The job is a remote job and is open to candidates in USA. Pinterest is a platform that helps people discover creative ideas and plan meaningful experiences. The Staff Machine Learning Engineer, Visual AI will develop large-scale visual encoders, vision-language models, and diffusion models, build visual reasoning and training-data tools, and collaborate with product and infrastructure engineers to ship AI capabilities to production.
Responsibilities
- Build state-of-the-art visual encoders, VLMs, and diffusion models that power Pinterest's visual AI capabilities
- Experiment with billion-scale image datasets, backed by large-scale GPU computing
- Build flexible visual reasoning tools such as composed image retrieval, promptable image feature computation, instruction-tuned embedding and generative editing models, and more
- Read research papers, participate in group discussions, and directly participate in brainstorming the company's overall AI strategy
- Help construct data agents to build training data that can be shared across multimodal representation, composed image retrieval, image-editing generation, and visual language modeling
- Collaborate directly with product engineers and infrastructure engineers to ship new capabilities in the core product
- Publish and share your work through conferences like CVPR and KDD, paper submissions, and blog posts
- Mentor junior researchers and research interns within the Pinterest Labs organization
- Collaborate across a team situated across San Francisco, Seattle, NYC, and remote roles
Skills
- Research engineers and scientists with experience building and training large scale vision models of all categories
- A track record of research contributions (e.g., publications, open-source work) and/or shipping ML models to production
- Hands-on experience with large-scale model training and modern deep learning frameworks (e.g., PyTorch)
- Strong collaboration skills and a demonstrated ability to work effectively in a small, fast-moving team
- M.S. or PhD in Machine Learning or related academic areas, or equivalent work experience
- Experience using AI-accelerated research tooling akin to auto-research, data agents, etc
- US based applicants only
- Experience with multimodal representations and visual language modeling is strongly preferred
Benefits
- The position is also eligible for equity.
- Flexible working arrangement: this role can be situated anywhere in the country and requires in-office collaboration 1-2 times per quarter.
Company Overview
Company H1B Sponsorship