Senior AI Engineer

We are seeking a Senior AI Engineer to join our dynamic team. In this role, you will be instrumental in building data-driven ML/AI algorithms that enhance our search and recommendation systems. Your primary focus will be on data engineering, analysis, transformations, model training, and serving, ensuring practical and scalable applications of machine learning within our products. This position emphasizes productization and the implementation of ML/AI solutions over pure data science and research, making it ideal for professionals thriving in the fast-paced generative AI era.

Key Responsibilities

Data Engineering & Analysis

• Slice and dice analytics data to formulate hypotheses and generate ideas to improve search and recommendation performance.
• Perform comprehensive data transformations to prepare datasets for model training and evaluation.
• Build and maintain data pipelines using tools like Airflow, Kubeflow, and MLflow to support ML/AI workflows.

Model Development & Deployment

• Design, develop, and enhance machine learning and AI models tailored to product discovery and search functionalities.
• Conduct feature engineering to extract meaningful insights from historical data, search queries, product catalogs, and images.
• Collaborate with Data Engineers to integrate and scale ML components to production-level systems capable of handling large-scale data.
• Ensure seamless deployment of models, maintaining high availability and performance in cloud environments.

Algorithm Implementation & Optimization

• Dive deep into algorithm applicability, performing impact analysis to ensure models meet performance and business objectives.
• Optimize and build new algorithms to address various challenges in product discovery and search.

Productization of ML/AI Solutions

• Translate data-driven insights and models into actionable product features that enhance user experience.
• Work closely with Data Science, Product and Engineering teams to implement practical ML/AI applications that drive business outcomes.

Continuous Learning & Improvement

• Stay abreast of the latest advancements in ML/AI, particularly in generative AI and large language models (LLMs).
• Continuously refine and improve existing models and workflows based on new research and industry trends.

Qualifications

Educational Background

• BS/MS degree in Computer Science, Engineering, Mathematics, or a related discipline with a strong mathematical foundation.

Experience

• 5-8 years of experience building ML-driven, fast, and scalable ML/AI algorithms in a corporate or startup environment.

Technical Skills

• Proficient in Python with excellent programming skills.
• Strong understanding of machine learning and natural language processing technologies, including classification, information retrieval, clustering, knowledge graphs, semi-supervised learning, and ranking.
• Experience with deep learning frameworks such as PyTorch, Keras, or TensorFlow.
• Proficient in SQL and experience with data warehouses like Redshift or BigQuery.
• Experience with big data technologies such as Hadoop, Spark, Kafka, and data lakes for large-scale processing.
• Strong understanding of data structures, algorithms, and system design for building highly available, high-performance systems.
• Experience with workflow orchestration and ML pipeline tools such as Airflow, Kubeflow, and MLflow.

Specialized Knowledge

• Strong awareness of recent trends in Generative AI and Large Language Models (LLMs).
• Experience working with the GenAI stack is highly desirable.

Soft Skills

• Excellent problem-solving and analytical skills with the ability to adapt to new ML technologies.
• Effective communication skills in English, both verbal and written.
• Ability to work collaboratively in a fast-paced, agile environment.