Affinity.co

Senior Machine Learning Engineer, AI Platform

Affinity.co

full-time

Posted on:

Location Type: Hybrid

Location: San FranciscoCaliforniaUnited States

Visit company website

Explore more

AI Apply
Apply

Salary

💰 $160,000 - $235,000 per year

Job Level

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