Principal Machine Learning Engineer (RAG Pipeline Optimiation)

$151,000 - $225,000/Yr

Palo Alto Networks - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Professional, Scientific, and Technical Services

About the position

The role at Palo Alto Networks focuses on leveraging advanced machine learning techniques to address network security challenges. The position involves creating and delivering AI and machine learning solutions that enhance threat detection and device protection. The successful candidate will work in a collaborative environment, contributing to the development of innovative security products and features that impact a wide range of customers.

Responsibilities

• Design and develop workflow frameworks using formal methods and algorithms.
• Create and deliver training materials for team members and stakeholders.
• Engage and contribute to the security community through collaborations.
• Develop and deliver new product and mission features to the security platform.
• Utilize advanced machine learning techniques on large datasets to discover threats and provide intelligence.

Requirements

• M.S. or Ph.D. degree in Computer Science, Mathematics, Electrical Engineering, or a related field, or equivalent military experience.
• 8+ years of industry experience in machine learning techniques and data analytics.
• 8+ years of experience in design, algorithms, and data structures.
• Expertise in one or more programming languages such as Java, Python, or Rust.
• Experience in designing and deploying high-performance RAG systems for real-world applications.

Nice-to-haves

• In-depth knowledge of reranking algorithms (RRF, BM25, DPR, etc.) and experience in fine-tuning models to optimize retrieval performance.
• Experience with NLP, Recommender Systems, and LLM.
• Familiarity with Formal Methods toolchain (z3, cvc5, TLA+).

Benefits

• FLEXBenefits wellbeing spending account with over 1,000 eligible items.
• Mental and financial health resources.
• Personalized learning opportunities.