Machine Learning Engineer

Ultralytics; London; Hybrid; GBP 85,000 - 120,000; Full time

Benefits offered by this job

Competitive salary

Startup equity

Generous time off

Flexible hours

New Apple tech gear

Supportive team environment

Job summary

A leading AI company in London is seeking an experienced Machine Learning Engineer to advance state-of-the-art YOLO models. The role includes model development across the lifecycle and requires deep expertise in Python and PyTorch. If you're passionate about innovation and eager to make an impact, apply to join a dynamic team committed to redefining AI capabilities.

Qualifications

• 5+ years of professional experience in Machine Learning Engineering or a similar role.
• Deep expertise in Python and preference for PyTorch.
• Hands-on experience with model optimization and deployment frameworks.
• Hands-on experience with model optimization (quantization, pruning) and deployment frameworks such as TensorRT, ONNX Runtime, and OpenVINO.
• Proficiency with CUDA programming and GPU optimization.
• Strong background in MLOps, including CI/CD with GitHub Actions and containerization with Docker.
• Excellent problem-solving skills in a fast-paced environment.

Responsibilities

• Develop, train, and validate state-of-the-art models for computer vision tasks.
• Write efficient, scalable code for high-performance model deployment.
• Manage large-scale datasets and implement advanced data augmentation strategies.
• Managing large-scale datasets and implementing advanced augmentation strategies.
• Designing and maintaining CI/CD pipelines with GitHub Actions for training, testing, and benchmarking.
• Collaborating with research and engineering to implement cutting-edge techniques and open-source contributions.
• Creating documentation, tutorials, and supporting users of Ultralytics technology.

Skills

Python

Deep learning frameworks

Computer vision

MLOps practices

Problem-solving skills

CUDA

MLOps

GitHub Actions

Docker

Tools

PyTorch

Docker

CUDA

TensorRT

OpenVINO

ONNX

Job description

Overview

Direct message the job poster from Ultralytics

Founder & CEO at Ultralytics | Democratizing Vision AI

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

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.

Job Details

• Seniority level: Associate
• Employment type: Full-time
• Job function: Research and Science
• Industries: Software Development and Information Services