ML Engineer

Role Mission:
We are looking for a senior Machine Learning Engineer who can build and improve ML models used in our product. Most of the tasks are related to computer vision and image processing. You will work on both common tasks like segmentation, detection, and classification, as well as more complex problems involving Diffusion Models, LLMs, VLMs, and other custom tasks without ready-made solutions.
You should be able to work independently, move fast, and quickly test new ideas.

Responsibilities:
Build and improve ML models, mostly for computer vision tasks.
Handle the full ML process: collect and prepare data, train and test models, and improve them step by step.
Work on both standard and new types of problems.
Create fast prototypes and help turn successful ones into real products.
Work closely with other teams to bring your models into the product.

Requirements:
4+ years of real-world experience with ML, including at least 2 years in computer vision.
Good understanding of how to build ML systems from start to finish.
Hands-on experience with one or more of these: object detection, segmentation, img2img, generative networks.
Ability to read, understand, and implement ideas from cutting-edge research papers. You stay current with top conferences (e.g., CVPR, NeurIPS, ICCV) and can turn academic innovations into practical solutions.
Strong skills in Python and PyTorch.
Experience optimizing models for efficient inference on local devices (CPU, GPU, NPU), including quantization, pruning, and runtime adaptation.
Able to work with and explore large datasets.
Comfortable working on your own and taking full responsibility for your tasks. Experience with prototyping and working in an R&D cycle.

Nice to Have
Experience in research (PhD, papers, or personal projects).
Experience training or fine-tuning foundational models (LLMs, VLMs).
Experience training or fine-tuning diffusion models for specific image generation or enhancement tasks.
Familiarity with classical image processing techniques.
Experience with deep learning inference frameworks like ONNX Runtime, OpenVINO, Core ML, or others.
Strong C++ skills, close to your Python level.
Enjoy building and testing product ideas quickly.

What we offer:
For personal growth:
A chance to work with a strong team and a unique opportunity to make substantial contributions to our award-winning photo editing tools;
An educational allowance to ensure that your skills stay sharp;
English and German classes to strengthen your capabilities and widen your knowledge.
For comfort:
A great environment where you’ll work with true professionals and amazing colleagues whom you’ll call friends quickly;
The choice of working remotely or in our office space located on Podil, equipped with everything you might need for productive and comfortable work.
For health:
Medical insurance;
Twenty-one days of paid sick leave per year;
Healthy fruit snacks full of vitamins to keep you energized.
For leisure:
Twenty-one days of paid vacation per year;
Fun times at our frequent team-building activities.

What to expect when you apply
An interview with our Talent Acquisition Specialist
Professional interview
Management interview