Staff Machine Learning Engineer - Retention
Taskrabbit2 hours ago
Base Salary
$170k - $225k/yr
Responsibilities
- Own the reliability and performance of Taskrabbit’s core ranking system and optimize tasker-to-job matching and First-Time Right rates.
- Build ML solutions for repeat purchase frequency, personalized recommendations, category discovery, customer lifetime value, dynamic pricing, segmentation, churn reduction, and predictive quality interventions.
- Own the complete ML lifecycle from feature engineering and model training through evaluation, deployment, monitoring, and production optimization.
- Build scalable ML infrastructure and data pipelines supporting reproducible feature engineering and deployment across real-time, near-real-time, and batch contexts.
- Develop monitoring and observability systems for data quality and model performance, and optimize training, inference, and evaluation algorithms.
- Write clean, efficient, maintainable code and contribute to code reviews, documentation, and software engineering best practices.
Requirements
- BS, MS, or PhD in Computer Science, Statistics, Operations Research, or a related quantitative field.
- 8+ years of industry experience building and deploying high-quality, production-grade machine learning models and systems.
- Strong machine learning theory and hands-on experience in search, ranking, recommender systems, pricing or elasticity modeling, or predictive analytics.
- Strong software engineering skills with proficiency in one or more programming languages, including Python.
- Experience with ML libraries such as Scikit-learn, LightGBM, XGBoost, TensorFlow, or PyTorch.
- Proficiency in SQL for writing complex queries and transforming data.
- Experience building REST API-based services.
- Experience with Docker, Kubernetes, Kafka, Airflow, data warehouses such as Snowflake, Redshift, or BigQuery, and data lakes.
- Familiarity with dbt for transforming and testing data is a plus.
- Familiarity with Infrastructure as Code tools such as GitHub Actions and CI/CD pipelines.
- Excellent communication skills for presenting complex findings and recommendations to technical and non-technical audiences.
- Experience in marketplace or platform contexts involving ranking, matching, and pricing is preferred.
Benefits
- Hybrid work arrangement requiring two days in the San Francisco or NYC hub every Tuesday and Wednesday.
- Employer-paid health insurance and a 401k match with immediate vesting for US-based employees.
- Generous and flexible time off for global employees, including two company-wide closure weeks.
- Taskrabbit product stipends, wellness, productivity, and education stipends.
- IKEA discounts and reproductive health support.
- Benefits vary by country of employment.
Tech Stack
Amazon RedshiftApache AirflowApache KafkadbtDockerGitHub ActionsGoogle BigQueryKubernetesLightGBMPythonPyTorchscikit-learnSnowflakeSQLTensorFlowXGBoost
Categories
About Taskrabbit
Taskrabbit builds a two-sided marketplace and mobile app that connects consumers with local Taskers for home services such as furniture assembly, moving help, and handyman work. The company generates revenue through service fees on bookings and partnerships, and was founded in 2008 and is headquartered in San Francisco. Taskrabbit operates across North America and Europe and has been a subsidiary of IKEA since its 2017 acquisition.