Senior Staff Machine Learning Engineer (AI/ML)

• Design and build agent-driven systems leveraging ADK to enable generative input/output orchestration across internal and customer-facing platforms.

• Define and implement the A2UI (Agent-to-User Interface) layer to translate agent reasoning and outputs into intuitive visual experiences and actionable workflows.

• Develop Model Context Protocol (MCP) integrations to enable secure and scalable context sharing between models, tools, and enterprise systems.

• Develop Agent-to-Agent (A2A) collaboration frameworks to enable multi-agent orchestration and task delegation aligned with the organization's AI North Star.

• Build and maintain reusable agent capabilities, tools, and connectors to accelerate development of AI-powered workflows.

• Create visualization and interaction layers for generative outputs, enabling traceability, explainability, and human-in-the-loop feedback loops.

• Establish best practices for prompt design, agent memory management, tool usage, and context optimization within the agent ecosystem.

• Develop evaluation frameworks and telemetry to measure agent performance, response quality, and user engagement across AI-powered experiences.

• Implement governance mechanisms for agent safety, observability, and policy compliance in generative AI systems.

• Collaborate with product and design teams to define next-generation AI-native user experiences powered by autonomous agents and generative interfaces.

• Continuously evaluate emerging agent frameworks, LLM tooling, and generative AI technologies to inform platform architecture and roadmap decisions.

• 2+ years of experience using an Object-Oriented programming language (Java/Python)

• Experience with cloud-native service development stack on GCP

• Knowledge of Object Oriented Programming design concepts

• Solid grasp of RESTful API design and micro services architecture.

• Basic understanding of machine learning concepts and familiarity with ML frameworks (e.g., TensorFlow, PyTorch) is a plus.

• Skilled in diagnosing and solving complex problems while providing detailed technical analysis

• Attention to details and high behavioral standards

• Team player with can-do attitude to tackle difficult problems and you inspire your team to do the same

• High energy and the ability to work in a fast-paced environment

• Excellent collaboration and communication with multiple teams

• Fast learner and eager to absorb new emerging technologies

• M.S./B.S. degree in Computer Science or Electrical Engineering or equivalent military experience

The compensation offered for this position will depend on qualifications, experience, and work location. For candidates who receive an offer at the posted level, the starting base salary (for non-sales roles) or base salary + commission target (for sales/com-missioned roles) is expected to be the annual range listed below. The offered compensation may also include restricted stock units and a bonus. A description of our employee benefits may be found here (https://benefits.paloaltonetworks.com/) .

$126,000.00 - $204,500.00/yr