Palo Alto Networks

Principal Machine Learning Engineer (AI-ML Data Scientist - ADEM)

Palo Alto Networks
Apply
1 day ago
Santa Clara, CA, USAStaff+
H1B sponsor

Base Salary

$163k - $264k/yr

Responsibilities

  • Lead the design and implementation of ML-, LLM-, and GenAI-powered agents supporting planning, multi-step execution, self-correction, and human or agent collaboration.
  • Build reasoning pipelines, RAG retrieval systems, and knowledge graph integrations for autonomous task execution and network troubleshooting.
  • Build, fine-tune, or adapt LLMs, multimodal models, and specialized agent models, integrating traditional AI/ML and statistical models.
  • Drive end-to-end development from prototype to production with emphasis on scalability, reliability, safety, observability, and performance.
  • Establish practices for prompt design, agent memory, tool usage, and context optimization.
  • Lead technical design for the AIOps platform and contribute to core backend rearchitecture and platform-level services.
  • Evaluate agent frameworks, LLM tooling, and generative AI technologies to guide architecture and roadmap decisions.
  • Mentor engineering teams and integrate services with adjacent ecosystems such as Prisma Access and Cortex.

Requirements

  • 8+ years of software engineering experience with production-grade ML, GenAI, or agentic workflows deployed at scale.
  • Strong Python experience for AI/ML and professional Go or Java experience for backend microservices.
  • Hands-on SQL experience, including designing and restructuring BigQuery schemas.
  • Experience building and managing scalable data pipelines using Dataflow, Google ADK, and GKE.
  • Understanding of retrieval systems, RAG, prompt engineering, model evaluation, and statistical modeling.
  • Ability to build platform-level services that integrate with adjacent ecosystems.
  • BS or MS in Computer Science, Data Science, Engineering, or a related field.
  • Knowledge of L3-L7 networking is desirable.

Benefits

  • Most teams work from the office full time, with flexibility when needed.
  • The role is eligible for immigration sponsorship.
  • Employee benefits are described by the company, with compensation potentially including restricted stock units and a bonus.

Tech Stack

GoGoogle BigQueryJavaPythonSQL
Palo Alto Networks

About Palo Alto Networks

10,000+ employees

Palo Alto Networks builds cybersecurity products for enterprises and governments, including next-generation firewalls, cloud security (Prisma Cloud/Prisma Access), and SOC tools (Cortex). It sells subscriptions and appliances with cloud-delivered threat intelligence, securing networks, endpoints, and applications across data centers and public clouds. Founded in 2005 and headquartered in Santa Clara, California, the company is publicly traded on the NYSE under the ticker PANW.

Contact me