ML Engineer

About Ultralytics: At Ultralytics https://ultralytics.com/, we commit to relentless innovation in the AI space and seek team members https://www.ultralytics.com/about who resonate with our ambition to produce the world's best YOLO AI models https://ultralytics.com/yolo. 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. 🌎 Where and how you can work This full-time Machine Learning Engineer position is based at our brand-new Ultralytics office https://www.ultralytics.com/about 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. 🚀 What you'll do As a Machine Learning Engineer at Ultralytics, you will be at the forefront of developing and refining our world-class Ultralytics YOLO https://docs.ultralytics.com/models/yolo26/ models. You will work on the entire lifecycle of our models, from research and development to high-performance deployment. Key responsibilities include: - Developing, training, and validating state-of-the-art models for a variety of computer vision tasks https://docs.ultralytics.com/tasks/, including detection, segmentation, and classification. - Writing highly efficient, scalable, and production-ready code in Python https://docs.ultralytics.com/usage/python/ using the PyTorch https://pytorch.org/ framework. - Optimizing models for high-performance inference on diverse hardware using tools like NVIDIA TensorRT https://docs.ultralytics.com/integrations/tensorrt/, OpenVINO https://docs.ultralytics.com/integrations/openvino/, and ONNX https://docs.ultralytics.com/integrations/onnx/. - Managing and processing large-scale datasets https://docs.ultralytics.com/datasets/ and implementing advanced data augmentation https://docs.ultralytics.com/guides/yolo-data-augmentation/ strategies. - Designing and maintaining robust CI/CD pipelines https://docs.ultralytics.com/help/CI/ 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 https://github.com/ultralytics. - Engaging with our global community by creating documentation, tutorials, and supporting users to solve real-world problems with our technology. Your expertise will be critical in advancing the capabilities of our models and supporting Ultralytics' mission https://www.ultralytics.com/about to make AI easy and accessible for everyone. 🧠 Skills and experience - 5+ years of professional experience in Machine Learning Engineering https://www.ultralytics.com/blog/aspiring-ml-engineer-8-tips-you-need-to-know or a similar role. - Deep expertise in Python https://www.python.org/ and deep learning frameworks, with a strong preference for PyTorch https://pytorch.org/. - Proven experience with computer vision https://www.ultralytics.com/glossary/computer-vision-cv and a strong understanding of model architectures like transformers and CNNs. - Hands-on experience with model optimization https://www.ultralytics.com/glossary/optimization-algorithm (i.e., quantization and pruning) and model deployment https://docs.ultralytics.com/guides/model-deployment-options/ frameworks such as TensorRT https://developer.nvidia.com/tensorrt, ONNX Runtime https://onnxruntime.ai/, and OpenVINO https://www.intel.com/content/www/us/en/developer/tools/openvino-toolkit/overview.html. - Proficiency with CUDA https://developer.nvidia.com/cuda-zone programming and optimizing code for GPU acceleration. - Strong background in MLOps https://www.ultralytics.com/glossary/machine-learning-operations-mlops practices, including CI/CD using GitHub Actions https://github.com/features/actions and containerization with Docker https://www.docker.com/. - Excellent problem-solving skills and the ability to thrive in a fast-paced, high-intensity environment. - Experience contributing to major open-source projects https://docs.ultralytics.com/help/contributing/ is a significant advantage. 🌟 Cultural fit 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 https://handbook.ultralytics.com/mission-vision-values/. Our team https://www.ultralytics.com/about 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. 💰 Compensation and 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 https://www.ultralytics.com/solutions. - Gear: Brand-new Apple MacBook Air/Pro, Apple Studio Display, and AirPods Pro 3. - Team: Become part of a supportive and passionate team environment https://www.ultralytics.com/about. If you are driven to redefine the capabilities of machine learning and eager to make a significant impact, Ultralytics https://www.ultralytics.com/careers offers an exceptional career opportunity. What we offer: Cutting-Edge, Next-Generation AI Computer Vision Technology: Contribute to building cutting-edge computer vision AI models based on the YOLO framework. Impactful Work: Shape the future of AI-powered solutions that transform industries. Collaborative Culture: Join a talented and passionate team that values open communication and innovation. Ultralytics Handbook Comprehensive guide to our company's mission, vision, values, and operational practices. This handbook is designed to provide key insights and resources for our (future) team members, collaborators, and community to align with Ultralytics' core principles. Link: https://handbook.ultralytics.com/ Ultralytics is an equal opportunity employer committed to building an inclusive workplace. We believe that everyone should be able to bring their whole selves to work, and we do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability, or any other legally protected status.