smartrecruiters
Staff AI/ML Engineer
SanDisk
Bengaluru, IND
Full-time
Senior · 9+ years experience
Hybrid
Discovered Today
PyTorchTensorFlowPythonApache SparkSQLMLflow
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
PyTorchTensorFlowPython
Key Skills for This Role
PyTorchTensorFlowPythonApache SparkSQLMLflow
Full Job Posting
Key Responsibilities
- Lead the end-to-end design, development, and deployment of machine learning models and GenAI solutions for manufacturing intelligence, yield optimization, supply chain analytics, and enterprise data platforms.
- Architect and implement data pipelines, feature stores, and model serving infrastructure on Lakehouse and cloud-native architectures (Databricks, Azure).
- Develop prompt engineering strategies and feedback loops for LLM optimization; capture and normalize LLM interactions into reusable Knowledge Artifacts.
- Implement Retrieval-Augmented Generation (RAG) pipelines, including document parsing, chunking, vectorization, and semantic search using embeddings and vector databases. Support domain-specific ontologies, taxonomies, governance and versioning.
- Architect Agentic Harness and Design workflows using industry-standard frameworks for autonomous task orchestration and multi-step reasoning, with safe and controlled execution via constraints, policies, and fallback paths.
- Collaborate with cross-functional teams to define KPIs, success metrics, and business impact measurements for AI/ML initiatives.
- Mentor intermediate and junior engineers; conduct code reviews and establish best practices for ML engineering.
- Present technical findings and strategic recommendations to senior leadership, distilling complex results into actionable business insights.
Required Qualifications
- Master's degree in Computer Science, Data Science, Statistics, Mathematics, or a related quantitative field.
- 9+ years of professional experience in AI/ML engineering, data science, or a related role, with demonstrated ownership of production ML systems and with at least 3+ years in AI/ML solution architecture or enterprise AI implementation.
- Deep expertise in machine learning algorithms, deep learning frameworks (PyTorch, TensorFlow), and statistical modeling.
- Expertise with vector databases, embedding models, and semantic search architectures.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at SanDisk
Power BI/Tableau Developer (3-5 Years Experience)
Bengaluru, IND
MidFull-time
Discovered 3 days agoView →
Senior AI/ ML Engineer
Bengaluru, IND
Discovered TodayFull-time
Sales Manager
, IND
Discovered YesterdayFull-time
Senior Staff VLSI Design Engineer
Bengaluru, IND
Discovered YesterdayFull-time
Early Talent Recruiter & Program Specialist
Bengaluru, IND
Discovered 3 days agoFull-time
HPC Engineer
Bengaluru, IND
Discovered 3 days agoFull-time
PMO Manager
Milpitas, USA
Discovered 3 days agoFull-time
Power BI/Tableau Developer (3-5 Years Experience)
Bengaluru, IND
Discovered 3 days agoFull-time
HPC & Cloud Engineer
Bengaluru, IND
Discovered 3 days agoFull-time