Machine Learning Engineer – Computer Vision, NLP & Backend

Overview

We are hiring a Machine Learning Engineer with strong experience in Computer Vision, Natural Language Processing (NLP), and Backend Development. You’ll be responsible for training and deploying deep learning models, implementing research papers, and building production-ready APIs using Flask, FastAPI, or Django.

Key Responsibilities

• Develop ML/DL models for:
• Computer Vision: classification, object detection, segmentation.
• NLP: text classification, NER, summarization.
• Implement research papers and build production-ready prototypes.
• Use libraries such as PyTorch, OpenCV, Pillow, TorchVision, and Transformers.
• Optimize models with techniques like quantization, pruning, ONNX export, and TorchScript.
• Build and deploy RESTful APIs using FastAPI, Flask, or Django.
• Containerize applications using Docker and deploy them to cloud or local servers.
• Write clean, efficient, and scalable code for backend and ML pipelines.

Required Skills

• Solid understanding of Machine Learning and Deep Learning.
• Experience with:
• PyTorch, OpenCV, Pillow
• Computer Vision (detection, segmentation, classification)
• NLP libraries like Hugging Face Transformers, spaCy, NLTK
• Strong backend skills using FastAPI, Flask, or Django.
• Familiar with Docker, Git, and Linux environments.
• Experience with model deployment and optimization tools (ONNX, TorchScript).

Nice to Have

• Knowledge of Generative AI / LLMs
• Experience with ONNX, TensorRT, or TorchScript
• MLOps tools: MLflow, DVC, Airflow
• Experience with Cloud platforms (AWS, GCP, Azure)

Qualifications

• Bachelor’s or Master’s in Computer Science, Artificial Intelligence, Data Science, or a related field.

Job Type: Full-time

Pay: ₹272,876.24 - ₹1,548,746.18 per year

Schedule:

• Evening shift
• Monday to Friday

Work Location: In person