Machine learning engineer

Responsibilities

• Design and build machine learning models for recommendation, search, and content understanding.
• Develop and optimize ranking systems that serve real-time decisions across 1B+ users.
• Train, evaluate, and iterate on state-of-the-art models in NLP, computer vision, and adjacent AI fields.
• Build data pipelines and experimentation frameworks to support rapid model development and deployment.
• Work closely with Product and Engineering leadership to translate customer needs into ML solutions.
• Analyze model performance and conduct experiments to improve relevance, personalization, and quality.
• Write production-quality Python and SQL.
• Iterate rapidly to improve Shaped’s platform in response to customer needs.

Requirements

• 5+ years of industry or research experience in machine learning, data science, or AI research.
• Previous experience working with recommendation or search systems in production.
• Experience in NLP, computer vision, or related AI fields.
• Strong proficiency in Python and SQL.
• Deep knowledge of mathematics, statistics, and algorithms.
• Comfortable owning projects end-to-end, from research and experimentation to production deployment.
• Track record of learning new concepts quickly
• Experience training and serving large-scale ranking or retrieval models.
• Familiarity with distributed data processing systems such as Apache Spark or Apache Kafka
• Experience with feature stores, experiment platforms, or model monitoring infrastructure.
• Experience implementing data infrastructure that processes millions of documents per day.
• Background in reinforcement learning, representation learning, or foundation models.
• Experience with Rust

Qualifications

We are looking for candidates who have a strong background in machine learning, data science, or AI research. You should have experience working with recommendation or search systems in production and be comfortable translating customer needs into ML solutions. You should also have a track record of learning new concepts quickly and experience training and serving large-scale ranking or retrieval models.

Skills

• Python
• SQL
• NLP
• Computer Vision
• Reinforcement Learning
• Representation Learning
• Foundation Models
• Rust

Benefits

• Comprehensive medical, dental, and vision insurance
• Flexible Spending Account (FSA), Health Savings Account (HSA)
• Expert Healthcare Guidance
• Retirement savings
• Company holidays each year
• Unlimited days for personal time each year
• Flex Commuter Benefits
• Office is fully-stocked with drinks and snacks to fuel your work day.
• Dinner is on us if you stay late

Pay

$140K - $200K

Schedule

Full-time