
Senior Machine Learning Engineer, AI Platform
Affinity.co
full-time
Posted on:
Location Type: Hybrid
Location: San Francisco • California • United States
Visit company websiteExplore more
Salary
💰 $160,000 - $235,000 per year
Job Level
Tech Stack
About the role
- Own the full ML lifecycle: Take projects from ideation to production, including feature engineering, model selection, deployment, and model observability and evaluation
- Translate business needs into ML solutions: Gather product requirements and translate them into robust ML system design requirements
- Build recommendation and ranking systems: Architect and launch ranking and recommendation infrastructure from scratch, initially via integrated off-the-shelf models, and evolving to targeted and customized solutions in the long term
- Solve complex problems: Work on a variety of information extraction, information storage and information retrieval problems for both structured and unstructured data
- Collaborate cross-functionally: Partner with cross-functional (product, infra, data engineering, and software engineering) teams to build robust, high-scale systems that underlie all of our data processing and ML Operations
Requirements
- 5+ years of experience in software engineering and/or Machine Learning experience in applying machine learning in production
- Hands-on experience developing ranking or recommendation systems from scratch, deployed at scale using techniques such as learn-to-rank, explainable recommendations
- Strong understanding of machine learning techniques, including clustering and decision trees
- Experience with serving ML models for streaming and batch inference at scale
- Experience with vector or graph databases
- Proficiency in Python and modern ML frameworks (PyTorch, Scikit-learn, or similar)
- Track record of building maintainable, testable, and production-grade codebases
- Experience with observability tools for online and offline model evaluation, A/B testing, and tracing for AI applications
Benefits
- We cover your medical, dental, and vision insurance premiums with comprehensive PPO, HDHP and HMO options (in CA)
- Offer flexible personal & sick days to support your well-being
- We offer a 401(k) plan to help you plan for your future
- Provide an annual education budget and a comprehensive L&D program
- Reimburse monthly for things like home internet, meals, and wellness memberships/equipment to support your overall health and happiness
- Virtual team-building activities and socials to keep our team connected, because building strong relationships is key to success.
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
machine learningranking systemsrecommendation systemsfeature engineeringmodel selectionmodel observabilityclusteringdecision treesPythonproduction-grade code
Soft Skills
problem solvingcollaborationcross-functional teamwork