Jobs Companies Snowflake Staff Software Engineer, Cortex AI Search Infrastructure

About this Staff Software Engineer, Cortex AI Search Infrastructure role at Snowflake

Snowflake · Hybrid · US-CA-Menlo Park

At Snowflake, we are powering the era of the agentic enterprise. To usher in this new era, we seek AI-native thinkers across every function who are energized by the opportunity to reinvent how they work. You don’t just use tools; you possess an innate curiosity, treating AI as a high-trust collaborator that is core to how you solve problems and accelerate your impact. We look for low-ego individuals who thrive in dynamic and fast-moving environments and move with an experimental mindset — who rapidly test emerging capabilities to discover simpler, more powerful ways to deliver results. At Snowflake, your role isn't just to execute a function, but to help redefine the future of how work gets done.

The Cortex team is building the future of AI for enterprise data. This role focuses on the Search infrastructure that powers our flagship products like CoWork, Cortex Code & Cortex Agents fast, reliable, scalable and secure at the enterprise level.

You will be building high-performance retrieval engines (leveraging vector search, hybrid search, and semantic indexing) that power Snowflake Cortex. This involves optimizing how billions of rows of data are indexed and retrieved in milliseconds.

What you will do in this role:

  • Architect Agentic Runtimes: Build and scale the orchestration engines that execute complex agentic workflows, ensuring low-latency tool execution and robust state management.

  • Scale Context Engineering Infra: Design high-performance systems for RAG (Retrieval-Augmented Generation), including vector database integration, scalable and efficient search indexing, query processing, and result ranking, semantic caching, and automated metadata extraction.

  • Build the "Evals Engine": Develop the automated infrastructure required to run massive-scale golden set simulations, error analysis pipelines, and "hillclimbing" experiments.

  • Productionize AI Workflows: Collaborate with the modeling team to take raw LLM capabilities and turn them into hardened, multi-tenant microservices with strict guardrails and observability.

  • Optimize Performance & Cost: Direct the infra strategy for model routing, prompt caching, and token optimization to ensure Snowflake’s AI features are the most efficient in the industry.

Requirements:

  • Education: Bachelor’s degree in Computer Science or a related technical field.

  • Experience: 9 + years of experience building distributed systems, high-throughput APIs, or backend infrastructure for AI/ML products.

  • Technical Stack: Deep proficiency in Go or Java (for systems) and Python (for AI orchestration).

  • Systems Thinking: Strong understanding of database internals, distributed state management, and cloud-native architecture (Kubernetes, FoundationDB, etc.).

  • Domain Expertise: Familiarity with the "plumbing" of AI: vector indices, agent platforms, and building scalable data pipelines.

(Bonus) Experience with:

  • Query optimization and SQL engine internals.

  • Designing multi-tenant systems that handle sensitive enterprise data at scale.

  • Developing search infrastructure for large-scale applications.

  • Direct experience with any of the subsystems outlined above.

Every Snowflake employee is expected to follow the company’s confidentiality and security standards for handling sensitive data. Snowflake employees must abide by the company’s data security plan as an essential part of their duties. It is every employee’s duty to keep customer information secure and confidential.

Snowflake is growing fast, and we’re scaling our team to help enable and accelerate our growth. We are looking for people who share our values, challenge ordinary thinking, and push the pace of innovation while building a future for themselves and Snowflake.

How do you want to make your impact?

For jobs located in the United States, please visit the job posting on the Snowflake Careers Site for salary and benefits information: careers.snowflake.com

Ready to apply to Snowflake?
Apply to Snowflake

How this Staff Engineer salary compares

This role pays $287,625/yrabove the typical range for Staff Engineer roles.

$255,400 median $287,600 $287,625

Typical range $287,600–$287,625/yr, from 14 comparable Staff Engineer listings on JobsRadar (pay annualized to USD). See Staff Engineer salary insights →

Similar jobs

Snowflake
Staff/Principal AI Software Engineer - Snowflake CoWork
Snowflake
⚡ Apply early US-CA-Menlo Park Onsite $264,000–$379,500
● New 👁 Seen ✓ Applied 2d ago
Snowflake
Staff Software Engineer - Query Processing (Snowtrail)
Snowflake
⚡ Apply early US-CA-Menlo Park Onsite $236,000–$339,200
● New 👁 Seen ✓ Applied 5d ago
Snowflake
Staff Software Engineer - Snowhouse
Snowflake
⚡ Apply early US-CA-Menlo Park Onsite $236,000–$339,200
● New 👁 Seen ✓ Applied 1w ago
Redwood Materials
Staff Mechanical Design Engineer, EPC 
Redwood Materials
⚡ Apply early Carson City, NV Onsite
● New 👁 Seen ✓ Applied 5h ago
Redwood Materials
Staff Mechanical Design Engineer, EPC 
Redwood Materials
⚡ Apply early San Francisco, California, Uni... Onsite $187,500–$247,500
● New 👁 Seen ✓ Applied 5h ago
Redwood Materials
Staff Electrical Design Engineer, EPC
Redwood Materials
⚡ Apply early Carson City, NV Onsite
● New 👁 Seen ✓ Applied 5h ago
Anduril Industries
Staff PCB Design Engineer
Anduril Industries
⚡ Apply early Costa Mesa, California, United... Onsite $191,000–$253,000
● New 👁 Seen ✓ Applied 6h ago
Anduril Industries
Staff Product Quality Engineer, Low Observables
Anduril Industries
⚡ Apply early Ashville, Ohio, United States Onsite $143,000–$191,000
● New 👁 Seen ✓ Applied 6h ago
Anduril Industries
Staff Mechanical Engineer, Omen
Anduril Industries
⚡ Apply early Costa Mesa, California, United... Onsite $191,000–$253,000
● New 👁 Seen ✓ Applied 6h ago

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at Snowflake

See all jobs at Snowflake →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free