Staff Machine Learning Engineer -Gen AI, Machine Learning, Graph ML, Big Data - 10030

Position: Staff Machine Learning Engineer -Gen AI, Machine Learning, Graph ML, Big Data

Experience: 8 to 14 Years

Position type: Full-time

Location: Bangalore/Remote

Our AI Core group is pioneering platforms and solutions for Generative AI, including AI Agents, RAG, Knowledge Bases, Data Mining, Anomaly Detection, and LLM fine-tuning. These innovations power flagship Extreme products while enabling entirely new offerings. Together, we are driving a fundamental shift in how businesses manage networks by building intelligent, high-performance multi-agent systems that perceive, learn, and act in real time. At Extreme, innovation is not just encouraged, it is expected. Advance with us and help shape the future of network intelligence.

• Degree in mathematics/computer science or related discipline.
• 8+ years of experience in the complete software development lifecycle including design, coding, code reviews, testing, build processes, deployments and operations.
• 6+ years of experience in Python with an in-depth knowledge of its advanced features and libraries.
• Expertise in designing RESTful APIs with hands-on experience with technologies such as FastAPI.
• Proficient in Docker, Kubernetes, and modern CI/CD practices.
• 4+ years of experience in leading the design and architecture of large distributed systems preferably on cloud platforms (e.g., AWS, Azure, Google Cloud).
• Experience as a mentor, tech lead or leading an engineering team.

• BE/MS or PhD in Computer Science or equivalent experience in ML.
• Experience working with ML technologies (PyTorch, Sagemaker, Triton, TensorRT, etc.).
• Experience with NoSQL and document databases.
• Proven ability to handle big data, optimize workflows, and improve system performance.
• Come work with a team of highly talented engineers, and advance with us to achieve new heights every day!