Machine Learning Engineer (Computer Vision Focus)

Job Description:
We are hiring a Machine Learning Engineer with strong Computer Vision expertise to develop, train, deploy, and operationalize ML models. This role involves hands-on work in PyTorch, building FASTAPI-based inference services, managing cloud GPU workloads (Runpod/AWS/GCP), and contributing to both CV-centric and occasional LLM-related tasks.

Key Responsibilities

• Develop, train, and optimize CV models using PyTorch/TensorFlow.
• Build and maintain FASTAPI REST APIs for model inference.
• Create reproducible training pipelines, experiment tracking, and data workflows.
• Work on CV tasks: detection, segmentation, OCR, tracking, vision transformers.
• Deploy and monitor ML workloads on Runpod, AWS, or GCP.
• Package and integrate ML components with backend systems.
• Research and evaluate state-of-the-art CV techniques.
• Support LLM tasks (fine-tuning, evaluation, prompting).
• Write clean, modular Python code and internal documentation.
• Participate in code reviews, benchmarking, and iterative improvements.

Required Skills

• Bachelor’s degree in CS, AI, Data Science, or related field.
• 1–2 years of practical ML/CV engineering experience.
• Strong in Python and the PyTorch ecosystem.
• Hands-on experience with FASTAPI.
• Exposure to CV workflows: classification, detection, segmentation, OCR.
• Familiarity with cloud GPUs (Runpod preferred, AWS/GCP acceptable).
• Good understanding of Git, Linux, and software engineering practices.
• Basic knowledge of MLOps (reproducibility, packaging, deployment).

Nice-to-Have

• Experience with model optimization (quantization, ONNX, pruning).
• Docker and containerized deployments.
• Basic knowledge of CI/CD.
• Familiarity with LLMs or prompt engineering.
• GPU performance optimization skills.

What We Offer

• Work on production-grade ML and CV systems.
• A collaborative engineering culture.
• Exposure to modern training and deployment stacks.
• Growth opportunities across research, engineering, and infrastructure.

Job: Full-Time

• Timings: 9 AM to 6 PM
• Working Days: Monday to Friday
• Budget: PKR 80k to 170k
• Location: ML Bench, 424, J3 Johar Town, Lahore

Join ML Bench by sending your resume here hr@ml-bench.com

Job Type: Full-time

Pay: Rs80,000.00 - Rs170,000.00 per month

Application Question(s):

• Apply only if you can attend an on-site interview?
• This is an onsite position apply only if you are residing in Lahore?
• What is your current salary?
• What is your expected salary?
• Please mention your linkedIn profile link here?

Work Location: In person