Senior AI Architect & Enterprise Advisory (Remote)

Pure Storage, Inc.; United States; Remote; USD 172,000 - 258,000; Full time

Benefits offered by this job

Flexible time off

Wellness resources

Company-sponsored team events

Job summary

Pure Storage, Inc. is building a global Applied AI team to work directly with enterprise customers and partners—turning AI ambitions into production outcomes on Pure's platform.

This role blends science, consulting, and communication at Staff/Principal level. You will lead technical discovery, advise on end-to-end deployment, and create proof-of-value prototypes and enablement content for field teams, partners, and customers.

Qualifications

• Advanced degree in a quantitative field (CS, physics, math, engineering) or equivalent demonstrated through publications and production system experience.
• 8+ years building and deploying AI/ML systems in cloud or on-prem environments.
• Deep expertise in modern AI/ML — LLMs, distributed training, inference optimization, and evaluation/alignment frameworks.
• Fluency with the modern AI stack (PyTorch, vLLM, Ray, Kubernetes) and data platforms (Spark, Snowflake, Kafka).
• Able to scope and carry out research projects independently and in teams.
• Excellent written, verbal, and presentation skills for both data scientists and C-suite.

Responsibilities

• Lead technical discovery and advisory engagements with customers to identify high-value AI use cases relevant to their industry and data.
• Advise on end-to-end AI deployment — model selection, training/fine-tuning, inference optimization, data pipeline design, and safety frameworks — optimized for Pure's platform.
• Collaborate cross-functionally to qualify and position AI opportunities, and develop proof-of-value prototypes translating performance into business outcomes.
• Create AI enablement content for field teams, partners, and customers — technical walkthroughs, guides, workshops, and vertical use-case frameworks.
• Present at industry conferences, publish technical content, and represent Pure in engagements with technology partners (NVIDIA, AMD, cloud providers).
• Operate autonomously in ambiguous environments, scoping high-impact initiatives and driving them to completion at pace.

Skills

PyTorch

Kubernetes

GPU clusters

XGBoost

Distributed training

Evaluation/alignment frameworks

Large language models

Data platforms

Inferenz optimization

Presentations to c-suite

Education

Advanced degree in CS/Physics/Math/Engineering

Tools

PyTorch

vLLM

Ray

Kubernetes

Spark

Snowflake

Kafka

Job description

Pure Storage, Inc. is building a global Applied AI team to work directly with enterprise customers and partners—turning AI ambitions into production outcomes on Pure's platform.

This role blends science, consulting, and communication at Staff/Principal level. You will lead technical discovery, advise on end-to-end deployment, and create proof-of-value prototypes and enablement content for field teams, partners, and customers.