AI / ML Full Stack Developer

The Opportunity

Are you an AI / ML engineer passionate about building intelligent systems from the ground up? Join the SaaS Engineering team at Nutanix to design, develop, and deploy production-scale machine learning solutions for our dynamic education platform serving employees, customers, and partners. You'll architect and optimize neural recommendation systems, build advanced NLP pipelines for semantic search, develop conversational AI agents using LLMs, and implement RAG frameworks. Your expertise in model training, fine-tuning, feature engineering, and MLOps will drive innovation as you work with cutting-edge frameworks and deploy models that power real-time intelligent experiences at scale.

About the Team

At Nutanix, you'll join the SaaS Engineering team's AI / ML division, driving innovation in our learning management system, Nutanix University. Our team is geographically distributed across India, San Jose, CA, and Durham, NC, bringing together machine learning engineers, data scientists, and MLOps specialists who collaborate on building production ML systems. We operate in a fast-paced environment where we ship models iteratively using Agile sprints, enabling rapid experimentation, model retraining, and continuous deployment of AI features. You'll work directly with distributed training infrastructure, experiment tracking platforms, and vector databases while building end-to-end ML pipelines from data ingestion to model serving. The team maintains a strong culture of knowledge sharing around emerging research, model architectures, and optimization techniques. You will report to the Director of Engineering, who champions ML innovation and provides technical mentorship to help you grow as an ML engineer. Our hybrid work model requires three days in office, facilitating collaborative model debugging sessions, architecture reviews, and hands-on pair programming while maintaining flexibility for focused deep work on complex ML problems.

Your Role

Participate in ML sprint planning, including model experimentation roadmaps, RAG pipeline optimization, agentic workflow design, feature engineering discussions, and training pipeline estimations.

Design, develop, and deploy machine learning models, RAG systems with vector databases and embedding models, and autonomous AI agents with tool calling capabilities, ensuring scalability, latency optimization, and alignment with business objectives.

Conduct peer reviews of ML code, RAG retrieval strategies, agent framework implementations, model architectures, and experiment results, contributing to team-wide evaluation of semantic search quality, agent performance, and model benchmarking.

Mentor junior ML engineers on best practices in model development, RAG architecture patterns, building multi-agent systems, prompt engineering, hyperparameter tuning, data preprocessing, and production ML systems.

Monitor deployed models, RAG pipelines, and agentic workflows in production, manage embedding model updates, optimize retrieval performance, debug agent behavior, handle model drift detection, and maintain MLOps infrastructure for continuous delivery.

Collaborate effectively with distributed ML teams across time zones, coordinating on shared vector stores, agent orchestration frameworks, model serving infrastructure, and cross-functional AI initiatives.

Stay current with latest ML research, RAG optimization techniques, agentic AI frameworks like LangGraph and AutoGen, emerging model architectures, fine-tuning techniques, and GenAI advancements, bringing innovative approaches to team discussions and technical implementations.

Document model architectures, RAG system designs, agent workflows, retrieval strategies, training procedures, feature specifications, and deployment processes through model cards, experiment tracking logs, architecture diagrams, and comprehensive ML documentation to enable reproducibility and knowledge transfer.

What You Will Bring

Bachelor's degree in Computer Science, Machine Learning, Data Science, or related technical field with solid foundation in Python and familiarity with ML frameworks (PyTorch, TensorFlow, scikit-learn, Hugging Face Transformers).

Experience or strong interest in building ML systems with understanding of model training, fine-tuning, and deployment concepts. Exposure to MLOps practices like experiment tracking, model versioning, or containerization is a plus.

Familiarity with modern AI / ML approaches including working with LLMs, prompt engineering, RAG systems using vector databases and embeddings, or agentic AI frameworks like LangGraph, LangChain, or AutoGen.

Understanding of NLP concepts, semantic search, and retrieval strategies. Experience with GPU infrastructure or optimizing model serving is beneficial but not required.

Ability to work in Agile environments with willingness to learn, experiment, iterate on models, and take ownership of assigned ML projects and features.

Strong communication skills, eagerness to learn from senior team members, and enthusiasm for contributing ideas, staying current with ML research, and growing as an AI / ML engineer.

Work Arrangement

Hybrid : This role operates in a hybrid capacity, blending the benefits of remote work with the advantages of in-person collaboration. For most roles, that will mean coming into an office a minimum of 3 days per week, however certain roles and / or teams may require more frequent in-office presence. Additional team-specific guidance and norms will be provided by your manager.

The pay range for this position at commencement of employment is expected to be between USD $ 109,600 and USD $ 218,400 per year. 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. We're an Equal Opportunity Employer Nutanix is an Equal Employment Opportunity and (in the U.S.) an Affirmative Action employer. Qualified applicants are considered for employment opportunities without regard to race, color, religion, sex, sexual orientation, gender identity or expression, national origin, age, marital status, protected veteran status, disability status or any other category protected by applicable law. We hire and promote individuals solely on the basis of qualifications for the job to be filled. We strive to foster an inclusive working environment that enables all our Nutants to be themselves and to do great work in a safe and welcoming environment, free of unlawful discrimination, intimidation or harassment. As part of this commitment, we will ensure that persons with disabilities are provided reasonable accommodations. If you need a reasonable accommodation, please let us know by contacting CandidateAccommodationRequests@nutanix.com.