
Senior Machine Learning Engineer, AI Platform
Affinity.co3 months ago
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.