Staff Software Engineer, Cortex AI Search Infrastructure

SNOWFLAKES ABA, LLC
Menlo Park, United States
6 days ago
Apply on startup.jobs
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Experience required
9 years minimum
Working hours
Regular working hours
Job source

Tech stack

Java (Programming Language) Application Programming Interfaces (APIs) Artificial Intelligence Cloud Engineering Databases Distributed Systems Python (Programming Language) Query Optimization SQL Databases Enterprise Data Management Enterprise Software Applications Retrieval-Augmented Generation
+9 more
Large Language Models Snowflake Multi-Agent Systems Backend Kubernetes Information Technology Data Pipelines Automation Anywhere Microservices

Job description

  • 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., * 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.

Benefits & conditions

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.

About the company

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.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on startup.jobs
Prepare application

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

1:52 min

Structuring and scaling the backend engineering team

Stefan Lingler Stefan Lingler +1 Ā· Coffee With Developers

3:04 min

Database evolution and the funding behind vector databases

Erik Bamberg Ā· LIVE

1:33 min

Integrating internal APIs and maintaining data sovereignty

Mahran Meißner Mahran Meißner · World Congress 2026 Europe

3:03 min

Career evolution in data engineering and AI platforms

Maria Apazoglou Ā· Coffee With Developers

1:12 min

Choosing TypeScript for complex backend applications

Maximilian Otto Maximilian Otto Ā· World Congress 2024

2:14 min

Solving complex platform architecture challenges at an enterprise scale

Maria Apazoglou Ā· Coffee With Developers

Videos

See all

Related articles

See all