smartrecruiters
AI SW Stack Deployment Architect (12+ years)
Sandisk
Bengaluru, IND
Full-time
Senior · 12+ years experience
Onsite
Discovered 3 weeks ago
vLLMPyTorchTransformersTensorFlowJAXXLA
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
vLLMPyTorchTransformers
Role Overview
We are looking for a Software Architect (12+ years experience) to lead the application/framework layer and deployment stack for the Next Generation Accelerator AI platform. This role owns how models run on Next Generation Accelerator—from vLLM / PyTorch / TensotFlow/XLA to production deployment—ensuring correctness, performance, and scalability.
Key Skills for This Role
vLLMPyTorchTransformersTensorFlowJAXXLA
Full Job Posting
Role Overview
We are looking for a Software Architect (12+ years experience) to lead the application/framework layer and deployment stack for the Next Generation Accelerator AI platform. This role owns how models run on Next Generation Accelerator—from vLLM / PyTorch / TensotFlow/XLA to production deployment—ensuring correctness, performance, and scalability.
Key Responsibilities
- Architect integration of vLLM, PyTorch, and TensorFlow, JAX/XLA into Next Generation Accelerator stack
- Define framework → compiler → runtime APIs and contracts
- Own LLM execution behavior (batching, KV cache, streaming inference)
- Design and implement end-to-end deployment workflows (packaging, versioning, reproducibility)
- Drive performance optimization across model → framework → runtime
- Work cross-functionally with compiler, runtime, and low-level SW teams
- Support customer workloads, model onboarding, and debugging
Impact
Own customer-visible AI execution and deployment on Next Generation Accelerator , closing the gap between models and system performance , and enabling enterprise-grade AI solutions
Required Qualifications
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
Vice President, Global Procurement
Milpitas, USA
ExecutiveFull-time
The mission of the Vice President of Global Procurement is to create enterprise-wide value by driving procurement excellence, achieving best cost, enabling business continuity, providing market intelligence, and acting a
Discovered TodayView →
Bachelors - Finance & Accounting Rotation Program (2027 New College Graduates)
Milpitas, USA
EntryFull-time
ESSENTIAL DUTIES AND RESPONSIBILITIES: We are seeking highly motivated candidates who are ready to contribute hands-on alongside our Finance & Accounting teams as you develop your career. Bachelors of Science or Arts can
Discovered TodayView →
Staff Data Scientist – AI/ML, Generative AI & Agentic Systems
Milpitas, USA
JuniorFull-time
We are seeking a highly skilled and forward-thinking Staff Data Scientist with 5–7 years of experience in AI/ML, Generative AI, and Agentic AI systems to join our Advanced Supply Chain Analytics team. This role will focu
Discovered TodayView →
Finance & Accounting Internship (Bachelors, Summer 2027)
Milpitas, USA
InternIntern
Essential Duties & Responsibilities: Our Summer Internship is an intensive, exciting opportunity to join our organization. Interns will take on challenging projects and solve complex problems to help drive the company fo
Discovered TodayView →
Principal Engineer, Memory Technology Development
Milpitas, USA
DirectorFull-time
Join Sandisk's advanced memory technology development team in a rare director-level technical leadership role. As part of the device and productization team, you will drive the development of breakthrough memory technolo
Discovered TodayView →
Finance & Accounting Internship (MS, Summer 2027)
Milpitas, USA
InternIntern
Essential Duties & Responsibilities: Our Summer Internship is an intensive, exciting opportunity to join our organization. Interns will take on challenging projects and solve complex problems to help drive the company fo
Discovered TodayView →
Finance & Accounting Internship (MBA, Summer 2027)
Milpitas, USA
EntryInternship, Full Time
Discovered TodayView →
Vice President, Global Procurement
Milpitas, USA
Discovered TodayFull-time
Bachelors - Finance & Accounting Rotation Program (2027 New College Graduates)
Milpitas, USA
Discovered TodayFull-time
Staff Data Scientist – AI/ML, Generative AI & Agentic Systems
Milpitas, USA
Discovered TodayFull-time
Finance & Accounting Internship (Bachelors, Summer 2027)
Milpitas, USA
Discovered TodayIntern
Principal Engineer, Memory Technology Development
Milpitas, USA
Discovered TodayFull-time
Finance & Accounting Internship (MS, Summer 2027)
Milpitas, USA
Discovered TodayIntern
Finance & Accounting Internship (MBA, Summer 2027)
Milpitas, USA
Discovered TodayInternship, Full Time
Lead Data Analyst
Milpitas, USA
Discovered TodayFull-time