Job Description
Job Title: Senior AI / Machine Learning Engineer (Databricks)
Core Technical Skills
- Strong proficiency in Python (mandatory); experience with SQL for data analysis and feature engineering.
- Handson experience with machine learning libraries such as Scikitlearn, TensorFlow, PyTorch, and MLflow.
- Strong experience working on Databricks for model development, training, and orchestration.
- Experience with Spark (PySpark) for largescale data processing and feature pipelines.
- Proficiency with cloud platforms (Azure Databricks / AWS / GCP), including storage and compute optimization.
Responsibilities
- Design, train, and validate machine learning models (classification, regression, risk scoring, NLPbased models) using Databricks.
- Own endtoend ML lifecycle: data ingestion feature engineering model training evaluation deployment.
- Build and manage ML pipelines using Databricks, MLflow, and Delta Lake.
- Work with large, complex datasets; clean, preprocess, and engineer features for highquality model training.
- Implement model tracking, versioning, and monitoring using MLflow.
- Deploy AI/ML models for consumption by downstream systems via APIs or batch inference jobs.
- Explain model logic and outputs clearly to business and engineering stakeholders (model explainability).
- Optimize and retrain models to improve accuracy, performance, and stability over time.
Production & Architecture
- Implement scalable AI solutions in production using Databricks Jobs, REST APIs, and cloud-native services.
- Experience with Docker (good to have) and production-grade ML practices.
- Ensure models meet performance, reliability, and compliance expectations.
- Handle data drift, model degradation, and retraining strategies.
Nice to Have
- Experience with NLP use cases (text classification, sentiment analysis, survey insights, complaint analysis).
- Familiarity with feature stores, model governance, and MLOps best practices.
Experience integrating ML models with
Salesforce, APIs, or enterprise data
Qualifications
BE
Range Of Year Experience-Min Year
2
Range Of Year Experience-Max Year
4
