We are seeking a Senior Staff AI/ML Engineer to lead the design, development, and deployment of advanced AI/ML and Generative AI solutions that drive innovation across our engineering, manufacturing, and business operations.
This role requires deep expertise in machine learning engineering, data science methodologies, and large-scale AI systems, combined with strong domain intuition in semiconductor/NAND environments. You will operate at the intersection of data, models, and real-world systems, translating complex, high-impact problems into scalable AI solutions.
Key Responsibilities
AI/ML, Data Science & Domain Integration
- Develop advanced machine learning, statistical, and optimization models to solve complex business and engineering problems
- Apply end-to-end data science rigor: problem framing, feature engineering, modeling, validation, and impact measurement
- Work with large-scale datasets (e.g., process, yield, test, operational data) to derive actionable insights
- Integrate domain knowledge (semiconductor/NAND processes, constraints, variability) directly into model design and interpretation
Generative AI & Intelligent Systems
- Design and deploy GenAI solutions (LLMs, RAG pipelines, agent-based systems) for engineering and operational use cases
- Build knowledge-driven systems leveraging enterprise data (documents, logs, process data)
- Develop evaluation frameworks to ensure quality, grounding, and reliability of GenAI outputs
- Apply GenAI to enable decision support, automation, and productivity at scale
End-to-End System Architecture
- Architect production-grade AI/ML systems, including:
- Data pipelines and feature engineering frameworks
- Model training and experimentation platforms
- Real-time and batch inference systems
- Design systems that balance accuracy, scalability, latency, and cost efficiency
- Establish patterns for integrating AI into existing enterprise platforms and workflows
MLOps & Production Excellence
- Build and scale MLOps pipelines (CI/CD, model registry, monitoring, drift detection)
- Ensure reproducibility, observability, and governance of AI systems
- Lead efforts to operationalize models into reliable, maintainable production systems
Collaboration & Mentorship
- Partner with cross-functional teams (engineering, manufacturing, product, IT) to translate requirements into solutions
- Mentor engineers and data scientists in advanced modeling, system design, and GenAI capabilities
Apply on company website Milpitas, CA On-site Full-time 172,783.00-286,178.00