Lead AI Engineer - Bengaluru, INDIA at Vytwo

Prosper, Texas, United States -

Full Time

Start Date

Immediate

Expiry Date

19 Jun, 26

Salary

0.0

Posted On

21 Mar, 26

Experience

5 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

Skills

Azure Databricks, Python, SQL, Azure Cloud, Generative AI, MLOps, PySpark, CI/CD, Azure DevOps, LLMs, RAG, MLflow, Data Ingestion, Model Integration, Monitoring, Optimization

Industry

IT Services and IT Consulting

Description

Role: Lead AI Engineer - Bengaluru, INDIA Full Time *Consultants local to INDIA only Primary Responsibilities: This role focuses on building production-ready AI applications and deploying them on Azure Databricks and Azure cloud infrastructure. You will work end-to-end: from data ingestion and model integration to scalable deployment, monitoring, and ongoing optimization. The expectation is to convert AI ideas into reliable, governed, and cost-efficient applications that run in production. You will design data and AI pipelines, integrate models (including ML and Generative AI), and deploy them using Databricks workflows and Azure-native services. Success in this role requires strong hands-on experience with Azure Databricks, Python, SQL, and Azure services, along with a clear understanding of how AI systems fail in production—and how to prevent it. You will collaborate closely with data scientists, platform engineers, and business stakeholders to ensure AI applications are usable, scalable, and maintainable beyond the first release. Key Responsibilities Design and build end-to-end data and AI pipelines using Azure Databricks. Develop robust ETL/ELT workflows using Python (PySpark) and SQL. Implement CI/CD pipelines for Databricks deployments (jobs, notebooks, workflows). Integrate Databricks with Azure services (Data Lake, Blob Storage, Key Vault, Azure OpenAI, Azure Functions, etc.). Optimize jobs for performance, cost, and reliability. Build reusable, modular code. Collaborate with data scientists and platform teams to move models from experimentation to production. Implement logging, monitoring, and error handling for production pipelines. Develop and deploy ML and Generative AI models (LLMs, embeddings, RAG pipelines) for NLP, computer vision, and predictive analytics. Fine-tune LLMs using LoRA/QLoRA and integrate with Azure OpenAI or Hugging Face models. Implement vector search and retrieval pipelines using FAISS or Azure Cognitive Search. Ensure responsible AI practices, including bias detection and model governance. Good to Have (Strong Advantage) Experience with ML and Generative AI workloads on Databricks. RAG, embeddings, or inference pipelines. Terraform / ARM / Bicep for infrastructure. Databricks Asset Bundles. Airflow or ADF orchestration. Production monitoring and cost optimization experience. Knowledge of LangChain or similar frameworks for AI application development. Experience with Azure AI services (Azure Machine Learning, Azure Cognitive Services). Requirements Azure Databricks (jobs, workflows, clusters, Unity Catalog preferred). Python (PySpark-heavy, not just pandas). SQL (complex joins, window functions, analytical queries). Azure Cloud (ADLS Gen2, ADF, Key Vault, IAM concepts). Pipeline orchestration & deployment (CI/CD, environment promotion). Azure DevOps. Strong understanding of ML lifecycle and MLOps best practices. Experience with model deployment using MLflow or similar frameworks. Flexible work from home options available.

Responsibilities

This role involves building production-ready AI applications and deploying them on Azure Databricks and Azure cloud infrastructure, covering the entire lifecycle from data ingestion to scalable deployment and optimization. Key tasks include designing data and AI pipelines, integrating models (including Generative AI), and ensuring applications are reliable, governed, and cost-efficient.