Principal Machine Learning Inference Engineer - Remote

About the Role

We are seeking a Principal Machine Learning Inference Engineer to join our dynamic team at Palo Alto Networks. This remote position is perfect for those who are passionate about leveraging AI to enhance cybersecurity solutions. As a Principal Machine Learning Inference Engineer, you will play a pivotal role in shaping the future of our AI security platform, Prisma AIRS.

What You'll Do

• Lead the architectural design of a highly scalable, low-latency ML inference platform.
• Provide technical leadership and mentorship to the engineering team, promoting best practices in MLOps and software engineering.
• Drive the strategy for model and system performance, implementing advanced optimization techniques.
• Establish engineering standards for automated model deployment and operational excellence.
• Act as a key technical liaison to ensure cohesion across the Prisma AIRS platform.
• Tackle complex technical challenges in large-scale inference, focusing on security threats and performance goals.

Requirements

• BS/MS or Ph.D. in Computer Science or a related field.
• Professional experience in software engineering with a focus on MLOps and ML systems.
• Expert-level programming skills in Python; knowledge of Go, Java, or C++ is a plus.
• Hands-on experience designing large-scale distributed systems on major cloud platforms (GCP, AWS, Azure).
• Proven track record of leading architecture for complex ML systems and MLOps pipelines.
• Mastery of ML frameworks such as TensorFlow and PyTorch.
• Experience with advanced inference optimization tools (ONNX, TensorRT).
• Familiarity with CI/CD pipelines and automation tools (e.g., Jenkins, GitLab CI).

Nice to Have

• Experience with data infrastructure technologies (e.g., Kafka, Spark).
• Understanding of modern LLM inference engines.
• Low-level performance optimization experience.

What We Offer

• Competitive salary ranging from $141,000 to $228,075 per year.
• Comprehensive relocation support for candidates moving to the United States.
• Visa sponsorship for eligible candidates.
• A collaborative work environment that values innovation and diversity.
• Opportunities for professional development and growth.