SIGN IN
AI/ML Architect with Databricks | Los Angeles CA (Hybrid) jobs in United States
info-icon
This job has closed.
company-logo

TestingXperts · 1 day ago

AI/ML Architect with Databricks | Los Angeles CA (Hybrid)

TestingXperts is seeking a skilled AI/ML Architect with hands-on experience in Databricks to join their team. The role involves developing and optimizing machine learning models, building scalable data pipelines, and collaborating with cross-functional teams to drive data-driven decision-making.
DevOpsInformation TechnologyPenetration TestingQuality AssuranceSoftwareUsability Testing
check
H1B Sponsor Likelynote
Hiring Manager
Riyaz Ahmad
linkedin

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 as required by the business
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
Communicate complex statistical concepts in a clear and concise manner

Qualification

DatabricksPythonPySparkMachine LearningData EngineeringMLflowDelta LakeStatistical TechniquesAnalytical SkillsCommunication Skills

Required

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
Advanced knowledge of Databricks, including: 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

Preferred

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) is a plus

Company

TestingXperts

company-logo
Next Gen QA & Software Testing Company

H1B Sponsorship

TestingXperts has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (1)
2020 (1)

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Manish Gupta
Founder & CEO
linkedin
leader-logo
Archana Gupta
CFO
linkedin
Company data provided by crunchbase