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 :Education : Bachelors or Masters degree in Computer Science, AI / ML, Data Science, or related field (Ph.D. : 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.(ref : hirist.tech)