Skip to content
SquareSquareUnited States

Staff Applied Machine Learning Engineer

Build and operate production ML systems for ranking, recommendations, search, and customer intelligence signals used across product, growth, risk, and decisioning teams. Requires 12+ years of production ML experience and deep expertise in intelligent systems.

277k – 415k/yr
Remote12+ YOEML Engineering

About the role

Responsibilities

  • Build and operate production ML systems that turn customer and product context into trusted signals, rankings, recommendations, and decision capabilities.
  • Design production data and signal contracts that define intended use, freshness, provenance, confidence, eligibility, and calibration for downstream consumers.
  • Own ranking, retrieval, recommendation, search, propensity, and next-best-action systems end to end, from feature and candidate generation through serving, experimentation, monitoring, and feedback loops.
  • Evaluate customer and business impact beyond short-term conversion, including trust, fairness, access, risk, compliance, long-term engagement, and segment-level performance.
  • Partner across product, growth, data, platform, modeling, risk, and compliance to translate ambiguous goals into measurable ML system designs.
  • Use AI and agents to accelerate development, analysis, testing, documentation, and operations while exposing reusable capabilities to product services, internal tools, and AI-assisted workflows.

Requirements

  • 12+ years building and operating production software and ML systems for business-critical products.
  • Deep expertise in intelligent systems such as ranking/retrieval, recommendations, search, personalization, growth and lifecycle ML, customer intelligence, propensity/churn/LTV, next-best-action, or model-derived risk signals.
  • Strong production ML judgment across feature pipelines, model serving, experimentation, monitoring, feedback loops, online/offline consistency, and reliable signal interfaces.
  • Ability to evaluate impact beyond short-term conversion, including trust, fairness, access, risk, compliance, and long-term engagement.
  • Experience using AI-assisted engineering tools with appropriate verification, testing, and review for customer-impacting systems.

Nice to Have

  • Experience with semantic retrieval, embeddings, two-tower models, graph features, LLM-powered retrieval or decision systems, entity resolution, or real-time personalization.
  • Experience with experimentation, online evaluation, interleaving, counterfactual evaluation, multi-objective optimization, or long-term holdouts.
  • Experience building reusable feature/signal platforms, decision services, customer intelligence layers, model-derived data products, or agent-assisted operations.

Technologies

  • Python, Java, Kotlin, SQL
  • TensorFlow, PyTorch, XGBoost/LightGBM, ranking/retrieval systems, embeddings, semantic search, recommendation frameworks
  • Event streams, batch pipelines, feature stores, model-serving infrastructure, workflow orchestration, experimentation systems, and data warehouses/lakehouses
  • Cloud infrastructure, Kubernetes, observability tooling, coding agents, evaluation harnesses, and agent-assisted operations tooling

Skills

PythonJavaKotlinSQLTensorFlowPyTorchXgboostLightgbmKubernetesFeature Stores

Similar roles

ML Engineering jobs
Square

Staff Machine Learning Engineer, Underwriting and Credit

SquareUnited States

Senior IC building and maintaining ML underwriting and credit decisioning models for Cash App Borrow and Afterpay. Owns full modeling lifecycle including experimentation, calibration, deployment, and monitoring.

277k – 415k/yr
Remote10+ YOEML Engineering
Square

Staff Applied Machine Learning Engineer - Fraud & Abuse

SquareSan Francisco, CA

Staff Applied ML Engineer building and operating production ML decision systems to detect and prevent payment fraud, scams, identity abuse, and marketplace risk across Block.

277k – 415k/yr
On-site12+ YOEML Engineering
Zoox

Senior Staff Machine Learning Engineer - Perception

ZooxBoston, MA

Leads development of ML algorithms for robot perception, including scene understanding, tracking, segmentation, and multi-modal foundation models using sensor data. Requires deep expertise in deep learning, computer vision, and production ML pipelines, collaborating across autonomy teams.

277k – 389k/yr
HybridML Engineering
Zoox

Staff / Senior Staff Software Engineer - Learned Trajectory Machine Learning Engineer

ZooxFoster City, CA

Develops deep learning models using imitation and reinforcement learning to generate safe, efficient driving trajectories for autonomous vehicles. Collaborates with Perception, Planning, and Simulation teams; requires ML expertise, Python fluency, and transformer experience.

277k – 407k/yr
HybridML Engineering
Datadog

Staff Applied Scientist - Dashboards

DatadogNew York, NY

Staff Applied Scientist defining evaluation strategy and quality metrics for Datadog's AI-native Dashboards product. Owns ML/GenAI evaluation systems, builds datasets and harnesses, and drives improvements in retrieval, tool selection, and agent performance.

276k – 345k/yr
Hybrid10+ YOEML Engineering