Machine Learning Engineer

🔥 Who We Are

AtUltralytics, we relentlessly drive innovation in AI, building the world's leadingopen-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. Joinour teamand help shape the future ofVision AI.

🌎 Location and Legalities

This full-time Machine Learning Engineer position is based onsite in our brand-newUltralytics officein 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-classUltralytics YOLOmodels. 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 ofcomputer vision tasks, including detection, segmentation, and classification.
• Writing highly efficient, scalable, and production-ready code inPythonusing thePyTorchframework.
• Optimizing models for high-performance inference on diverse hardware using tools likeNVIDIA TensorRT,OpenVINO, andONNX.
• Managing and processing large-scaledatasetsand implementing advanceddata augmentationstrategies.
• Designing and maintaining robustCI/CD pipelineswith GitHub Actions for automated model training, testing, and benchmarking.
• Collaborating with our research and engineering teams to implement cutting-edge techniques and contribute to ouropen-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 supportingUltralytics' missionto make AI easy and accessible for everyone.

🛠️ Skills and Experience

• 5+ years of professional experience inMachine Learning Engineeringor a similar role.
• Deep expertise inPythonand deep learning frameworks, with a strong preference forPyTorch.
• Proven experience withcomputer visionand a strong understanding of model architectures like transformers and CNNs.
• Hands-on experience withmodel optimization(i.e. quantization, pruning) andmodel deploymentframeworks such asTensorRT,ONNX Runtime, andOpenVINO.
• Proficiency withCUDAprogramming and optimizing code for GPU acceleration.
• Strong background inMLOpspractices, including CI/CD usingGitHub Actionsand containerization withDocker.
• Excellent problem-solving skills and the ability to thrive in a fast-paced, high-intensity environment.
• Experience contributing to majoropen-source projectsis 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 ourmission, vision, and values. Ourteamthrives 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-edgeAI projects.
• Gear:Brand-new Apple MacBook and Apple Display provided.
• Team:Become part of a supportive and passionateteam environment.

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