AI/ML Full Stack Developer

Nutanix is a technology company that focuses on cloud computing and enterprise solutions. They are seeking an AI/ML Full Stack Developer to design, develop, and deploy machine learning solutions for their education platform, working with advanced technologies and collaborating with distributed teams.ResponsibilitiesParticipate in ML sprint planning, including model experimentation roadmaps, RAG pipeline optimization, agentic workflow design, feature engineering discussions, and training pipeline estimationsDesign, 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 objectivesConduct 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 benchmarkingMentor 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 systemsMonitor 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 deliveryCollaborate effectively with distributed ML teams across time zones, coordinating on shared vector stores, agent orchestration frameworks, model serving infrastructure, and cross-functional AI initiativesStay 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 implementationsDocument 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 transferSkillsBachelor'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 conceptsAbility to work in Agile environments with willingness to learn, experiment, iterate on models, and take ownership of assigned ML projects and featuresStrong 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 engineerExposure to MLOps practices like experiment tracking, model versioning, or containerizationFamiliarity 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 AutoGenUnderstanding of NLP concepts, semantic search, and retrieval strategiesExperience with GPU infrastructure or optimizing model serving is beneficial but not requiredBenefitsSign-on bonusRestricted stock unitsDiscretionary awardsFull range of medical, financial, and/or other benefits401(k) eligibilityVarious paid time off benefits, such as vacation, sick time, and parental leaveCompany OverviewNutanix specializes in cloud software, offering organizations a single platform for running apps and data across clouds. It was founded in 2009, and is headquartered in San Jose, California, USA, with a workforce of 5001-10000 employees. Its website is http://www.nutanix.com.Company H1B SponsorshipNutanix has a track record of offering H1B sponsorships, with 260 in 2025, 152 in 2024, 300 in 2023, 282 in 2022, 137 in 2021, 143 in 2020. Please note that this does not guarantee sponsorship for this specific role.



Apply Now