Skills required
ml_engineering
Databricks
mlflow
automl
mosaic_ai
pytorch_and_tensorflow
hyperopt
Education/Qualification
Bachelor’s degree in a related field
Desirable skills
Ability to influence and build trust with C-suite stakeholders.
Years of experience
5–12 years
Position: ML Engineer – Databricks
Duties & Responsibilities:
• MLflow: experiment tracking, model registry, and model versioning.
• Databricks Feature Store for feature engineering and sharing.
• Model Serving for real-time and batch inference endpoints.
• AutoML for baseline model generation and feature importance.
• Distributed training with Spark MLlib, XGBoost, and Horovod.
• PyTorch and TensorFlow on Databricks clusters (single-node and distributed).
• Hyperopt for hyperparameter tuning at scale.
• Lakehouse Monitoring for model drift and data quality tracking.
Experience:
• 3+ years in ML engineering with at least 1 year deploying models on Databricks.
• Delivered 2+ production ML models with MLflow tracking and serving.
• Clear communication skills for client interactions and technical documentation.
Company Description:
At Mastech Digital, we solve meaningful business problems using data, AI, and modern digital technologies. Our teams work closely with global enterprises to build solutions that create real, measurable impact.
We bring deep industry expertise across Utilities & Energy, Financial Services, Healthcare, Retail, and Technology. With an AI-first mindset and a collaborative culture, we empower our people to innovate, grow, and help clients move forward with confidence.