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

Extremenetworks; San Jose (CA); Hybrid; USD 170,000 - 240,000; Full time

Job summary

Extreme Networks is seeking an AI Staff Machine Learning Engineer to lead innovative machine learning solutions in a hybrid/remote role. The ideal candidate will have a strong background in Python, experience with distributed systems on cloud platforms, and a passion for mentoring other engineers.

Salary for this position ranges from USD 170,000 to 240,000, depending on qualifications and experience. Join a team dedicated to transforming the way we do business with advanced technologies.

Qualifications

• 5 to 10 years of experience in the complete software development lifecycle.
• 5 to 10 years of experience in Python with advanced knowledge.
• 3+ years of experience in leading large distributed systems design.

Responsibilities

• Drive innovative vision for products and platforms.
• Lead end-to-end software development lifecycle.
• Mentor and establish technical direction for the team.

Skills

Python

RESTful API Design

Docker

Kubernetes

Machine Learning Technologies

Big Data Handling

CI/CD Practices

Education

Degree in Mathematics, Computer Science or related discipline

MS or PhD in Computer Science or equivalent experience in ML

Tools

FastAPI

AWS

Azure

Google Cloud

PyTorch

Triton

TensorRT

NoSQL and Document Databases

Job description

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

Experience: 5 to 14 years

Location: Hybrid/Remote

Responsibilities

• Be a thought leader and forward thinker, help drive an innovative vision for products and platforms, design and launch strategic machine learning solutions and drive business‑wide innovation.
• Take the lead in the end‑to‑end software development lifecycle, encompassing design, testing, deployment, and operations, 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.
• Come work with a team of highly talented engineers and advance with us to achieve new heights every day!

Qualifications

• 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.
• MS or PhD in Computer Science or equivalent experience in ML.
• Experience working with ML technologies such as PyTorch, Sagemaker, Triton, TensorRT, etc.
• Experience with NoSQL and document databases.
• Proven ability to handle big data, optimize workflows, and improve system performance.

Benefits

Salary based on qualifications, experience and region up to USD 170,000 to 240,000 plus benefits.

Equal Opportunity Employer

We encourage people from underrepresented groups to apply. In keeping with our values, no employee or applicant will face discrimination or harassment based on race, color, ancestry, national origin, religion, age, gender, marital domestic partner status, sexual orientation, gender identity, disability status or veteran status. Extreme Networks strives to prevent other subtle forms of inappropriate behavior from gaining a foothold in our organization.