Lead AI/ML Engineer – Security AI & LLMs Pune, Maharashtra, India

SonicWall; Pune District; On-site; INR 4,000,000 - 7,000,000; Full time

Benefits offered by this job

Hybrid work model

Competitive salary

Job summary

SonicWall in Pune, India, seeks a Lead AI/ML Engineer to design, build, and scale the intelligence layer powering our next-generation security products. You will translate complex datasets into production-ready AI capabilities driving automated analysis and intelligent recommendations.

The role requires 6+ years deploying ML in production, with 1+ year focused on LLMs and generative AI; strong Python, PyTorch/TensorFlow, LangChain, LlamaIndex, and Hugging Face are essential.

Qualifications

• 6+ years of professional experience deploying ML models in production.
• 1+ year focused on LLMs and generative AI.
• Experience in the cybersecurity domain is highly valued.
• Strong AI/NLP stack: Python, PyTorch/TensorFlow, LangChain, LlamaIndex, Hugging Face.

Responsibilities

• Architect AI Systems: Design and deploy Retrieval-Augmented Generation pipelines and LLM apps for security telemetry and unstructured logs.
• Model Development & Tuning: Train, fine-tune, and evaluate ML/DL models for anomaly detection and classification.
• Ensure AI Trust & Alignment: Implement guardrails, evaluation frameworks, and safety alignment for deterministic outputs.
• Establish MLOps: Build scalable ML pipelines from data preprocessing to production monitoring.

Skills

Python

PyTorch

TensorFlow

LangChain

LlamaIndex

Hugging Face

LLMs

Vector DBs

Semantic Search

Docker

Kubernetes

SageMaker

Azure ML

Vertex AI

Education

Bachelor's degree

Job description

Lead AI/ML Engineer – Security AI & LLMs

Pune, Maharashtra, India

SonicWall is a cybersecurity forerunnerwith more than 30 years of expertise and is recognized as a leading partner-first company, ensuring our partners and their customers are never alonein the fight against cybercrime. With the ability to build, scale and manage security across the cloud, hybrid and traditional environments in real-time, SonicWall provides relentless security against the most evasive cyberattacks across endless exposure points for increasingly remote, mobile and cloud-enabled users. With its own threat research center, SonicWall can quickly and economically provide purpose-built security solutions to enable any organization—enterprise, government agencies and SMBs—around the world. For more information, visitwww.sonicwall.com or follow us onTwitter ,LinkedIn ,Facebook andInstagram .

Role Overview

As our lead AI/ML Engineer , you will design, build, and scale the intelligence layer powering our next-generation security products. You will turn complex datasets—including configurations, security alerts, and raw network logs—into production-ready AI capabilities that drive automated analysis, reasoning, and intelligent recommendations.

Key Responsibilities

• Architect AI Systems: Design and deploy robust Retrieval-Augmented Generation (RAG) pipelines, agentic workflows, and LLM applications tailored to parse and reason over security telemetry and unstructured logs.
• Model Development & Tuning: Train, fine-tune, and evaluate ML/DL models for pattern matching, anomaly detection, and classification across massive, heterogeneous security datasets.
• Ensure AI Trust & Alignment: Implement strict guardrails, evaluation frameworks, and safety alignment techniques to ensure all model outputs and recommendations are deterministic, safe, and highly accurate.
• Establish MLOps: Build and maintain scalable ML pipelines (from data preprocessing to model monitoring in production), collaborating closely with Data and Product Engineers to ensure low-latency, secure inferences.

Required Experience & Skills

• Experience: 6+ years of professional experience deploying machine learning models into production, with 1+ years focused on LLMs and generative AI. Experience in the cybersecurity domain is highly valued.
• AI/NLP Stack: Deep expertise in Python, PyTorch/TensorFlow, and frameworks like LangChain, LlamaIndex, or Hugging Face.
• Data & Vector Systems: Hands-on experience with vector databases (e.g., Pinecone, Qdrant, Milvus) and semantic search strategies over complex, technical datasets.
• Production Engineering: Solid understanding of API design, containerization (Docker/Kubernetes), and cloud ML platforms (AWS SageMaker, Azure ML, or GCP Vertex AI).
• Mindset: A builder's mentality—comfortable setting technical direction, navigating ambiguity, and executing fast in a small team.

#LI-DS9

#LI-Pune

#LI-Hybrid

SonicWall is an equal opportunity employer.

We are committed to creating a diverse environment and are an equal opportunity employer. All qualified applicants receive consideration for employment without regard to race, color, ethnicity, religion, sex, gender, gender identity and expression, sexual orientation, national origin, disability, age, marital status, veteran status, pregnancy, or any other basis prohibited by applicable law.

At SonicWall, we pride ourselves on recruiting a diverse mix of talented people and providing active security solutions in 100+ countries.