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 will develop and productionize large-scale visual encoders, vision-language models, and diffusion models, while collaborating with research, product, and infrastructure teams. The role also involves advancing visual AI research, building multimodal data and reasoning tools, publishing research, and mentoring junior researchers.
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
- Experience with multimodal representations and visual language modeling is strongly preferred
Benefits
- The position is also eligible for equity.
- This role will need to be in the office for in-person collaboration 1-2 times/quarter and therefore can be situated anywhere in the country.
Company Overview
Company H1B Sponsorship