Job Type: Full time

Role: AI/ML Engineer

Location: Bengaluru, India

As a AI/ML Engineer, you will be working in one of the world's largest and most complex data Lake and warehouse environments. You will design, implement, and support scalable data infrastructure solutions to integrate with multi heterogeneous data sources, aggregate and retrieve data in a fast and safe mode, curate data that can be used in reporting, analysis, machine learning models and ad-hoc data requests. You will be exposed to cutting edge cloud(AWS, Azure, GCP) big data technologies. This will be a great fit if you get excited by partnering with key stakeholders to dig deep into the business challenges to understand and identify insights that will enable us to meet associate needs at a globally scale. You need excellent analytical abilities as well as business acumen and an ability to tie together from across Amazon to help enable effective decision making. An ability to communicate insight to the business via development and implementation of ongoing and subject-specific reporting artifacts is a critical skill in this role.

Job Requirements (JR)

• Education: Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence, Machine Learning, or a closely related quantitative field.
• Experience: 3+ years of hands-on experience in Data Science and Machine Learning, with a demonstrable focus on:
• Generative AI: Practical experience with LLMs, prompt engineering, fine-tuning concepts, Retrieval-Augmented Generation (RAG) principles, embeddings, and vector similarity search techniques.
• Computer Vision: Hands-on experience applying CV models for image analysis (e.g., object detection, classification, OCR, graph/diagram understanding).
• Technical Proficiency:
• Strong programming skills in Python and proficiency with relevant AI/ML libraries/frameworks (e.g., PyTorch, TensorFlow, Hugging Face Transformers, LangChain, Langgraph, OpenCV, scikit-learn).
• Experience working with vector databases/search engines.
• Proficiency with SQL and NoSQL for data querying and analysis.
• Hands on Experience in developing Multi Agent based AI Applications with minimum two or more project implementations and successful deployments
• Cloud & Deployment:
• Hands-on experience building, deploying, and managing AI/ML models in production on cloud platforms, with a preference for Azure (e.g., Azure Machine Learning, Azure OpenAI Service, Azure Cognitive Services, Azure Kubernetes Service - AKS).
• Multi-Modal Experience: Demonstrated ability to work with and integrate data from different modalities (text, images/visual data).
• Problem Solving: Strong analytical, critical thinking, and problem-solving skills, capable of contributing to the resolution of complex challenges in AI development and deployment.
• Communication: Good collaboration and communication skills (verbal and written), with the ability to explain technical concepts clearly within the team.

Key Roles and Responsibilities:

• Design, develop, and implement Generative AI solutions for specific tasks within document summarization, knowledge retrieval, content generation, and structured information extraction from PDFs, excel and other databases utilizing LLMs and relevant frameworks.
• Develop and apply Computer Vision (CV) models to analyze and interpret images, focusing on graphs, charts, and diagrams found in documents or other sources.
• Build and optimize components of Retrieval-Augmented Generation (RAG) systems, including developing embedding pipelines and integrating with vector databases (e.g., Azure Cognitive Search, Pinecone, Milvus) for semantic search and context retrieval.
• Work with multi-modal data, contributing to the integration of insights from both text and image/visual analysis within specific project modules.
• Evaluate, select, and fine-tune pre-trained foundation models (LLMs, Vision models) for specific tasks, contributing to performance improvements on domain-specific data.
• Design, implement, and manage specific parts of MLOps pipelines on Azure (using Azure Machine Learning, Azure DevOps, AKS) for model training, deployment, and monitoring.
• Monitor the performance of deployed models (accuracy, latency, cost, robustness) in production environments, contributing to the identification and implementation of improvements.
• Collaborate closely with Data Engineers, Software Engineers, and Product Managers to integrate AI/ML capabilities into applications, taking ownership of specific integration points.
• Stay updated on the latest advancements in Generative AI, Computer Vision, and MLOps, proactively learning and contributing to the exploration of new methodologies within the team.

Preferred Qualifications:

• Adequate experience with Azure OpenAI Service and Azure Cognitive Search for vector capabilities.
• Exposure to multi-modal foundation models.
• Familiarity with Databricks for AI/ML workloads.
• Experience with other cloud platforms (AWS, GCP).
• Contributions to relevant AI/ML/CV projects (including personal projects or academic work).
• Experience working in Agile development environments.

Skills

• Python Programming & ML/DL Frameworks (PyTorch/TensorFlow)
• Generative AI Expertise (LLMs, RAG, Embeddings, Vector DBs)
• Cloud Platform Experience (Specifically Azure AI/ML Services)
• Problem-Solving & Analytical Thinking