Machine Learning Engineer II, Computer Vision

Location & Employment

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

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:

• Medical, dental, and vision insurance
• 401(k) retirement savings plan
• Life insurance
• Flexible Spending Account (FSA)
• Short-term and long-term disability coverage
• Access to an Employee Assistance Program (EAP)
• Paid time off and company-paid holidays

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

• 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.