Machine Learning Engineer / Computer Vision Engineer

Job Overview

We are seeking a Machine Learning Engineer to join our team and help deploy vision models into production systems with tight latency and throughput constraints.

Responsibilities

• Fine-tuning open-source PyTorch vision models on proprietary datasets.
• Optimizing and deploying models into real-time C++ runtimes for inference at the edge.
• Designing experiments to explore accuracy vs latency trade-offs.
• Owning model integration across the product and measuring impact in production.
• Maintaining and improving ML systems deployed across a distributed environment.
• Staying up-to-date with the latest research in deep learning, vision, and foundation models, and turning ideas into deployed features.

Qualifications

• Strong foundations in ML: linear algebra, probability, supervised/self-supervised learning.
• Solid software engineering skills with the ability to write clean, production-grade code.
• Curiosity and fluency in modern AI research, particularly in vision and LLMs.
• Understanding of Docker and containerized workflows.

Bonus Experience

• Experience with PyTorch: training, fine-tuning, model export.
• Proficiency in C++ for production inference and deployment.
• Familiarity with tools like TorchScript, ONNX Runtime, TensorRT.
• Knowledge of quantization (INT8, FP16) and performance optimization for inference.