EN
Hiring companyExtreme Networks
Verified today
Staff Machine Learning Engineer, Generative AI, Machine Learning, Graph ML, Big Data
Bangalore, India Full-time Hybrid
About the role
AI Core group at Extreme Networks builds platforms for Generative AI, including AI agents, RAG, knowledge bases, and LLM fine‑tuning. The Staff Machine Learning Engineer designs and scales large distributed ML systems, leads architecture, and mentors engineers while integrating Python, FastAPI, Docker, and Kubernetes on cloud platforms. Bangalore, India, hybrid work model with remote flexibility.
What you’ll do
- Design and implement generative AI platforms
- Develop RESTful APIs using FastAPI
- Build and optimize large‑scale ML pipelines
- Lead architecture decisions for cloud deployments
- Mentor and guide engineering team
What you’ll bring
- 8+ years software development lifecycle
- 6+ years Python
- 4+ years leading design of large distributed systems
- Experience with FastAPI, Docker, Kubernetes
- ML expertise with PyTorch, SageMaker, Triton
- Leadership or tech‑lead experience
Nice to have
BS/MS or PhD in Computer Science or related field
Skills
PythonMachine LearningDistributed SystemsDockerKubernetesFastAPICloud PlatformsBig Data
Benefits
- Inclusive workplace culture
- Global impact with 50,000+ customers