Senior Software Engineer
Infinite Computer Solutions
About the role
Job Description
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
Don't send a generic resume
Paste this job description into Mimi and get a resume tailored to exactly what the hiring team is looking for.
Get started free