We are looking for a skilled Machine Learning Engineer with a strong background in Computer Vision to join our team. The ideal candidate will take ownership of developing, training, deploying, and operationalizing machine learning models, primarily using PyTorch. This role involves hands-on experience in building FASTAPI-based inference services and managing cloud GPU workloads on platforms such as Runpod, AWS, or GCP. The engineer will contribute to various computer vision projects and occasionally support large language model (LLM) related tasks. This position offers the chance to work on advanced ML systems within a collaborative and innovative engineering environment.

Key Responsibilities
- Develop, train, and optimize computer vision models utilizing PyTorch and TensorFlow frameworks.
- Build and maintain FASTAPI REST APIs to serve model inference efficiently.
- Design and implement reproducible training pipelines, manage experiment tracking, and create streamlined data workflows.
- Work on diverse computer vision tasks including detection, segmentation, OCR, tracking, and vision transformers.
- Deploy, monitor, and manage machine learning workloads on cloud GPU platforms such as Runpod, AWS, or GCP.
- Package and integrate machine learning components with backend systems to ensure smooth deployment.
- Conduct research and evaluate state-of-the-art computer vision techniques to improve model accuracy and performance.
- Support LLM-related activities including fine-tuning, evaluation, and prompt engineering.
- Write clean, modular Python code with comprehensive internal documentation.
- Participate actively in code reviews, benchmarking, and iterative improvements to uphold high-quality standards.

Required Qualifications
- Bachelor’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field.
- 1 to 2 years of practical experience in machine learning and computer vision engineering.
- Strong proficiency in Python and the PyTorch ecosystem.
- Hands-on experience with FASTAPI for developing APIs.
- Familiarity with computer vision workflows such as classification, detection, segmentation, and OCR.
- Experience working with cloud GPU environments, preferably Runpod; AWS and GCP experience is also acceptable.
- Solid understanding of Git version control, Linux operating systems, and software engineering best practices.
- Basic knowledge of MLOps principles including reproducibility, packaging, and deployment.

Preferred Qualifications and Benefits
- Experience with model optimization techniques such as quantization, ONNX, and pruning.
- Proficiency in Docker and containerized deployment workflows.
- Basic understanding of Continuous Integration/Continuous Deployment (CI/CD) pipelines.
- Familiarity with large language models (LLMs) and prompt engineering.
- Skills in GPU performance optimization are considered a plus.

What We Offer
- The opportunity to work on production-grade machine learning and computer vision systems.
- A collaborative and supportive engineering culture that encourages growth.
- Exposure to modern training and deployment technology stacks.
- Career development opportunities across research, engineering, and infrastructure domains.

Additional Information
This is a full-time, on-site position based at ML Bench, Johar Town, Lahore. Working hours are Monday to Friday, from 9 AM to 6 PM. The salary range for this role is PKR 80,000 to 170,000 per month. Candidates must be able to attend an on-site interview and reside in Lahore to be considered.

This role presents an excellent opportunity for professionals eager to advance their careers in machine learning and computer vision within a dynamic and forward-thinking environment.

Job Details

Total Positions:
1 Post
Job Shift:
First Shift (Day)
Job Type:
Job Location:
Gender:
No Preference
Age:
18 - 65 Years
Minimum Education:
Bachelors
Career Level:
Entry Level
Maximum Experience:
2 Years
Apply Before:
Jan 08, 2026
Posting Date:
Dec 08, 2025

ML Bench

· 11-50 employees - Lahore

What is your Competitive Advantage?

Get quick competitive analysis and professional insights about yourself
Talk to our expert team of counsellors to improve your CV!
Try Rozee Premium
I found a job on Rozee!