Lead AI Engineer - Bengaluru, INDIA

Vytwo Technologies Inc.; Prosper (TX); Hybrid; USD 120,000 - 150,000; Full time

Benefits offered by this job

Flexible work from home options

Job summary

Vytwo Technologies Inc. is seeking a Lead AI Engineer to focus on building production-ready AI applications and deploying them on Azure Databricks and Azure cloud infrastructure. The role involves end-to-end responsibilities, from data ingestion to scalable deployment and optimization.

The ideal candidate should have strong experience in Azure Databricks, Python, SQL, and Azure services, working closely with data scientists and business stakeholders to ensure effective AI applications.

Qualifications

• Solid Azure Databricks experience, handling jobs and workflows.
• Proficient in Python with a focus on PySpark, not just pandas.
• Ability to write complex SQL queries including joins and window functions.

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.

Skills

Azure Databricks

Python (PySpark)

SQL

Azure Cloud

ML lifecycle and MLOps

Tools

Azure DevOps

MLflow

Terraform

Job description

Role: Lead AI Engineer - Bengaluru, INDIA

Full Time

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.