Lead AI/ML Engineer – Security AI & LLMs

Posted Yesterday
Be an Early Applicant
Pune, Mahārāshtra, IND
In-Office
Senior level
Security • Cybersecurity
The Role
Lead design and build AI capabilities for security products: architect RAG/LLM pipelines and agentic workflows, train and tune ML/DL models for anomaly detection and classification, implement safety guardrails and evaluation frameworks, and establish scalable MLOps (data preprocessing to production monitoring) for low-latency, secure inferences.
Summary Generated by Built In

SonicWall is a cybersecurity forerunner with more than 30 years of expertise and is recognized as a leading partner-first company, ensuring our partners and their customers are never alone in 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, visit www.sonicwall.com or follow us on TwitterLinkedInFacebook and Instagram.

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.

Applicant Privacy Notice

Skills Required

  • 6+ years professional experience deploying machine learning models into production
  • 1+ years experience focused on LLMs and generative AI
  • Experience in the cybersecurity domain
  • Proficiency in Python
  • Experience with PyTorch or TensorFlow
  • Experience with frameworks like LangChain, LlamaIndex, or Hugging Face
  • Hands-on experience with vector databases (Pinecone, Qdrant, Milvus) and semantic search strategies
  • API design experience and production engineering skills
  • Containerization and orchestration (Docker, Kubernetes)
  • Experience with cloud ML platforms (AWS SageMaker, Azure ML, or GCP Vertex AI)
  • Build and maintain scalable MLOps pipelines including model monitoring in production
  • Implement guardrails, evaluation frameworks, and safety/alignment techniques for model outputs
  • Builder's mentality; ability to set technical direction and execute in a small team

SonicWall Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about SonicWall and has not been reviewed or approved by SonicWall.

  • Healthcare Strength Health, dental, vision, mental-health support, and wellness programs are part of the package. Feedback suggests coverage breadth is a dependable component of total rewards.
  • Leave & Time Off Breadth Generous PTO, paid holidays and sick time, and bereavement leave are described alongside flexible or remote options. Feedback suggests ample time away meaningfully supports balance.
  • Retirement Support A 401(k) and core financial protections such as life and disability insurance are included. Feedback suggests these fundamentals meet standard expectations for many.

SonicWall Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Milpitas, CA
1,832 Employees
Year Founded: 1991

What We Do

SonicWall has been fighting the cyber-criminal industry for over 25 years defending small, medium-size businesses and enterprises worldwide. Backed by research from the Global Response Intelligent Defense (GRID) Threat Network, our award-winning real-time breach detection and prevention solutions, coupled with the formidable resources of over 10,000 loyal channel partners around the globe, are the backbone securing more than a million business and mobile networks and their emails, applications and data. This combination of products and partners has enabled a real-time cyber defense solution tuned to the specific needs of the more than 500,000 global businesses in more than 215 countries and territories.

Similar Jobs

JPMorganChase Logo JPMorganChase

Client Data

Financial Services
Remote or Hybrid
2 Locations
289097 Employees

LogicMonitor Logo LogicMonitor

Senior Sales Engineer

Artificial Intelligence • Cloud • Information Technology • Machine Learning • Software
Easy Apply
Remote or Hybrid
India
1100 Employees

Mondelēz International Logo Mondelēz International

Organization Capability Specialist

Big Data • Food • Hardware • Machine Learning • Retail • Automation • Manufacturing
Remote or Hybrid
India
90000 Employees

Capco Logo Capco

Product Manager

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
India
6000 Employees

Similar Companies Hiring

Credal.ai Thumbnail
Software • Security • Productivity • Machine Learning • Artificial Intelligence
Brooklyn, NY
Milestone Systems Thumbnail
Artificial Intelligence • Security • Software • Analytics • Big Data Analytics
Lake Oswego, OR
1500 Employees
NODA AI Thumbnail
Artificial Intelligence • Information Technology • Software • Cybersecurity
Sydney, AU
54 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account