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

AI Staff Machine Learning Engineer (AI Engineering): Pioneering platforms and solutions for Generative AI, including AI Agents, RAG, Knowledge Bases, Data Mining, Anomaly Detection, and LLM fine-tuning with an accent on building intelligent, high-performance multi-agent systems. Focus on designing and launching strategic machine learning (ML) solutions and driving business-wide innovation.

Location: San Jose, California, United States . Hybrid/Remote

Salary: $170,000 - $240,000 a year

Company

Extreme Networks creates effortless networking experiences that enable all of us to advance.

What you will do

• Lead technical discussions and strategy, and participate hands‑on in design reviews, code reviews, and implementation.
• Craft high-performance, high-scale microservices architectures, including synchronous and asynchronous web services.
• Develop real-time online inferencing for highly complex models using Triton, TensorRT and mixed precision computing.
• Mentor and develop other engineers on the team, establish technical direction and foster team culture.
• Uphold the highest standards of technical rigor in engineering and operational excellence, build highly resilient and scalable systems, and champion operational and process improvements.

Requirements

• Degree in mathematics/computer science or related discipline.
• 5 to 10 years of experience in the complete software development lifecycle including design, coding, code reviews, testing, build processes, deployments and operations.
• 5 to 10 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.
• 3+ 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.

Nice to have

• 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.

Culture & Benefits

• Team of highly talented engineers.