Machine Learning Engineer

We are looking for a Machine Learning Engineer to join our team on a 6-month contract. The ideal candidate will have hands-on experience in Python-based machine learning development, deep learning frameworks, image processing, and data handling. This role involves working closely with cross-functional teams to build, experiment with, and optimize machine learning models and data pipelines.

Key Responsibilities

• Develop, train, and evaluate machine learning and deep learning models using PyTorch.
• Implement MLOps pipelines for model training and deployment.
• Use Jupyter notebooks for experimentation, analysis, and prototyping.
• Design and optimize image processing pipelines.
• Preprocess, manage, and analyze large-scale datasets.
• Collaborate using GitHub or GitLab for version control and code reviews.
• Document experiments, results, and implementation details.
•  Support model optimization and performance tuning, including GPU acceleration where applicable.

Required Technical Skills

• Strong proficiency in Python.
• Strong Fundamental knowledge of ML/DL model training and optimization.
• Strong mathematical and algorithm skills.
• Hands-on experience with setting up an MLOps pipeline.
• Hands-on experience with machine learning frameworks, particularly PyTorch.
• Experience with image processing libraries, including OpenCV and scikit-image.
• Experience working with image-based or medical datasets.
• Familiarity with GPU-based training and inference performance optimization.
• Experience in applied machine learning research or production ML systems.

Educational Qualifications

• Bachelor’s degree in Computer Science, Engineering, Data Science, or a related field

Soft skills

• Strong problem-solving and analytical skills.
• Ability to work independently.
• Good communication and documentation skills.
• Ability to adapt quickly to changing requirements.

Contract duration: 06 months
Work type: Hybrid