Skip to content
SquareSquareSan Francisco, CA

Senior Machine Learning Engineer, Risk Modeling

Senior ML Engineer building and deploying production risk and fraud detection models at scale for Cash App and Square. Owns end-to-end ML systems and partners cross-functionally to reduce financial loss.

Salary not listed
On-site8+ YOEML Engineering

About the role

Responsibilities

  • Partner with product, engineering, data science, policy, and operations to design and productionize ML-driven risk solutions at scale
  • Own end-to-end machine learning systems — from problem definition and modeling to deployment, monitoring, and iteration
  • Lead technical decision-making within your workstreams and influence ML strategy and planning
  • Build tooling and processes that improve the speed, reliability, and impact of the ML development lifecycle
  • Apply state-of-the-art modeling techniques and third-party data sources to improve detection and decision-making
  • Investigate emerging fraud, abuse, and risk patterns to proactively inform product safeguards and policy
  • Collaborate closely with ML platform and engineering teams to ensure models operate reliably in real time and at scale

Requirements

  • 8+ years of industry experience in machine learning, applied AI, or related fields
  • Bachelor’s degree in a quantitative field (Computer Science, Engineering, Statistics, Physics, Applied Math); Master’s or PhD preferred
  • Proven experience independently designing, deploying, and maintaining ML solutions in production
  • Strong familiarity with techniques such as tree-based models, deep learning, transfer learning, or reinforcement learning
  • Experience influencing technical direction and collaborating with cross-functional partners at scale
  • Strong communication skills, sound judgment, and an ownership mindset

Technologies

  • Python (NumPy, Pandas, scikit-learn, XGBoost, PyTorch, TensorFlow/Keras)
  • PySpark, MLflow, workflow orchestration tools (Airflow, Prefect)
  • GCP (Vertex AI), AWS
  • Snowflake, MySQL, Tableau, Mode
  • Containerization, CI/CD, and production ML best practices

Skills

PythonNumPypandasscikit-learnXgboostPyTorchTensorFlowPysparkMLflowAirflowGCPAWSSnowflakeMySQL

Similar roles

ML Engineering jobs
Scale AI

Senior Machine Learning Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Senior ML Engineer building observability, evaluation frameworks, and improvement loops for production agentic AI systems. Requires 5+ years production ML/LLM experience, strong grounding in agent design or evaluation, and hands-on work taking systems from prototype to scale.

216k – 270k/yr
On-site5+ YOEML Engineering
Scale AI

Senior Software Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Build platform infrastructure for observing, evaluating, and improving production AI agents at scale. Requires 4+ years software engineering experience with ML/LLM systems, strong backend/distributed systems skills, and collaboration with ML engineers.

216k – 270k/yr
On-site5+ YOEML Engineering
Ambient.ai

Senior Software Engineer, AI Infrastructure

Ambient.aiRedwood City, CA

Build and optimize scalable AI infrastructure for real-time inference, evaluation, and continuous improvement of LLMs, LVMs, computer vision, and multimodal models on large-scale video data. Requires 4+ years production ML systems experience, strong Python skills, and expertise in inference optimization and model serving.

168k – 205k/yr
Hybrid4+ YOEML Engineering
SpotOn

Senior Software Engineer

SpotOnChicago, IL +3

Senior AI Engineer building agentic workflows and orchestration layers to automate manual business processes for operations, sales, and customer support teams. Requires 7+ years software engineering experience, production LLM/agent expertise, and strong Python/TypeScript skills.

160k – 190k/yr
Hybrid7+ YOEML Engineering
Snowflake

Senior AI Engineer, Enterprise

SnowflakeMenlo Park, CA

Senior AI Engineer building scalable backend services, distributed systems, and reusable AI platform components (agent frameworks, evaluation pipelines, SDKs) that power Snowflake's internal agentic applications for GTM teams. Requires 5+ years software engineering experience, strong Python/FastAPI skills, and production LLM experience.

156k – 224k/yr
On-site5+ YOEML Engineering