Staff Engineer - AI Inference

Nutanix — Canada · Posted ~5 hours ago

Lead Full-time

Skills

Generative AI LLMs LLM Inference AI Gateways Agentic AI Machine Learning

🔓 Log in to save this job, tailor your resume & track your apply process — 7 days free, no card needed.

Log in to add to target list

Summary ✨ AI‑Generated

A Staff Engineer role within a globally distributed AI infrastructure team, responsible for foundational technologies that enable organizations to deploy and operate generative AI, large language models, and agentic applications. The position involves high-performance AI infrastructure and inference capabilities spanning on-premises, edge, and public-cloud environments.

Highlights

Staff-level opportunity on foundational AI infrastructure, focused on generative AI, LLM inference, AI gateways, and agentic AI. The role offers exposure to cutting-edge AI infrastructure across on-premises, edge, and public-cloud environments.

Description

The Opportunity At Nutanix, we're simplifying the future of AI with the Nutanix Cloud Platform for AI, enabling organizations to easily build, fine-tune, and run Generative AI, Large Language Models (LLMs), and next-generation Agentic AI applications without the complexity of managing AI infrastructure themselves. Our high-performance, full-stack machine learning cloud platform delivers AI-ready capabilities out of the box ("GPT-in-a-Box") through a software-defined, full-stack infrastructure solution that simplifies AI deployment across on-premises data centers, edge locations, and public clouds. The Enterprise AI team is at the forefront of this innovation, driving strategic products such as LLM Inference, the AI Gateway, and the Agentic AI Platform, recently showcased at NVIDIA GTC and NEXT 2026. Join the team responsible for the foundational AI technologies powering the next wave of intelligent applications at Nutanix. About The Team We are a fast-paced, globally distributed team building the foundational layers of the enterprise AI stack. By joining our team, you'll help shape the next generation of enterprise AI platforms, working at the intersection of large-scale distributed systems and machine learning infrastructure. This is an opportunity to solve complex technical challenges, influence the direction of key AI technologies, and build systems that power AI workloads at enterprise scale. You will report to a seasoned Technical Lead/Manager who will provide mentorship and guidance as you navigate through your responsibilities. The work setup at Nutanix AI is a hybrid model, offering a blend of in-office collaboration and remote work flexibility. As a new hire, you will be expected to be in the office for 3 days a week, ensuring that you have the opportunity to engage with your team and foster strong working relationships. Your Role Architect, design, and develop horizontally scalable, containerized, fault-tolerant services on Kubernetes for enterprise AI and LLM workloads.Build and operate high-performance inference and platform services that deliver low-latency, high-throughput experiences for Generative AI and Agentic AI applications.Design and optimize critical system components across the stack, including distributed systems, storage, networking, and low-level infrastructure layers.Develop and enhance multi-tenant platform services supporting on-premises, hybrid, and cloud-based AI deployments.Design and implement scalable observability architectures using technologies such as Prometheus, Grafana, Datadog, OpenTelemetry, and related cloud-native monitoring frameworks.Debug complex production issues, perform root-cause analysis, and improve reliability, resiliency, and operational efficiency of platform services.Build and maintain CI/CD pipelines and deployment automation to accelerate delivery of production-grade services.Design and implement foundational LLM serving capabilities including request routing, rate limiting, token streaming, load balancing, quota management, and usage budgeting.Collaborate closely with globally distributed product management, AI, and software engineering teams to deliver high-quality products in a fast-paced environment.Contribute to all stages of the product lifecycle, including architecture, design, development, testing, experimentation, performance analysis, deployment, and operations.Leverage and contribute to relevant open-source cloud-native and AI ecosystem projects.Review code and design documents, provide feedback on product requirements, and champion engineering excellence across the team.Continuously evaluate emerging technologies and help shape the technical direction of Nutanix's Enterprise AI Platform. Required Qualifications What You Will Bring 8+ years of experience developing maintainable, modular, resilient, fail-safe, and long-lived software products within a product development organization.Strong computer science fundamentals including data structures, algorithms, operating systems, networking, and distributed systems.Hands-on experience with Docker, Kubernetes, and cloud-native architectures.Production experience developing backend systems using Go, Python, C++, or Rust.Experience building, owning, and maintaining CI/CD pipelines and release automation end-to-end.Strong understanding of datacenter architecture including compute, storage, networking, and virtualization.Experience designing and deploying software across on-premises, cloud, and hybrid environments.Demonstrated experience designing and tuning high-performance, performance-sensitive system software.Solid understanding of distributed computing, distributed data stores, and large-scale service architectures.Experience diagnosing and resolving production performance issues using observability and monitoring platforms such as Prometheus, Grafana, Datadog, Open Telemetry, or similar tools.Familiarity with LLM serving concepts including rate limiting, token streaming, request scheduling, load balancing, quota management, and usage budgeting.Familiarity with modern LLM concepts including reasoning workflows, tool calling, prompt templates, and agent.Experience building multi-tenant services running on virtualized or containerized infrastructure.Strong communication, collaboration, and problem-solving skills with the ability to work effectively across globally distributed teams.Master's degree in Computer Science or equivalent practical experience. Bonus Points If You Have Experience With Machine learning frameworks such as PyTorch or TensorFlow.GPU-based systems and acceleration technologies.Modern model-serving platforms such as vLLM, DeepSpeed, Hugging Face TGI, or Triton.Retrieval-Augmented Generation (RAG), vector databases, and AI orchestration frameworks.Open-source contributions or experience working in large distributed codebases.Production AI platforms, LLM APIs, agentic systems, or inference infrastructure.Building or scaling production LLM APIs, including streaming via SSE/WebSockets, prompt guardrails, rate limiting, and usage budgeting. Learn More About the Technology: https://www.nutanixbible.com/ [nutanixbible.com] #NAI Highlighted Benefits (Vancouver, Canada) Retirement: RRSP with dollar-for-dollar matching up to 7% of base salary Mental Health: Dedicated mental health coverage plus top-tier paramedical benefits Family: Fully paid maternity and parental leave and generous bereavement leave, including time for the loss of a pet Equity: RSUs and Employee Stock Purchase Plan at a 15% discount Work Arrangement Hybrid: This role operates in a hybrid capacity, blending the benefits of remote work with the advantages of in-person collaboration. In locations where our workplace policy applies (i.e. San Jose, Durham, Mexico City, Vancouver, Bangalore, Pune, Hoofddorp, Belgrade, Barcelona, Singapore, Sydney and Tokyo), employees are expected to work onsite a minimum of 3 days per week to foster collaboration, team alignment, and access to in-office resources. Workplace type may vary based on location and team requirements. Please speak with your recruiter for details. Additional team-specific guidance and norms will be provided by your manager. Pay Transparency - Role Location The pay range for this position at commencement of employment is expected to be between CAD $171,000 and CAD $257,000 per annual. However, base pay offered may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. The total compensation package for this position may also include other elements, including a sign-on bonus, restricted stock units, and discretionary awards in addition to a full range of medical, financial and/or other benefits (including 401(k) eligibility and various paid time off benefits, such as vacation, sick time, and parental leave), dependent on the position offered. Details of participation in these benefit plans will be provided if an employee receives an offer of employment. If hired, employee will be in an “at-will position” and the Company reserves the right to modify base salary (as well as any other discretionary payment or compensation program) at any time, including for reasons related to individual performance, Company or individual department/team performance, and market factors. Our application deadline is 40 days from the date of posting. In good faith, the posting may be removed prior to this date if the position is filled or extended in good faith.