Position: AI/ML Architect with Databricks , AWS
Location: Los Angeles, CA
Job Mode: Fulltime & Contract
Job Description:
Role 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.
The ideal candidate combines architectural thinking with strong engineering execution, demonstrating the ability to build modern lakehouse systems, optimize large scale pipelines, and drive analytical and ML capabilities across the organization.
This role requires working with large, multi-terabyte datasets, advanced analytics, and end to end ML lifecycle management using Databricks, Python, PySpark, and AWS-native services.
Skills & Qualifications
Required
- 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.
Preferred
- Experience with MLOps and production deployment pipelines.
- Strong grasp of AWS-native data and compute services.
- Understanding of CI/CD using GitHub Actions, GitLab CI, or similar.
- Familiarity with deep learning frameworks (TensorFlow, PyTorch).