Staff Software Engineer - Artificial Intelligence/Machine Learning

About the Role :

We are seeking a highly experienced Staff Software Engineer (AI/ML) to join our advanced engineering team. In this role, you will drive the design, development, and deployment of scalable AI/ML solutions for enterprise-grade applications. As a senior technical leader, you will architect end-to-end machine learning systems, mentor engineering teams, and collaborate with cross-functional stakeholders to translate complex business challenges into production-ready AI-driven solutions.

Key Responsibilities :

Architecture & Design :

- Define the technical vision and architecture for AI/ML solutions at scale.

- Design and implement robust, distributed ML systems leveraging cloud-native technologies (AWS/Azure/GCP).

- Lead model lifecycle management, including training, deployment, monitoring, and retraining.

AI/ML Model Development :

- Research, design, and implement advanced algorithms in areas such as NLP, Computer Vision, Generative AI, Predictive Analytics, and Recommendation Systems.

- Optimize models for performance, scalability, and real-time inference.

- Leverage frameworks like TensorFlow, PyTorch, Keras, Hugging Face, and Scikit-learn.

Data Engineering & Pipeline Management :

- Design and maintain ETL pipelines, feature stores, and data preprocessing workflows.

- Work with structured/unstructured data, large-scale datasets, and real-time streaming data.

- Collaborate with data engineering teams to ensure high-quality, reliable datasets for training and inference.

MLOps & Deployment :

- Implement MLOps best practices using CI/CD pipelines, containerization (Docker), and orchestration tools (Kubernetes).

- Ensure reproducibility, model versioning, and scalable deployment of ML services.

- Monitor model performance (drift, bias, accuracy) and ensure compliance with responsible AI standards.

Leadership & Mentoring :

- Provide technical leadership and mentorship to junior engineers and data scientists.

- Conduct code reviews, design reviews, and enforce engineering best practices.

- Collaborate with product managers and business stakeholders to define AI/ML roadmaps and KPIs.

Required Skills & Expertise

Core Technical Expertise :

- 1015 years of software engineering experience with at least 68 years in AI/ML system development.

- Strong proficiency in Python, Java, or C++ for production-grade development.

- Hands-on experience with ML frameworks (PyTorch, TensorFlow, Scikit-learn, Hugging Face).

- Strong mathematical foundation in Linear Algebra, Statistics, Probability, and Optimization.

MLOps & Cloud :

- Deep experience with AWS SageMaker, Azure ML, or Google Vertex AI.

- Expertise in CI/CD pipelines, Docker, Kubernetes, MLflow, Kubeflow.

- Familiarity with data pipelines and distributed data processing (Spark, Kafka, Flink).

Domain Knowledge :

- Experience in at least one domain such as Healthcare, FinTech, Retail, or Enterprise SaaS AI applications.

- Strong focus on building responsible, explainable, and ethical AI systems.

Soft Skills :

- Excellent leadership, mentoring, and stakeholder management abilities.

- Strong communication skills with the ability to explain technical concepts to non-technical stakeholders.

Qualifications :

Education : Bachelors or Masters degree in Computer Science, AI/ML, Data Science, or related field (Ph.D. preferred).

Certifications : AI/ML certifications from AWS, Azure, GCP, or DeepLearning.ai are a plus.

- Proven experience leading end-to-end AI/ML product development and managing cross-functional teams.