AI / ML Engineer

Codvo Private Limited; Pune District; On-site; INR 800,000 - 1,200,000; Full time

Job summary

A tech company is seeking a highly skilled AI/ML Engineer to design and develop scalable machine learning and deep learning solutions. The ideal candidate will focus on computer vision and deep learning frameworks, deploying models in production environments. Responsibilities include building ML pipelines and collaborating with cross-functional teams. Strong proficiency in Python and hands-on experience with PyTorch is essential. This role offers an opportunity to work on impactful AI solutions in a dynamic team environment.

Qualifications

• Strong proficiency in Python for machine learning and software development.
• Hands-on experience with PyTorch for deep learning model development.
• Solid understanding of deep learning architectures (CNNs, transfer learning, etc.).
• Practical experience in computer vision applications.

Responsibilities

• Design, develop, and optimize machine learning and deep learning models using PyTorch.
• Build and deploy computer vision solutions for real-world use cases.
• Develop end-to-end ML pipelines, including data ingestion and deployment.
• Implement and maintain MLOps workflows for model versioning and monitoring.

Skills

Python

PyTorch

Computer vision

MLOps

AWS services

Databricks

Tools

Docker

Kubernetes

Job description

Job Title: AI/ML Engineer

Job Summary

We are seeking a highly skilled AI/ML Engineer to design, develop, and deploy scalable machine learning and deep learning solutions. The ideal candidate will have strong experience in computer vision, deep learning frameworks, and cloud-based ML deployment, along with solid software engineering and MLOps practices. You will work closely with cross-functional teams to build production‑ready AI systems that deliver real business impact.

Key Responsibilities

• Design, develop, and optimize machine learning and deep learning models using PyTorch.
• Build and deploy computer vision solutions for real‑world use cases.
• Develop end‑to‑end ML pipelines, including data ingestion, preprocessing, training, validation, and deployment.
• Implement and maintain MLOps workflows for model versioning, monitoring, CI/CD, and retraining.
• Deploy and scale ML models on AWS cloud infrastructure.
• Work with large‑scale datasets using Databricks and distributed computing frameworks.
• Collaborate with data scientists, product managers, and software engineers to translate business requirements into AI solutions.
• Ensure high code quality by following software engineering best practices (modular design, testing, documentation).
• Monitor model performance in production and continuously improve accuracy, efficiency, and reliability.

Required Skills & Qualifications

• Strong proficiency in Python for machine learning and software development.
• Hands‑on experience with PyTorch for deep learning model development.
• Solid understanding of deep learning architectures (CNNs, transfer learning, etc.).
• Practical experience in computer vision applications.
• Experience working with Databricks and large‑scale data processing.
• Strong knowledge of AWS services for ML deployment (EC2, S3, SageMaker, etc.).
• Experience with MLOps tools and practices (model deployment, monitoring, CI/CD).
• Good understanding of software engineering principles and production‑grade system design.

Preferred Qualifications

• Experience deploying ML models in production environments.
• Familiarity with containerization tools such as Docker and orchestration platforms like Kubernetes.
• Exposure to real‑time or batch inference systems.
• Experience working in agile or fast‑paced development environments.

Nice to Have

• Experience with optimization and performance tuning of ML models.
• Knowledge of data security and compliance in cloud environments.
• Experience with monitoring tools for ML model performance and drift detection.