Principal Machine Learning Platform Engineer (Prisma AIRS) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Palo Alto Networks · 3 hours ago

Principal Machine Learning Platform Engineer (Prisma AIRS)

Palo Alto Networks is dedicated to protecting the digital way of life through innovation and cutting-edge technology. They are seeking a Principal Machine Learning Platform Engineer to lead the architectural design and strategy of their AI security platform, Prisma AIRS, while providing technical leadership and mentoring to the team.

Agentic AICloud SecurityCyber SecurityNetwork SecuritySecurity
check
Growth Opportunities
check
H1B Sponsorednote

Responsibilities

Lead the architectural design of a highly scalable, low-latency, and resilient ML inference platform capable of serving a diverse range of models for real-time security applications
Provide technical leadership and mentorship to the team, driving best practices in MLOps, software engineering, and system design
Drive the strategy for model and system performance, guiding research and implementation of advanced optimization techniques like custom kernels, hardware acceleration, and novel serving frameworks
Establish and enforce engineering standards for automated model deployment, robust monitoring, and operational excellence for all production ML systems
Act as a key technical liaison to other principal engineers, architects, and product leaders to shape the future of the Prisma AIRS platform and ensure end-to-end system cohesion
Tackle the most ambiguous and challenging technical problems in large-scale inference, from mitigating novel security threats to achieving unprecedented performance goals

Qualification

MLOpsML systemsPythonDistributed systemsKubernetesDockerTensorFlowPyTorchGoJavaC++ONNXTensorRTTransformersCNNsGNNsVLLMSGLangCUDAKafkaSparkFlinkCI/CDJenkinsGitLab CITekton

Required

BS/MS or Ph.D. in Computer Science, a related technical field, or equivalent practical experience
Extensive professional experience in software engineering with a deep focus on MLOps, ML systems, or productionizing machine learning models at scale
Expert-level programming skills in Python are required
Deep, hands-on experience designing and building large-scale distributed systems on a major cloud platform (GCP, AWS, Azure, or OCI)
Proven track record of leading the architecture of complex ML systems and MLOps pipelines using technologies like Kubernetes and Docker
Mastery of ML frameworks (TensorFlow, PyTorch) and extensive experience with advanced inference optimization tools (ONNX, TensorRT)
Demonstrated expertise with modern LLM inference engines (e.g., vLLM, SGLang, TensorRT-LLM)

Preferred

Experience in a systems language like Go, Java, or C++
A strong understanding of popular model architectures (e.g., Transformers, CNNs, GNNs)
Open-source contributions in these areas
Experience with low-level performance optimization, such as custom CUDA kernel development or using Triton Language
Experience with data infrastructure technologies (e.g., Kafka, Spark, Flink)
Familiarity with CI/CD pipelines and automation tools (e.g., Jenkins, GitLab CI, Tekton)

Benefits

Restricted stock units
Bonus

Company

Palo Alto Networks

company-logo
Palo Alto Networks is a cybersecurity company that offers cybersecurity solutions for organizations.

H1B Sponsorship

Palo Alto Networks has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (579)
2024 (482)
2023 (341)
2022 (452)
2021 (493)
2020 (235)

Funding

Current Stage
Public Company
Total Funding
$65M
Key Investors
Icon VenturesLehman HoldingsGlobespan Capital Partners
2012-07-20IPO
2008-11-03Series C· $10M
2008-08-18Series C· $27M

Leadership Team

leader-logo
Helmut Reisinger
CEO EMEA
linkedin
leader-logo
Nikesh Arora
Chairman CEO
linkedin
Company data provided by crunchbase