San Francisco, CA; USA (Remote) Hybrid Full-time USD 160,000 – 235,000 / year

Affinity is hiring a Senior Machine Learning Engineer, AI Platform

Responsibilities

  • 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.

Nice to Have

  • Experience with dataset engineering, including data curation, augmentation, and synthesis, to assist ML model improvement.
  • Experience with graph-based recommendation systems, such as graph NN.
  • Experience with packaging, CI/CD and pipeline automation.

Benefits

  • We cover your medical, dental, and vision insurance premiums with comprehensive PPO, HDHP and HMO options (in CA), and offer flexible personal & sick days to support your well-being.
  • We offer a 401(k) plan to help you plan for your future.
  • We provide an annual education budget and a comprehensive L&D program.
  • We 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.

Additional Information

  • We are dedicated to building a diverse, inclusive, and authentic workplace, so if you’re excited about this role, but your past experience doesn’t perfectly align with the qualifications above, we encourage you to apply anyways. You may be just the right candidate for this or other roles.
  • We iterate quickly. As such, you must be comfortable embracing ambiguity, be able to cut through it, and deliver value to our customers. We are candid, transparent, and speak our minds while simultaneously caring personally with each person we interact with. We make data-driven decisions and make the best decision for the moment based on the information available.
Required Skills
Python
About company
Affinity
Affinity is a relationship intelligence platform that provides CRM solutions for venture capital, private equity, investment banking, and other financial sectors, leveraging AI and data enrichment technologies.
All jobs at Affinity Visit website
Job Details
Department AI Platform
Category data
Posted 11 days ago