Machine Learning Engineer II, Computer Vision

Location: Houston, Texas/Remote/Hybrid Employment Type: Full-Time Business Unit: Technology Department: Engineering Reports To: Senior Machine Learning / AI Engineer Direct Reports: No

Company Overview

We are an innovative start-up at the forefront of Machine Learning and Computer Vision technology, revolutionizing the retail industry through cutting-edge solutions.

Position Overview

Imagine a warehouse team, a farmer, or a city planner training a state-of-the-art object detector on their own images, without needing an ML team. You will build the platform that makes it possible. You will design the systems that turn raw images into labeled datasets, run training at scale on GPU clusters, benchmark every model, and put it into production with a single click. Every improvement you ship gives thousands of customers a better model. You will own real features from day one, work closely with data, platform, and product teams, and have senior engineers in your corner as you grow.

You will write and maintain new parts of the platform that give users the power to build their own models. Your work will scale across a wide range of use cases, and along the way you will gain deep insight into how computer vision models work and how to make them better.

Key Responsibilities

• Train, fine-tune and evaluate object detection models (for example YOLO-family, DETR-family, Faster R-CNN) for production use cases.
• Build and maintain data pipelines for image and video datasets, including labeling workflows and quality checks.
• Define evaluation metrics (mAP, precision/recall per class, latency, IoU) and run error analysis to decide what to fix next.
• Optimize models for deployment through quantization, pruning, and export to ONNX or TensorRT.
• Deploy and monitor models in production, tracking accuracy drift and data distribution shift.
• Run experiments with clear tracking and reproducible results.
• Write clean, tested Python code and take part in code reviews.
• Share findings with product and engineering stakeholders in plain terms.

Required Qualifications

• 2–5 years of professional experience building ML systems, with 2+ years focused on computer vision.
• BS in Computer Science, Computer or Electrical Engineering, or a related field.
• Have shipped at least one object detection model to production and maintained it after launch.
• Strong Python skills and hands-on experience with PyTorch.
• Working knowledge of modern detection architectures, including one-stage and two-stage detectors, and transformer-based detectors.
• Understands detection fundamentals: anchors and anchor-free methods, IoU, Non-Maximum Suppression, mAP, and class imbalance.
• Experience with image augmentation, dataset curation, and annotation quality control.
• Comfortable with Git, Linux, Docker and training on GPUs.

Preferred Qualifications

• MS or PhD with a Computer Vision focus.
• Experience with multi-object tracking, instance segmentation, or video-based detection.
• MLOps experience: experiment tracking, model registries, CI/CD for models, and production monitoring.
• Experience with distributed training on multiple GPUs.
• Experience with edge deployment (NVIDIA Jetson, mobile, or embedded) and inference optimization with ONNX Runtime or TensorRT.
• Publications, open-source contributions, or competition results in computer vision.

Tech Stack

Area

Tools

Language

Python

Frameworks

PyTorch, torchvision, Ultralytics

Image processing

OpenCV, Albumentations, Pillow

Inference

ONNX Runtime, TensorRT, Triton Inference Server

Infrastructure

Docker, Kubernetes, AWS/GCP GPUs

Compensation and Benefits

• Salary Range: $100,000 – $150,000
• Benefits: We offer a comprehensive benefits package designed to support the health, financial well-being, and work-life balance of our employees. Benefits include medical, dental, and vision insurance; a 401(k)-retirement savings plan; life insurance; a Flexible Spending Account (FSA); short-term and long-term disability coverage; and access to an Employee Assistance Program (EAP). Employees also enjoy paid time off and company-paid holidays, helping them recharge and maintain a healthy balance between their personal and professional lives.

What We Offer

• Opportunity to make a significant impact in a rapidly growing innovative industry.
• Collaborative and forward-thinking team culture.
• Competitive compensation package with robust benefits offering.

Our Core Values

At the heart of our company culture are the core values that guide everything we do—from how we work together to how we serve our customers. These values are more than just words; they are the principles that shape our decisions, define our relationships, and drive our success. We seek individuals who not only align with these values but are also inspired to contribute to a purpose-driven, inclusive, and growth-oriented environment.

• Pursue Excellence Relentlessly: We strive for the highest standards in everything we do, never settling for mediocrity

• Own the Outcome: We are responsible from start to finish. We don’t just complete tasks—we deliver results.

• Proactively Seek Solutions: Innovation starts with initiative. We anticipate, innovate, and act.

• Collaborate to Win: Teamwork fuels success. We share knowledge, support each other, and win together.

• Adapt with Purpose: Change is constant; We move forward intentionally, not reactively.

Equal Opportunity Employer

Amniscient is an equal opportunity employer. We consider all qualified applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic.