Skip to content
SquareSquareSeattle, WA

Staff Machine Learning Engineer, Modeling, Support

Lead end-to-end delivery of ML initiatives for customer support, focusing on conversational AI and generative AI. Partner with cross-functional teams to define ML roadmaps and drive R&D efforts for next-generation chatbot architectures.

161k – 284k/yr
On-site10+ YOEML Engineering

About the role

The Role

Block's Support ML Modeling team is a central driver of innovation in customer support experiences across our entire ecosystem—including Cash App, Square, and other business units. We are dedicated to advancing the state of intelligent, automated support through machine learning and generative AI. From customer-facing chatbots to smart internal tools for agents, our team builds high-impact, scalable systems that improve support quality, efficiency, and accessibility.

We're building the future of support at Block: one powered by AI, voice interfaces, and smart automation. We're looking for candidates with a passion for intelligent systems, practical ML experience, and a desire to build product-driven solutions. Our ideal candidate is a leader in the field, with a proven track record of building ML systems that can accelerate our ability to scale our conversational AI initiatives.

You Will

  • Lead the end-to-end delivery of multiple ML initiatives, from planning and design through rollout, documentation, and long-term maintenance
  • Partner strategically with risk, product, engineering, design, and operations leaders to define and drive long term ML roadmaps, informed by domain expertise and industry trends
  • Guide the team’s direction, identifying new ML/AI opportunities and advising leadership on strategic tradeoffs and opportunities
  • Drive R&D efforts exploring next-generation chatbot architectures using LLMs, RAG, fine-tuning, and real-time inference
  • Design, deploy, and maintain ML models powering conversational agents including support chatbots across Cash App, Square, and other Block products
  • Develop ML-powered tools and real time recommendation systems that enhance support agent effectiveness and customer outcomes
  • Act as the technical representative for your team’s systems and programs, clearly communicating work and results to stakeholders, cross-functional partners, and external audiences

You Have

  • 10+ years of experience in machine learning, applied AI, or product ML roles, with deep technical expertise
  • Demonstrated leadership capabilities, with the ability to influence and align cross-functional teams while directly shaping the work of peers through communication, context sharing, and technical guidance
  • A track record of delivering organizational wide impact, shaping systems, frameworks, or initiatives that raise the bar across multiple teams or functions
  • Proven ability to ship end to end ML features from problem framing through deployment and long-term maintenance
  • Demonstrated experience with language models, dialog systems, or generative AI in production
  • Strong foundation in NLP, deep learning, or ML infrastructure best practices.
  • Excellent communication skills, with the ability to represent the team’s work to leadership and stakeholders
  • Enthusiasm for R&D and pushing the boundaries of applied AI to transform conversational systems

Technologies We Use and Teach

  • Python, PyTorch, TensorFlow, or JAX

Every benefit we offer is designed with one goal: empowering you to do the best work of your career while building the life you want. Remote work, medical insurance, flexible time off, retirement savings plans, and modern family planning are just some of our offering.

Skills

Machine LearningGenerative AINLPDeep LearningPythonPyTorchTensorFlowJAXLanguage ModelsDialog Systems

Similar roles

ML Engineering jobs
Mariana Minerals

Staff Machine Learning Engineer

Mariana MineralsAnn Arbor, MI

Staff ML Engineer setting technical direction for autonomous mineral refining using reinforcement learning and simulation. Owns modeling, validation, and deployment of control systems on live industrial equipment.

160k – 200k/yr
On-site8+ YOEML Engineering
Roe AI

Founding Staff Engineer, AI Agent

Roe AISan Mateo, CA

Builds sophisticated AI agent products for investigating financial crimes, leads technical designs and teams. Requires 4-10 years experience in software development, Python, cloud infrastructure, and hybrid SF Bay Area work.

160k – 200k/yr
Hybrid4+ YOEML Engineering
Harper

Senior Member of Technical Staff

HarperSan Francisco, CA

Senior engineer owning end-to-end AI infrastructure including LLM orchestration, agent architectures, evaluation systems, and scalable platforms for insurance quoting. Requires 3-6 years experience with production AI systems and strong architectural skills.

160k – 230k/yr
On-siteML Engineering
Gusto

Retirement AI Staff Engineer

GustoDenver, CO +2

Staff Engineer building and shipping LLM-powered retirement features end-to-end. Owns architecture decisions, integrates models, and collaborates closely with product and design.

164k – 247k/yr
Hybrid7+ YOEML Engineering
Harvey

Mid/Senior/Staff Software Engineer, Agents

HarveySan Francisco, CA +1

Build AI agent systems for legal workflows, optimizing performance via prompt engineering, model selection, tools, and evals. Requires 3+ years experience, Python proficiency, and LLM/agent framework expertise for mid/senior/staff levels.

165k – 312k/yr
On-siteML Engineering