ML Engineer

The role

You will develop and refine Ultralytics YOLO models through the entire lifecycle from research to deployment. This includes training computer vision models, optimizing them for hardware, and managing large-scale datasets.

What you will do

• Developing, training, and validating state-of-the-art models for computer vision tasks
• Writing efficient, scalable, and production-ready code in Python using PyTorch
• Optimizing models for high-performance inference on diverse hardware
• Managing and processing large-scale datasets and implementing data augmentation
• Designing and maintaining robust CI/CD pipelines with GitHub Actions
• Collaborating with research and engineering teams to implement techniques and contribute to open-source repositories
• Engaging with the global community by creating documentation and tutorials

Required qualifications

• 5+ years of professional experience in Machine Learning Engineering or a similar role
• Deep expertise in Python and deep learning frameworks
• Proven experience with computer vision
• Strong understanding of model architectures like transformers and CNNs
• Hands-on experience with model optimization and model deployment frameworks
• Proficiency with CUDA programming and optimizing code for GPU acceleration
• Strong background in MLOps practices including CI/CD and containerization with Docker
• Excellent problem-solving skills

Nice to have

• Strong preference for PyTorch
• Experience contributing to major open-source projects

Benefits

• Competitive salary
• Equity packages
• Hybrid flexibility: 3 days per week in office, 2 days remote
• 24 days vacation
• Birthday off
• Local holidays
• Flexible working hours
• Apple MacBook Air/Pro, Apple Studio Display, and AirPods Pro 3

Before you apply

• Applicants must have legal authorization to work in China
• Ultralytics does not provide visa sponsorship

About the company

Ultralytics commits to relentless innovation in the AI space to produce the world's best YOLO AI models.