Skip to content
SquareSquareUnited States

Principal Engineer, AI Systems

Principal-level IC building and scaling production autonomous agents and agentic workflows across Block's ecosystem using frontier LLMs. Requires 15+ years experience shipping AI systems from zero to production scale.

319k – 479k
Remote15+ YOEML Engineering

About the role

Responsibilities

  • Ship, architect, build, and own end-to-end delivery of autonomous agents and agentic workflows that deliver real business value for Block, ensuring reliability, safety, and performance at scale
  • Design agent orchestration systems including planning, tool use, memory, evaluation, and multi-agent coordination at production scale
  • Integrate and optimize frontier LLMs into agent architectures, making decisions on model selection, fine-tuning, prompt engineering, and context retrieval strategies
  • Drive deeper model optimization work (fine-tuning, distillation, RLHF) where it unlocks agent capability or efficiency
  • Lead detailed technical planning by breaking down ambitious objectives into concrete, sequenced tasks with clear ownership and execution paths
  • Provide technical mentorship and guidance to engineers across experience levels, elevating team capabilities through code review, pairing, and knowledge sharing
  • Partner closely with technical and non-technical stakeholders to translate business objectives into agent-powered product experiences
  • Keep Block at the frontier by continuously evaluating emerging AI capabilities and making pragmatic tradeoffs across model performance, latency, cost, and user experience
  • Foster a culture of technical excellence, high-quality delivery, rapid experimentation, and learning within your team and beyond

Requirements

  • 15+ years of experience in software engineering or machine learning, with recent professional experience building autonomous agents or agentic workflows in production
  • Deep experience building autonomous agents or agentic workflows in production environments—not just prototypes or demos
  • Fluency in the core primitives of agentic systems: context management, planning, tool use, memory, evaluation, and multi-step reasoning
  • Experience bringing frontier LLM capabilities into production products, with hands-on experience in prompt engineering, retrieval-augmented generation, and model optimization
  • A track record of taking AI-powered products from zero to scale in fast-paced, product-driven environments, with the judgment that comes from operating in production
  • Strong software engineering fundamentals with the ability to write production-quality code and make sound architectural decisions
  • Experience providing technical leadership within teams—you've shaped technical direction, driven execution, and elevated others
  • Product-minded engineering approach—you think in terms of user outcomes, not just model metrics
  • Excellent collaboration and communication skills, with ability to build alignment across engineering, product, and design
  • Comfort navigating extreme ambiguity in a domain that's evolving weekly
  • Alignment with Block's mission of economic empowerment and using technology to create access and opportunity

Nice-to-Haves

  • Experience at leading AI organizations with a track record of translating research into production agent systems
  • Background building or scaling agentic products at startups (including early-stage or pivoting companies)
  • Experience with model fine-tuning, distillation, or RLHF to improve agent performance
  • Familiarity with agent evaluation, safety, and alignment challenges in production contexts

Skills

PythonLLMsPrompt EngineeringRAGFine-TuningRLHFAgent OrchestrationMulti-Agent SystemsModel OptimizationProduction Ai Systems

Similar roles

ML Engineering jobs
Shield AI

Principal Engineer, AI and Data Platform Engineering (R4941)

Shield AISan Francisco, CA

Leads the design, build, and operation of AI and data platform for autonomy systems, managing training, simulation, data pipelines, MLOps, and deployment across on-premise, cloud, and edge environments. Requires deep expertise in scalable ML infrastructure and compute strategy.

320k – 490k
On-siteML Engineering
Pinterest

Principal Machine Learning Engineer, Ads Delivery

PinterestSan Francisco, CA +2

Develop backend systems, statistical models, and experiments to optimize Pinterest's ads marketplace, balancing short- and long-term objectives. Requires 10+ years experience, CS/ML degree, and strong engineering/math skills.

315k – 551k
Hybrid10+ YOEML Engineering
Zoox

Principal Software Engineer - Autonomy Behaviors

ZooxFoster City, CA

Lead technical development of ML algorithms for next-generation ML Planner. Drive innovations in imitation learning, reinforcement learning, and model scaling while mentoring ML developers.

324k – 492k
Hybrid10+ YOEML Engineering
Airbnb

Principal Machine Learning Engineer- LLM Fine-tuning and Optimization

AirbnbUnited States

Principal MLE focused on fine-tuning and optimizing large language models for production, building scalable AI pipelines, and shaping Airbnb's ML platform strategy. Requires 10+ years experience and a PhD.

292k – 365k
Remote10+ YOEML Engineering
OpenAI

Principal Software Engineer, Codex Cyber

OpenAISan Francisco, CA

Principal engineer on OpenAI's Codex Cyber team building AI-powered security products for trusted defenders. Define roadmaps, shape model training and safeguards, build evaluations for cyber capabilities and risks, and translate frontier research into production tools while collaborating across product, research, safety, and external partners.

347k – 490k
On-site10+ YOEML Engineering