Computer Vision Engineer (PyTorch/TensorRT)
Flatgigs
Posted: January 16, 2026
Interested in this position?
Create a free account to apply with AI-powered matching
Quick Summary
We are seeking a Computer Vision Engineer with strong software and AI fundamentals to build and deploy high-performance AI models. You will handle the full pipeline—from training detection and segmentation models to optimizing them for production using NVIDIA TensorRT and Docker. This role requires expertise in Python programming, machine learning, and computer vision.
Required Skills
Job Description
We are seeking a Computer Vision Engineer with strong software and AI fundamentals to build and deploy high-performance AI models. You will handle the full pipeline—from training detection and segmentation models to optimizing them for production using NVIDIA TensorRT and Docker.
Core Responsibilities
• Model Training: Train and fine-tune models for Detection, Classification, and Segmentation (e.g., YOLO, ResNet, U-Net).
• Tracking: Implement Multi-Object Tracking (MOT) algorithms for complex video streams.
• Engineering: Write production-grade Python code with a focus on modularity and scalability.
• Deployment: Containerize applications using Docker for consistent deployment.
Requirements:
• 3+ years in CV/Deep Learning.
• Python, PyTorch, OpenCV.
• Strong preference for experience with NVIDIA TensorRT and model optimization (quantization/pruning).
• Solid grasp of software engineering principles (Git, testing, CI/CD).
• Can work on other non-vision AI implementations