Machine Learning Engineer; OSS in AI Startup

Machine Learning Engineer (OSS) in a fast-growing AI Startup

Join to apply for the Machine Learning Engineer (OSS) role at Lightly.

At the ETH spin‑off 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.

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.

• PyTorch, Python, Typescript, C++, Rust, React, Node.js
• Docker, Kubernetes, Google Cloud Platform, AWS

• Work on our open‑source Python package for self‑supervised learning:
• Add support for fine‑tuning pretrained models for downstream tasks including many different computer vision tasks like object detection, segmentation, tracking, and more.
• Keep up with current research in self‑supervised learning and multi‑modal models.
• All these tasks will include working with state‑of‑the‑art technology such as self‑supervised 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.

• MSc or PhD in Computer Science, Engineering, or a related field.
• Experience in deep learning and computer vision, ideally in self‑supervised 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.

We offer an attractive cash‑and‑equity package for you to become a part‑owner of Lightly. Remote work is also possible.

Interested? Contact us! We look forward to getting to know you.

Job :

Updated: 1 week ago