AI ML Engineer

Title

AI ML Engineer

Location

Bangalore

Exp

5+ Yrs

Job Description

• 3+ years of hands-on data engineering on production pipelines.

Python (strong)

production-quality code, pytest, PySpark, pandas, boto3.

SQL (strong)

window functions, CTEs, incremental logic, tuning queries on large, partitioned tables.

APIs

Building APIs

has designed and built REST APIs with FastAPI or Flask, and deployed them on AWS (API Gateway + Lambda, or containers)

Fetching data from APIs

has built production ingestion from REST/GraphQL APIs with requests/httpx. Handles pagination, rate limits, retries and incremental loads

API integration

OAuth 2.0 and API-key authentication, JSON/XML payloads, webhooks, and OpenAPI/Swagger specs

AWS data stack

AWS Glue

ETL jobs, crawlers, Data Catalog, Glue Studio

Amazon Athena

query authoring, workgroups, federated queries, cost controls

Amazon SageMaker Unified Studio

projects, Lakehouse, Catalog, unified notebooks and query editor

Amazon S3

data lake layout, partitioning, lifecycle policies

Supporting services

Lambda, API Gateway, Step Functions, EventBridge, IAM, CloudWatch, Secrets Manager, Lake Formation

Agentic and AI tools

has used AI coding agents in real work, and understands Amazon Bedrock (agents, knowledge bases), Model Context Protocol (MCP), tool and prompt design, and RAG.

• Workflow orchestration with Airflow, Step Functions or an equivalent tool.
• Data modelling (dimensional and event-based), plus Git and CI/CD practice.