Machine Learning Engineer in a fast-growing AI Startup

At the ETH spinoff Lightly we build software solutions to help companies improve their deep learning models. The available data for deep learning grows faster than human tagging and available compute power. Our customers already face this problem and have too much data to process. As a solution we built software to identify and work with the most relevant samples only. Learn more about us on our website.

We are a team of talented developers and researchers in Zurich Prague and San Francisco and are backed by YCombinator. Every member of the team is going to have a massive impact.

Thus we are looking for an experienced engineer with lots of passion for the subject. We offer attractive cash & equity package for you to become partowner of Lightly. Remote work is also possible.

We use a modern stack of:

• PyTorch Python Typescript C Rust React NodeJS
• Docker Kubernetes Google Cloud Platform AWS

Tasks

As a Machine Learning Engineer in the core ML team you will be part of a team which works mainly on our opensource package for selfsupervised learning and similar internal frameworks.

Particularly you will get to:

• Work on our opensource Python package for selfsupervised learning:
• Add support for finetuning pretrained models for downstream tasks including many different computer vision tasks like object detection segmentation tracking and more.
• Keep up with current research in selfsupervised learning and multimodal models.
• All these tasks will include working with stateoftheart technology such as selfsupervised learning model distillation vision language models and deep learning for computer vision in general.
• Participate in and hold paper sessions about the latest papers in the field.
• Represent the company at machine learning conferences and engage with the community.

In general for this role expect to use the stack we use for our opensource package

Requirements

Ideally you have:

• MSc or PhD in Computer Science Engineering or a related field
• Experience in deep learning and computer vision ideally in selfsupervised learning and you have worked with PyTorch.
• Excellent Python coding skills and knowledge of software engineering best practices. Experience with testing and typed Python is a big plus.
• Experience with building and shipping software solutions.
• Strong passion for new technologies and interest in keeping up with the latest technologies and publications in deep learning.

Benefits

We offer attractive cash & equity package for you to become partowner of Lightly. Remote work is also possible.

Interested Contact us!

We look forward to getting to know you!