Staff Machine Learning Engineer
Taskrabbitabout 3 hours ago
Base Salary
$170k - $225k/yr
Responsibilities
- Own the architecture, model design, and production reliability of the core ranking system.
- Serve as the primary technical point of contact for cross-functional teams on matching and ranking.
- Build automated retraining pipelines and rollback capabilities for model reliability.
- Address biases in the ranking pipeline to improve training data quality.
- Leverage user behavior signals to enhance conversion rates.
- Identify additional ML opportunities beyond the ranking system.
- Mentor junior engineers and elevate the technical standards of the team.
Requirements
- BS, MS, or PhD in Computer Science, Statistics, Operations Research, or a related field.
- 8+ years of experience in building and deploying production-grade ML systems.
- Deep expertise in ranking, recommender systems, and A/B experimentation.
- Strong architectural thinking and understanding of complex systems.
- High standard in code writing and reviewing.
- Ability to conduct critical R&D for novel problems.
- Vision for integrating state-of-the-art ML advancements.
- Effective communication with cross-functional partners.
- Proficiency in Python and SQL, with experience in ML libraries and data infrastructure.
- Experience with Infrastructure as Code and CI/CD pipelines.
- Proficiency in using AI coding tools throughout the software development lifecycle.
Benefits
- Employer-paid health insurance and a 401k match with immediate vesting for US employees.
- Generous and flexible time off, including two company-wide closure weeks.
- Taskrabbit product stipends and wellness, productivity, and education stipends.
- IKEA discounts and reproductive health support.
Tech Stack
Apache AirflowApache KafkaDockerGoogle BigQueryKubernetesLightGBMPythonPyTorchscikit-learnSnowflakeSQLTensorFlowTerraformXGBoost
Categories
AI & MLData Science