Software Engineer, AI Infrastructure

BE PART OF BUILDING THE FUTURE.

What do NASA and emerging space companies have in common with COVID vaccine R&D teams or with Roblox and the Metaverse? 

The answer is data, -- all fast moving, fast growing industries rely on data for a competitive edge in their industries. And the most advanced companies are realizing the full data advantage by partnering with Pure Storage. Pure’s vision is to redefine the storage experience and empower innovators by simplifying how people consume and interact with data. With 11,000+ customers including 58% of the Fortune 500, we’ve only scratched the surface of our ambitions. 

Pure is blazing trails and setting records:

• For ten straight years, Gartner has named Pure a leader in the Magic Quadrant 
• Our customer-first culture and unwavering commitment to innovation have earned us a certified Net Promoter Score in the top 1% of B2B companies globally
• Industry analysts and press applaud Pure’s leadership across these dimensions
• And, our 5,000+ employees are emboldened to make Pure a faster, stronger, smarter company as we go
  
  

If you, like us, say “bring it on” to exciting challenges that change the world, we have endless opportunities where you can make your mark.

At Pure Storage, we are on a mission to revolutionize AI infrastructure. Our AI Infrastructure team is building an innovative platform to simplify the lives of data scientists, model engineers, and machine learning practitioners—abstracting away infrastructure challenges so they can focus on delivering AI-driven solutions.

This is not an extension of something that already exists, we are rethinking the data stack for Data Scientists and Model Engineers from ground up. What you will build sits higher up in the stack than the classic storage product and we are agnostic to on-prem or cloud deployments.

This is a rare opportunity to join a small but high-impact team that is moving fast to build a V1.0 product. We have a fully functional prototype and a working demo, and our focus for 2025 is to get our first customer engagements working on production-level design and getting early customer feedback. If you thrive in a high-velocity environment with minimal bureaucracy, this is the place for you. We seek people who live in code and knows how to implement but also can do high-quality design that stands the test of time.

 

SHOULD YOU ACCEPT THIS CHALLENGE...

• Architect and develop highly distributed, performant, and scalable backend systems.
• Design robust data models and implement efficient database schemas with a strong focus on versioning, migrations, and lifecycle management.
• Work on large-scale AI workflows, ensuring seamless integration with existing AI/ML tooling.
• Collaborate closely with user experience engineers to define the product vision and shape how different personas interact with the platform.
• Drive technical excellence by mentoring junior engineers, establishing best practices, and contributing to architectural decisions.
• Optimize system performance and reliability, ensuring that the platform scales to handle enterprise AI workloads.

WHAT YOU’LL NEED TO BRING TO THIS ROLE...

• 8+ years of backend engineering experience with a focus on high-scale, distributed systems.
• Deep expertise in Python, Go, C/C++ or Rust for backend development.
• Strong understanding of database design, indexing, schema migration, and versioning. We desire experience in PostgreSQL, MySQL, NoSQL or similar database technologies.
• Experience working with message queues, event-driven architectures, and streaming data processing.
• Hands-on experience with containerization (Docker, Kubernetes) and cloud-native development (AWS, GCP, or Azure).
• Ability to work end-to-end, from system design to implementation, testing, and deployment.
• Strong problem-solving skills with a bias toward action and ownership.
• We are primarily an in-office environment and therefore, you will be expected to work from the Santa Clara, California office in compliance with Pure’s policies, unless you are on PTO, or work travel, or other approved leave.

The annual base salary range is: $207,000– $312,00. Salary ranges are determined based on role, level and location. For positions open to candidates in mult