Machine Learning Software Engineer, Plumerai

• Strong software engineering skills and proficiency in Python
• Very comfortable with frameworks such as TensorFlow and Keras (which we use), PyTorch or JAX

Desirable

• We are looking for a Machine Learning Software Engineer to help us improve our training algorithms, data pipeline and model architectures
• Your work has a major impact on our roadmap of new AI features and your code will be shipped to a very large number of camera devices in the field
• Together we are building the most advanced AI for embedded devices
• Our deep learning models are truly tiny - the memory footprint of our computer vision models is sometimes . You will train and design more accurate models, while also enabling new and more complex AI applications on low-cost and low-power hardware
• You will improve our data pipeline, model architectures and training software. Sometimes there is relevant literature available, but novel approaches and clever hacks are often required for the problems that we are working on
• You will use and improve our large data warehouse that allows us to train our tiny deep learning models. Our automated data ingestion pipeline feeds unlabelled images to large auto labeling models, processes our large datasets (30 million images and videos), and generates datasets for model training
• You will use our Kubernetes cluster to deploy TensorFlow training jobs, Weights & Biases to analyze results, Snowflake and Dataflow to build datasets, large multi-modal models for yet to be announced features, Streamlit to prototype new demos, and lots of GPUs on GCP for training new models and auto-labeling data
• Our software is deployed in the field on a very large number of camera devices from major consumer brands. You will ship new features fast, use feedback from users to make our AI models ever more accurate, and make our developer friendly
• We encourage you to take initiatives and take on many responsibilities. You will be part of a small team where everybody is expected to make a big impact on our products and the success of the company. Career advancement opportunities are likely as the company grows