Machine Learning Team Lead

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&DDemonstrate 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 positiveBuild a robust framework to seamlessly apply these models to live video camera feedsTraining 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.