ML Engineer

📋 Job Overview

At Ultralytics, we commit to relentless innovation in the AI space and seek team members who resonate with our ambition to produce the world's best YOLO AI models. If you're obsessed with AI, eager to make an impact on the world, and thrive in dynamic, high-intensity environments, we invite you to apply for a position on our team.

🏢 About Ultralytics

Ultralytics is a high-performance environment for world-class talent obsessed with achieving extraordinary results. We operate at a relentless pace, demanding exceptional dedication and an unwavering commitment to excellence, guided by our mission, vision, and values. Our team thrives on audacious goals and absolute ownership. This is not a conventional workplace. If your priority is predictable comfort or a standard work-life balance over the relentless pursuit of progress, Ultralytics is not for you. We seek driven individuals prepared for the profound personal investment required to make a defining contribution to the future of AI.

🎯 The Role

This full-time Machine Learning Engineer position is based at our brand-new Ultralytics office in Shenzhen, China, with a hybrid model of three days in-office and two days remote. Applicants must have legal authorization to work in China, as Ultralytics does not provide visa sponsorship.

✅ Key Responsibilities

• Developing, training, and validating state-of-the-art models for a variety of computer vision tasks, including detection, segmentation, and classification.
• Writing highly efficient, scalable, and production-ready code in Python using the PyTorch framework.
• Optimizing models for high-performance inference on diverse hardware using tools like NVIDIA TensorRT, OpenVINO, and ONNX.
• Managing and processing large-scale datasets and implementing advanced data augmentation strategies.
• Designing and maintaining robust CI/CD pipelines with GitHub Actions for automated model training, testing, and benchmarking.
• Collaborating with our research and engineering teams to implement cutting-edge techniques and contribute to our open-source repositories.
• Engaging with our global community by creating documentation, tutorials, and supporting users to solve real-world problems with our technology.

📌 Required Qualifications

• 5+ years of professional experience in Machine Learning Engineering or a similar role.
• Deep expertise in Python and deep learning frameworks, with a strong preference for PyTorch.
• Proven experience with computer vision and a strong understanding of model architectures like transformers and CNNs.
• Hands-on experience with model optimization (i.e., quantization and pruning) and model deployment frameworks such as TensorRT, ONNX Runtime, and OpenVINO.
• Proficiency with CUDA programming and optimizing code for GPU acceleration.
• Strong background in MLOps practices, including CI/CD using GitHub Actions and containerization with Docker.
• Excellent problem-solving skills and the ability to thrive in a fast-paced, high-intensity environment.

⭐ Desirable Experience

• Experience contributing to major open-source projects is a significant advantage.

🎁 Benefits

• Competitive salary: Highly competitive based on experience.
• Equity packages: Participate directly in our company's growth and success.
• Hybrid flexibility: 3 days per week in our brand-new office — 2 days remote.
• Generous time off: 24 days vacation, your birthday off, plus local holidays.
• Flexible working hours: Tailor your working hours to suit your productivity.
• Tech & tools: Engage with cutting-edge AI projects.
• Gear: Brand-new Apple MacBook Air/Pro, Apple Studio Display, and AirPods Pro 3.
• Team: Become part of a supportive and passionate team environment.

🛂 Visa & Eligibility

Applicants must have legal authorization to work in China, as Ultralytics does not provide visa sponsorship.