Skip to content
mimi

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

Lightly AG

Remote (Global) Senior 5d ago

About the role

About

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.

Team

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.

Role

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

Stack

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

Responsibilities

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.

  • Work on our open‑source Python package for self‑supervised learning: https://github.com/lightly-ai/lightly.
  • Add support for fine‑tuning pre‑trained 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.
  • Work 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.
  • 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

  • Attractive cash‑& equity package to become part‑owner of Lightly.
  • Remote work is also possible.

Contact

Interested? Contact us!

We look forward to getting to know you!

#J-18808-Ljbffr

Requirements

  • 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 building and shipping software solutions.
  • Strong passion for new technologies and interest in keeping up with the latest technologies and publications in deep learning.

Responsibilities

  • Work on our open-source Python package for self-supervised learning.
  • Add support for fine-tuning pre-trained 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.
  • 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.

Benefits

cash packageequity package

Skills

AWSC++DockerGoogle Cloud PlatformKubernetesNodeJSNumpyPyTorchPythonReactRustTypescript

Don't send a generic resume

Paste this job description into Mimi and get a resume tailored to exactly what the hiring team is looking for.

Get started free