AI/ML Architect with Databricks

About This Role

Join our team as an AI/ML Architect with strong Databricks expertise to build scalable data pipelines and machine learning models. Collaborate with cross-functional teams to drive data-driven decision-making.

Responsibilities

• Develop, train, and optimize machine learning and statistical models using Databricks, Python, PySpark, and MLflow
• Perform exploratory data analysis (EDA) to identify trends, patterns, and insights in large datasets
• Deploy ML models into production using Databricks MLflow, Delta Live Tables, or other MLOps pipelines
• Conduct A/B testing, forecasting, segmentation, anomaly detection, or recommendation systems
• Build scalable, high-performance ETL/ELT pipelines using PySpark, SQL, and Databricks workflows
• Work with Delta Lake to ensure high-quality, reliable, and performant data
• Optimize cluster usage and job performance within the Databricks environment
• Collaborate with data engineers to ensure high-quality data availability for modeling
• Translate business problems into analytical solutions and present findings to non-technical stakeholders
• Partner with product, engineering, and business teams to drive data-informed decisions

Requirements

• 3+ years of experience in data science or machine learning roles
• Advanced knowledge of Databricks (PySpark / Spark SQL, Databricks notebooks, Delta Lake, MLflow, Databricks Jobs & Workflows)
• Strong programming skills in Python (pandas, numpy, scikit-learn)
• Experience working with large-scale data processing
• Solid understanding of machine learning algorithms and statistical techniques

Qualifications

• Bachelor’s degree or higher in Computer Science, Data Science, Mathematics, Statistics, Engineering, or related field.
• 3+ years of experience in data science or machine learning roles

Nice to Have

• Experience deploying models in production using MLOps frameworks
• Knowledge of Azure Databricks or AWS Databricks environments
• Understanding of CICD pipelines and DevOps concepts (Azure DevOps, GitHub Actions, etc.)
• Familiarity with deep learning frameworks (TensorFlow, PyTorch)

Skills

* Required skills