Machine Learning Engineer

🔥 Who We Are

At Ultralytics, we relentlessly drive innovation in AI, building the world\’s leading open-source models. We\’re looking for passionate individuals obsessed with AI, eager to make a global impact, and ready to excel in a dynamic, high-energy environment. Join our team and help shape the future of Vision AI.

🌎 Location and Legalities

This full-time Machine Learning Engineer position is based onsite in our brand-new Ultralytics office in London, UK. Applicants must have legal authorization to work in the UK, 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 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, 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.

Your expertise will be critical in advancing the capabilities of our models and supporting Ultralytics\’ mission to make AI easy and accessible for everyone.

🛠️ Skills and Experience

• 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, 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.
• Experience contributing to major open-source projects is a significant advantage.

🌟 Cultural Fit – Intensity Required

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.

💶 Compensation and Benefits

• Competitive Salary: Highly competitive based on experience.
• Startup Equity: 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 Hours: Tailor your working hours to suit your productivity.
• Tech: Engage with cutting-edge AI projects.
• Gear: Brand-new Apple MacBook and Apple Display provided.
• Team: Become part of a supportive and passionate team environment.

If you are driven to redefine the capabilities of machine learning and eager to make a significant impact, Ultralytics offers an exceptional career opportunity.