Staff Machine Learning Engineer, Computer Vision

Segment (Twilio); San Francisco (CA); Hybrid; USD 189,308 - 389,753; Full time

Job summary

Pinterest Labs in the United States seeks researchers to advance state-of-the-art visual encoders and multimodal models for recommender systems. You will explore billion-scale datasets on GPUs and publish work at top venues.

You will mentor junior researchers and help shape Pinterest's visual generative strategy while collaborating with a small, fast-moving team. Strong publications or production ML experience preferred.

Qualifications

• Experience building and training computer vision models.
• Experience with multimodal representations and visual language modeling.
• 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 PyTorch.
• Strong collaboration skills and a demonstrated ability to work effectively in a small, fast-moving team.
• MS or PhD in Machine Learning or related academic areas, or equivalent work experience.
• Publications at top ML conferences.
• Experience using AI coding assistants for development, debugging, testing, and refactoring.

Responsibilities

• Prototype state-of-the-art visual encoders powering recommender systems and internal visual language models.
• Experiment with billion-scale datasets and large-scale GPU computing.
• Build flexible visual reasoning tools such as composed image retrieval, promptable detection/segmentation, and embedding/generative models.
• Read research papers, participate in group discussions, and help brainstorm the company's visual generative strategy.
• Help collect relevant visual instruction training data shared across multimodal representation, image retrieval, text-to-image generation, and visual language modeling.
• Publish and share work through conferences, paper submissions, and blog posts.
• Mentor junior researchers and research interns within Pinterest Labs.

Skills

Computer vision

Multimodal representations

PyTorch

Publications

Model training at scale

Collaboration

AI coding assistants

Education

MS/PhD in ML or related

Equivalent work experience

Tools

PyTorch

Cursor/Copilot/Codex

Job description

Responsibilities

• Prototype state‑of‑the‑art visual encoders that power Pinterest's recommender systems and internal visual language models.
• Experiment with billion‑scale datasets and gain hands‑on experience with large‑scale GPU computing.
• Build flexible visual reasoning tools such as composed image retrieval, promptable detection/segmentation, and instruction‑tuned embedding and generative models.
• Read research papers, participate in group discussions, and help brainstorm the company's overall visual generative strategy.
• Help collect relevant visual instruction training data that can be shared across multimodal representation, composed image retrieval, text‑to‑image generation, and visual language modeling.
• Publish and share your work through conferences, paper submissions, and blog posts.
• Mentor junior researchers and research interns within the Pinterest Labs organization.

Qualifications

• Research engineers and scientists with experience building and training computer vision models.
• Experience with multimodal representations and visual language modeling is strongly preferred.
• 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 Ph.D. in Machine Learning or related academic areas, or equivalent work experience.
• Publications at top ML conferences.
• Experience using Cursor, Copilot, Codex, or similar AI coding assistants for development, debugging, testing, and refactoring.

Relocation

This position is not eligible for relocation assistance. Visit our PinFlex page to learn more about our working model.

In‑Office Requirement

The role will need to be in the office for in‑person collaboration 1-2 times per quarter and therefore can be situated anywhere in the country.

Salary

US based applicants only: $189,308 - $389,753 USD.

Our Commitment to Inclusion

We are an equal opportunity employer and make employment decisions on the basis of merit. All qualified applicants will receive consideration for employment without regard to race, color, ancestry, national origin, religion or religious creed, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender, gender identity, gender expression, age, marital status, status as a protected veteran, physical or mental disability, medical condition, genetic information, or characteristics (or those of a family member), or any other consideration made unlawful by applicable federal, state or local laws. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements.