GrepJob
Affinity.co

Senior Machine Learning Engineer, AI Platform

Affinity.co
Apply
3 months ago
Remote, United StatesSenior
H1B Sponsor

Base Salary

$160k - $235k/yr

Responsibilities

  • Own the full ML lifecycle from ideation to production.
  • Translate business needs into robust ML system design requirements.
  • Build recommendation and ranking systems from scratch.
  • Solve complex information extraction and retrieval problems.
  • Collaborate with cross-functional teams to build high-scale systems.

Requirements

  • 5+ years of experience in software engineering and/or machine learning in production.
  • Hands-on experience developing ranking or recommendation systems at scale.
  • Strong understanding of machine learning techniques, including clustering and decision trees.
  • Experience with serving ML models for streaming and batch inference.
  • Experience with vector or graph databases.
  • Proficiency in Python and modern ML frameworks like PyTorch or Scikit-learn.
  • Track record of building maintainable, testable, and production-grade codebases.
  • Experience with observability tools for model evaluation and A/B testing.

Benefits

  • Comprehensive medical, dental, and vision insurance coverage.
  • 401(k) plan for retirement planning.
  • Annual education budget and comprehensive learning and development program.
  • Monthly reimbursements for wellness-related expenses.
  • Virtual team-building activities to foster team connection.

Tech Stack

PythonPyTorchscikit-learn

Categories

AI & MLData Science