Staff Machine Learning Engineer
Core
Build and deploy large-scale foundational models (LLMs, vision) and AI agents to solve complex eCommerce scenarios and enhance customer experience.
Role type
Staff Machine Learning Engineer (GenAI/Foundation Models)
Builds
E2E AI pipelines, predictive/compound AI models, and foundational model training systems for eCommerce.
Domain
eCommerce, Generative AI, Large Language Models (LLM), Knowledge Graphs, Multi-modal AI
Deliverable
production ML models
Required skills
Large-scale foundational model development, LLM post-training and deployment, AI/LLM architecture design, statistical modeling, SQL, Python, PySpark, Scala, R, PyTorch, TensorFlow, Spark, BigQuery, AI coding tools (Cursor, Windsurf, Claude Code, Codex), complex data visualization, dimension reduction, cloud analytical stacks (lakes, warehouse, ETL), inference platforms (AWS, Azure)
Preferred skills
Experience with semantic knowledge bases, AI agents, NLP, AIGC, ML, RL, and multi-modal models
Technologies
PyTorch, TensorFlow, Spark, BigQuery, AWS, Azure, Cursor, Windsurf, Claude Code, Codex
Responsibilities
Develop large language and vision understanding/generation models; Train foundational models for learning systems; Define processes and experiments for E2E foundation models; Build E2E AI pipelines for storing/servicing foundational models; Mentor team members on data lakes, normalization, dimension reduction, and statistical models; Ensure model training security and compliance; Collaborate cross-functionally to align models with company AI strategy
Seniority
Staff, hands-on IC with mentorship
