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

Lightly AG; Zürich; Hybrid; CHF 90’000 - 130’000; Vollzeit

Zusammenfassung

A leading machine learning company in Zurich is seeking a Machine Learning Engineer. You will work on self-supervised learning packages and leverage state-of-the-art technology in deep learning. Ideal candidates have an MSc or PhD in a related field, strong Python skills, and experience in deep learning. An attractive cash and equity package is offered along with the possibility of remote work.

Leistungen

Attractive cash & equity package

Remote work possible

Qualifikationen

• Experience in deep learning and computer vision, ideally in self-supervised learning.
• Excellent Python coding skills and knowledge of software engineering best practices.
• Experience with testing and typed Python is a big plus.

Aufgaben

• Work on open-source Python package for self-supervised learning.
• Add support for finetuning pretrained models for downstream tasks.
• Keep up with current research in self-supervised learning and multi-modal models.

Kenntnisse

Deep learning

Computer vision

PyTorch

Python

Software engineering best practices

Ausbildung

MSc or PhD in Computer Science, Engineering, or a related field

Tools

Docker

Kubernetes

Google Cloud Platform

AWS

Jobbeschreibung

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. 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 part-owner 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 open-source package for self-supervised learning and similar internal frameworks.

Particularly, you will get to :

Work on our open-source Python package for self-supervised learning : https : / / github.com / lightly-ai / lightly .

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 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.

In general, for this role expect to use the stack we use for our open-source package https : / / github.com / lightly-ai / lightly .

Requirements

Ideally, you have :

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.

Benefits

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

Interested? Contact us!

We look forward to getting to know you!

Lightly AG is an official ETH Spin-off in the field of machine learning. We're funded by YCombinator and top investors.