Computer Vision Engineer (PyTorch/TensorRT)

6 months, 1 week 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

Artificial Intelligence Engineer

MKS2 Technologies 51-250 Internet Software & Services

MKS2 Technologies is seeking a remote Artificial Intelligence Engineer to technically lead the design, deployment, and optimization of AI and machine learning systems supporting a federal government program.

Cybersecurity Feature Engineering Generative AI Machine Learning NLP
16 hours, 23 minutes ago

Senior Full-Stack AI Product Engineer

CGS Immersive 5001-10000 Enterprise learning

CGS is seeking a Senior Full-Stack AI Product Engineer to turn product requirements and UX concepts into polished, LLM-powered web products and evolve successful prototypes into maintainable production systems.

LLM Next.js React UX Design
17 hours, 38 minutes ago

Python/GenAI Solutions Architect

Provectus 251-1K Professional Services

Provectus is seeking a Senior Software Engineer or Solutions Architect to embed with enterprise client teams and deliver production-ready Python, cloud, and generative AI systems from discovery through deployment.

AWS CI/CD Django Docker FastAPI Flask GCP Generative AI GitHub Actions GitLab CI Go Kubernetes Machine Learning Microservices Node.js Pytest Python React REST API Rust SQL Vue.js
1 day, 17 hours ago

[Job - 31015] Master Software Developer ( AI Generative)

CI&T 5K-10K Internet Software & Services

CI&T is seeking a Master Software Developer to design, build, and scale secure, production-grade Generative AI and Agentic AI solutions for enterprise clients in a global delivery environment.

AWS Azure CI/CD Databricks Docker Generative AI Kubernetes MLOps Python Secrets Management Vertex AI
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