Staff Machine Learning Engineer

Role details

Zscaler (NASDAQ: ZS) accelerates digital transformation so customers can be more agile, efficient, resilient, and secure. The Zscaler Zero Trust Exchange platform protects thousands of customers from cyberattacks and data loss by securely connecting users, devices, and applications in any location. Distributed across 200+ public data centers globally and thousands of private sites at the edge, the SASE-based Zero Trust Exchange is the world’s largest in-line cloud security platform. We believe the future of work is Human + AI and are building an AI-native enterprise where human potential is amplified by machine intelligence to solve the world’s hardest security challenges. Driven by deep customer obsession, we are committed to the mission, outcome, and to each other. We bring these commitments to life through three core behaviors: ownership and collaboration, trust through outcomes and impact, and a challenge culture with ongoing feedback. Ready to make an impact at the company pioneering security transformation in the AI era? Join us at Zscaler. Role We are looking for a Machine Learning Engineer to join our Artificial Intelligence Guard team in a remote capacity within the United States (with a hybrid preference for Santa Clara, CA), reporting directly to the Director of AI and Machine Learning Engineering team. Operating at the core of the team that built the world’s largest cloud security platform processing over 400 billion transactions daily, you will design, build, and deploy end-to-end machine learning pipelines while integrating advanced AI capabilities into production-ready SaaS offerings to directly impact our global strategic roadmap. What You’ll Do (Role Expectations) Build and deploy end-to-end ML pipelines spanning data curation, training/fine-tuning, evaluation, and high-scale serving Develop deep learning models for security use cases, including transformer-based classifiers, embedding models, and sequence modeling across high-volume traffic data Adapt, fine-tune, and productionize open-weight LLMs and small language models using techniques such as LoRA/QLoRA, instruction tuning, and distillation Optimize models for production through quantization, batching, and high-throughput serving with frameworks like Hugging Face, PEFT, vLLM, TensorRT-LLM, and ONNX Runtime to balance latency, cost, and quality Architect and operationalize robust ML services across cloud platforms (AWS, GCP) leveraging cloud-native microservice architectures Who You Are (Success Profile) You enjoy being on top of the latest advancements and research in the deep learning space and you learn fast. You thrive on uncovering complex patterns within large, sparse datasets, leveraging a rigorous, highly numerate background to solve multifaceted business problems. You operate with an uncompromising sense of ownership and an execution-focused mindset, seamlessly bridging the gap between theoretical modeling and production deployment. You are a proactive, independent problem solver energized by engineering elegant, resilient solutions for massive-scale technical challenges. You possess a growth mindset and a continuous drive to learn, actively adapting to and implementing cutting-edge machine learning advancements. You are a collaborative partner who excels at working cross-functionally alongside engineering teams to champion and execute organizational AI strategies. What We’re Looking For (Minimum Qualifications) Demonstrated experience utilizing modern AI/ML frameworks and foundational model workflows to design, train, and deploy intelligent systems at scale Bachelor's or advanced degree in Computer Science, Machine Learning, Mathematics, Physics, Statistics, Engineering, or a related field, with 2+ years of applied ML experience Solid grounding in machine learning fundamentals, including loss functions, optimization, regularization, evaluation metrics, and handling imbalanced or noisy data Strong Python programming expertise and hands-on experience with modern deep learning frameworks such as PyTorch, TensorFlow, or JAX Proven experience training/fine-tuning large language models (LLMs) and deploying deep learning models (e.g., transformers, embedding models, generative AI architectures etc) in production environments Ability to work with large-scale datasets, write clean, testable code, and operate with high autonomy on ambiguous technical challenges What Will Make You Stand Out (Preferred Qualifications) Proven experience developing generative AI architectures or building scalable inference optimization pipelines Peer-reviewed publications or prominent open-source contributions demonstrating depth in modern deep learning techniques (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, IEEE) #LI-Remote #LI-YC2 Zscaler’s salary ranges are benchmarked and are determined by role and level. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations and could be higher or lower based on a multitude of factors, including job-related skills, experience, and relevant education or training. The base salary range listed for this full-time position excludes commission/ bonus/ equity (if applicable) + benefits. Base Pay Range $152,000 - $190,000 USD At Zscaler, we are committed to building a team that reflects the communities we serve and the customers we work with. We foster an inclusive environment that values all backgrounds and perspectives, emphasizing collaboration and belonging. Join us in our mission to make doing business seamless and secure. Our Benefits program is one of the most important ways we support our employees. Zscaler proudly offers comprehensive and inclusive benefits to meet the diverse needs of our employees and their families throughout their life stages, including: Various health plans Time off plans for vacation and sick time Parental leave options Retirement options Education reimbursement In-office perks, and mor