AI/ML Architect with Databricks , AWS

Job Overview

We are seeking an experienced AI/ML Architect with deep hands-on expertise in Databricks on AWS to lead the design and implementation of scalable, high-performance data and machine learning platforms.

Key Responsibilities

• Develop, train, and optimize ML models using Python, PySpark, MLflow, and Databricks Machine Learning.
• Conduct exploratory data analysis (EDA) to identify patterns, trends, and insights in large datasets.
• Deploy ML models into production using MLflow, Databricks Workflows, or other MLOps pipelines.
• Build analytics solutions such as forecasting, anomaly detection, segmentation, or recommendation systems.
• Design ML architectures aligned with Databricks Lakehouse on AWS.
• Architect and build scalable ETL/ELT pipelines using PySpark, SQL, and Databricks Workflows.
• Implement Delta Lake best practices, including OPTIMIZE, ZORDER, partitioning, and schema evolution.
• Design lakehouse layers (Bronze/Silver/Gold) with strong separation of compute and serving layers.
• Optimize cluster performance and jobs using Spark tuning, caching, and shuffle minimization.
• Work with multi-terabyte, time-series, high-velocity data in a distributed environment.
• Ensure robust data availability for downstream ML and analytics workloads.
• Architect end-to-end data and ML solutions using AWS services, including:
  • S3 for storage
  • IAM for identity & access
  • Glue Catalog for metadata management
  • Networking for secure, high-throughput data movement
• Integrate Databricks with AWS-native compute, API layers, and low-latency endpoints.
• Translate business problems into scalable analytical or ML architectures.
• Communicate complex statistical and architectural concepts to non-technical stakeholders.
• Collaborate with product, engineering, and business leaders to drive data-informed initiatives.
• Provide design leadership while remaining hands-on in execution.

Skills & Qualifications

• Bachelor’s or Master’s in Computer Science, Data Science, Engineering, Statistics, or related field.
• 10+ years of experience in data engineering, ML engineering, or AI/ML architecture roles.
• Deep expertise in Databricks on AWS, including:
  • PySpark / Spark SQL
  • Databricks Notebooks
  • Delta Lake
  • Unity Catalog
  • MLflow
  • Databricks Jobs & Workflows
• Strong programming ability in Python (pandas, numpy, scikit-learn).
• Demonstrated experience with large-scale, multi-terabyte data processing.
• Strong understanding of ML algorithms, distributed systems, and data optimization.

Compensation

$60.00 - $70.00 per hour

About the Company

Vytwo Technologies is a global leader in enterprise application integration, delivering end-to-end IT consulting and business services for mid to large-scale organizations. We offer a comprehensive suite of solutions including business and technology consulting, Cloud, e-business and digital transformation services, system integration, custom application development, re-engineering, and long-term application support.