Arkhya Tech. Inc. · 4 hours ago
AI/ML Architect with Databricks
Los Angeles, CA
Full-time
Hybrid
Mid Level
3+ years expArkhya Tech. Inc. is seeking a skilled AI/ML Architect with hands-on experience in Databricks to join their team. The ideal candidate will develop and optimize machine learning models, build scalable data pipelines, and collaborate with cross-functional teams to drive data-driven decision-making.
ConsultingInformation TechnologySoftware
Hiring Manager
Eshwar S
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
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
Arkhya Tech. Inc.
Arkhya Technologies, Inc.
201-500 employees
H1B Sponsorship
Arkhya Tech. Inc. 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
2023 (1)
2022 (1)
2020 (5)
Funding
Current Stage
Growth StageCompany data provided by crunchbase