Machine Learning Team Leader

Robbie AI

Robbie AI

Software Engineering

Spain

Posted 6+ months ago

Job Description

Please note, this is an Individual Contributor (IC) role.

Robbie AI is seeking a highly skilled and motivated Machine Learning Team Leader with

expertise in computer vision to join our dynamic team. As the team leader, you will play a pivotal

role in designing, building, and deploying cutting-edge self-learning applications focused on

body pose movement, facial expression recognition, basic emotions detection, gender

detection, and more. For this, being fluent in reading and writing papers associated with

state-of-the-art computer vision and ANN/RNN architectures is desirable.

Responsibilities:

Project Leadership:

● Manage and lead ML in-house teams projects like external labelers and outsourcing of

R&D

● Demonstrate expertise in building Computer Vision projects from scratch, focusing on

real-time processing rather than relying solely on off-the-shelf models.

Data Preprocessing and Architecture:

● Design and develop efficient code for preprocessing and storing image-based datasets

using custom-built ANN architectures.

● Utilize your skills to translate literature into practical, real-world products.

State-of-the-Art Models Implementation:

● Demonstrate a strong ability to read and implement cutting-edge deep learning models

from research papers related to body pose movement, face detection, basic emotions

detection, gender and age detection, and face recognition.

● Any experience with LLM and ViT Transformers applied to computer vision is very

positive

● Build a robust framework to seamlessly apply these models to live video camera feeds

Training and Deployment:

● Develop and streamline model training and deployment processes, focusing on

automation and efficiency.

● Optimize deep neural networks and associated preprocessing/postprocessing code to

ensure smooth operation on servers, edge devices, and specific hardware.

Project Management:

● Utilize your project management skills to guide the ML team towards achieving goals

and delivering successful outcomes.

● Foster a collaborative and innovative environment, promoting knowledge sharing and

continuous learning.

Required Qualifications:

● Proven experience using data science tools, including proficient Python scripting, numpy,

scipy, matplotlib, scikit-learn, jupyter notebooks, bash scripting, and working in a Linux

environment.

● Demonstrated ability to manage a team of at least 5 engineers, ensuring effective

collaboration and successful project outcomes.

● Experience working with custom hardware platforms such as Jetson/ARM.

● Proficiency in deploying and delivering highly optimized inference models at a production

level.

● Strong programming skills in Python3, Bash, Git, and experience with popular deep

learning frameworks such as PyTorch or TensorFlow.

● Solid knowledge of libraries and frameworks like Numpy, OpenCV, CUDA, and

TensorRT.

● Familiarity with C/C++ programming languages for optimizing performance.

● Desirable experience with Inference servers like Nvidia Triton, KubeML, and other

pipeline frameworks.

● Previous experience working in teams following SCRUM/Kanban methodologies.

● Excellent communication skills in English, enabling effective collaboration and reporting.

● Desirable exposure to scientific papers, demonstrating an ability to stay informed about

the latest advancements in the field.

Desired Qualifications:

● Experience working with Google Cloud Platform or Azure services, particularly storage

services and on-demand virtual machines.

● Proficiency in using Make and Makefiles to create targets and streamline build

processes.

● Familiarity with CI/CD (Continuous Integration/Continuous Deployment) tools such as

GitLab Jobs, GitHub Actions, and Travis, to automate software testing and deployment

workflows.

● Expertise in encoding/decoding h265 video feeds and knowledge of codecs and

transports like HSL (HTTP Live Streaming).

● Strong sense of teamwork and the ability to collaborate effectively with colleagues from

diverse backgrounds.

● Leadership skills to motivate and empower a team of developers, fostering their growth

and improving their capabilities.