ML Engineer
Core
Developing, training, validating, and deploying state-of-the-art computer vision models (detection, segmentation, classification) using the YOLO framework.
Role type
Senior Machine Learning Engineer (Computer Vision)
Builds
Production-ready AI models and inference pipelines for computer vision tasks
Domain
Artificial Intelligence / Computer Vision
Deliverable
production ML models
Required skills
Python, PyTorch, Computer Vision, Model Optimization (quantization/pruning), CUDA programming, MLOps, CI/CD, Docker
Preferred skills
Open-source contributions, experience with transformers and CNNs
Technologies
PyTorch, NVIDIA TensorRT, OpenVINO, ONNX, GitHub Actions, Docker, CUDA
Responsibilities
Develop and train models for detection, segmentation, and classification; Optimize models for high-performance inference on diverse hardware; Manage large-scale datasets and implement data augmentation; Design and maintain CI/CD pipelines for automated training and benchmarking; Collaborate with research teams to implement cutting-edge techniques; Create documentation and support users with the technology
Seniority
Senior, hands-on IC