Technical Lead – ML Engineer

Job Purpose

Design, architect, and deliver scalable AI/ML solutions that solve complex business problems while leading engineering initiatives across multiple client engagements. Drive the adoption of modern Machine Learning, MLOps, and Generative AI technologies to build intelligent, production-ready applications with high performance, reliability, and scalability.

Who you are

• 8+ years of software development experience with 4+ years of hands-on experience building and deploying production-grade Machine Learning/AI systems.
• 2+ years of experience in a Technical Lead role, driving architecture decisions, mentoring engineers, and leading technical initiatives.
• Strong expertise in Python, Object-Oriented Programming, SQL, and REST/gRPC API development using FastAPI or Flask.
• Hands-on experience with Machine Learning frameworks such as PyTorch, TensorFlow, scikit-learn, or XGBoost.
• Experience building, training, evaluating, and deploying Machine Learning models into production.
• Strong knowledge of MLOps tools including MLflow, Kubeflow, Airflow, or dbt for experiment tracking, workflow orchestration, and model lifecycle management.
• Hands-on experience developing Generative AI applications using LangChain, LlamaIndex, OpenAI, Anthropic, Hugging Face, and Retrieval-Augmented Generation (RAG) architectures.
• Experience working with Vector Databases such as Pinecone, Qdrant, Chroma, or Milvus.
• Experience deploying AI/ML workloads on AWS SageMaker, Azure ML, or GCP Vertex AI using Docker and Kubernetes.
• Experience working in software services, consulting, or product engineering environments delivering solutions for multiple clients.
• Strong analytical, problem-solving, communication, and stakeholder management skills.

What will excite us

• Ability to architect scalable, cloud-native AI/ML solutions for enterprise applications.
• Strong understanding of Machine Learning lifecycle, MLOps, and production deployment best practices.
• Experience designing and implementing end-to-end Generative AI solutions including RAG pipelines, AI Agents, and LLM integrations.
• Ability to lead technical discussions, mentor engineering teams, and establish development best practices.
• Experience collaborating with Product Managers, Architects, Data Engineers, and DevOps teams to deliver high-quality AI products.
• Strong knowledge of containerization, orchestration, CI/CD pipelines, and cloud-native deployments.
• Ability to optimize model performance, scalability, inference latency, and deployment efficiency.
• Experience participating in solution architecture, technical estimation, sprint planning, and client-facing discussions.
• Strong ownership mindset with a passion for innovation and continuous learning in emerging AI technologies.

What will excite you

• Opportunity to architect and build enterprise-scale AI and Machine Learning platforms.
• Work on cutting-edge Generative AI, Large Language Models (LLMs), AI Agents, and Retrieval-Augmented Generation (RAG) solutions.
• Collaborate with cross-functional teams including Product, Engineering, Data Science, and Business stakeholders.
• Own the complete AI solution lifecycle—from architecture and model development to deployment and optimization.
• Mentor talented engineers and influence technical direction across multiple projects.
• Exposure to modern cloud-native technologies, MLOps platforms, and enterprise AI ecosystems.
• Collaborative, innovation-driven environment with opportunities to solve complex real-world business challenges.

Location: Ahmedabad (Work From Office)