Staff Machine Learning Engineer

Extreme Networks India Private Limited; Bengaluru; Remote; INR 4,200,000 - 8,500,000; Full time

Job summary

Extreme Networks India Private Limited is seeking an AI-Staff/Principal Software Systems Engineer specializing in Gen AI, ML, Graph ML and Big Data. The role requires 8+ years of end-to-end software development and a strong background in Python, RESTful APIs, and cloud platforms.

You will lead architecture and design of large distributed systems, mentor engineers, and drive initiatives across AI core platforms including LLM fine-tuning, data mining, and knowledge bases.

Qualifications

• BE/MS or PhD in Computer Science or related field
• 8+ years of end-to-end software development lifecycle
• 6+ years of Python with advanced features
• Experience designing RESTful APIs with FastAPI
• Proficient in Docker, Kubernetes and CI/CD practices
• 4+ years leading the design and architecture of large distributed systems on cloud platforms (AWS/Azure/GCP)
• Experience as a mentor or tech lead
• Experience with ML technologies (PyTorch, Sagemaker, Triton, TensorRT)
• Experience with NoSQL/document databases

Responsibilities

• Design and architect large distributed systems on cloud platforms
• Lead and mentor engineering teams
• Develop RESTful APIs using FastAPI and modern stacks
• Optimize data workflows and system performance
• Collaborate with ML research to productionize models

Skills

Python programming

RESTful APIs

Docker

Kubernetes

CI/CD

Distributed systems design

Mentorship/tech leadership

ML frameworks

NoSQL databases

Big data

Cloud platforms

Education

BE/MS or PhD in Computer Science

Tools

FastAPI

AWS/Azure/GCP

PyTorch

Sagemaker

TensorRT/Triton

Job description

Position

AI-Staff/Principal Software Systems Engineer-Gen AI, Machine Learning, Graph ML, Big Data

Experience

8 to 18 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!

Extreme Networks, Inc. ( EXTR ) creates effortless networking experiences that enable all of us to advance. We push the boundaries of technology leveraging the powers of machine learning, artificial intelligence, analytics, and automation. Over 50,000 customers globally trust our end-to-end, cloud-driven networking solutions and rely on our top-rated services and support to accelerate their digital transformation efforts and deliver progress like never before.

For more information, visit Extreme's website or follow us on Twitter, LinkedIn, and Facebook.

We encourage people from underrepresented groups to apply. Come Advance with us!

In keeping with our values, no employee or applicant will face discrimination/harassment based on: race, color, ancestry, national origin, religion, age, gender, marital domestic partner status, sexual orientation, gender identity, disability status, or veteran status. Above and beyond discrimination/harassment based on “protected categories,” Extreme Networks also strives to prevent other, subtler forms of inappropriate behavior (e.g., stereotyping) from ever gaining a foothold in our organization. Whether blatant or hidden, barriers to success have no place at Extreme Networks.

Experience Level

Senior Level