Computer Vision Engineer (PyTorch/TensorRT)

5 months, 3 weeks ago
Mid Level
Software Development
Flatgigs

Flatgigs

FlatGigs tailors recruitment strategies for flat-structured companies, aligning teams with skills and culture. Utilizing tech tools for precise candidate organization.

Professional Services

Description

  • Train and fine-tune detection, classification, and segmentation models (e.g., YOLO, ResNet, U-Net).
  • Implement multi-object tracking (MOT) algorithms for complex video streams.
  • Optimize models for production deployment, including quantization, pruning, and conversion for inference runtimes.
  • Write production-grade Python code with a focus on modularity, scalability, and maintainability.
  • Containerize applications and manage consistent deployments using Docker.
  • Own the end-to-end pipeline from data and model training to production rollout and monitoring.
  • Support and implement non-vision AI solutions as needed.

Requirements

  • 3+ years of experience in computer vision and deep learning.
  • Proficiency in Python, PyTorch, and OpenCV.
  • Experience training and fine-tuning detection, classification, and segmentation models (examples: YOLO, ResNet, U-Net).
  • Experience implementing multi-object tracking (MOT) on video streams.
  • Strong preference for experience with NVIDIA TensorRT and model optimization techniques (quantization, pruning).
  • Familiarity with containerization and deployment using Docker.
  • Solid software engineering fundamentals, including Git, testing, and CI/CD practices.
  • Ability to work on other non-vision AI implementations when required.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

AI Developer

AI Developer (Contract) at a remote company, responsible for building AI-powered workflows and internal tools that improve business efficiency and create new capabilities.

JavaScript LLM Python
17 hours, 24 minutes ago

AI Agent Implementations Specialist

Extreme Networks 1K-5K IT Services

Extreme Networks is hiring an AI Agent Developer & Marketo Administrator to support marketing automation, AI agent implementation, and MarTech operations across its global marketing organization.

CRM JavaScript Machine Learning Node.js Python REST API
1 day, 17 hours ago

Entrepreneur in Residence - Technical Co-founder (CTO)

FutureSight 11-50 Internet Software & Services

FutureSight is seeking a Co-Founder & CTO to lead the technical direction of a new B2B AI venture from inception to launch and early growth.

LLM System Design
1 day, 17 hours ago

Technical Co-founder (CTO) - AI Collections Agent

FutureSight 11-50 Internet Software & Services

Argentl is seeking a Co-Founder & CTO to lead product and engineering for an AI collections platform that automates debtor follow-up for finance teams at B2B companies.

LLM
1 day, 17 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers