Computer Vision Engineer (PyTorch/TensorRT)

1 month 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 Engineer - FDE (Forward Deployed Engineer)

Databricks 1K-5K IT Services

Databricks is hiring an AI Forward Deployed Engineer (FDE) to work directly with customers and internal teams to design, deliver, and productionize GenAI applications that solve real-world problems and influence product and strategic priorities.

Apache Spark AWS Azure Databricks GCP MLflow Pandas Python PyTorch Scikit-learn SQL
1 month ago

Senior ServiceNow ITSM Architect with AI & ITAM exposure

Muller Internet Software & Services

ServiceNow AI & ITSM Solution Architect at Müller Solutions responsible for designing, leading, and delivering AI-enabled ServiceNow solutions across ITSM, FSM, and Asset Management to align platform capabilities with business processes, data foundations, and measurable operational outcomes.

Agile Generative AI JavaScript Machine Learning
1 month ago

Application Engineer II

HighLevel 251-1K Internet Software & Services

HighLevel is hiring an AI-focused Application Engineer II to design, implement, and operate LangGraph-driven agent workflows and production AI pipelines that improve transcription, scoring, and automation across its high-scale sales and marketing platform.

AWS Cloudflare Go Python REST API Snowflake SQL TypeScript WebSockets
1 month ago

Applied AI Operations Lead

Canvas Medical 51-250 Health Care Providers & Services

Canvas Medical is hiring an Applied AI Operations Lead to partner with healthcare organizations to deploy, govern, and scale AI agents that automate clinical and operational workflows and deliver measurable business and clinical value.

HIPAA
1 month 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