Machine Learning Engineer

As a machine learning engineer, you will build the machine learning models which power PicCollage’s creative and visual applications, using the latest in computer vision and generative AI technology. You will work with PMs, designers and other ML engineers to solve exciting problems and build new features which leverage the latest breakthroughs in AI. Specifically, you will work closely with engineers to train and accelerate models, and build novel solutions involving technologies like transformers and diffusion models.

Responsibilities

• Design and implement machine learning models for visually creative applications, including computer vision and generative AI.
• Keep up to date with the latest advancements in machine learning, and apply new technologies to improve our products.
• Collaborate with other engineers and cross-functional members to identify new challenges/opportunities and translate open problems into specific technical problems to work on.

Qualifications

• Experience in training or finetuning deep learning models, preferably in computer vision
• A deep and clear understanding of transformers, self- and cross-attention, ViTs.
• Strong grasp on a deep learning framework like PyTorch, Tensorflow or JAX (we use PyTorch here).
• A MS degree or higher in a related field (machine learning, computer science, etc), or equivalent experience.
• Ability to delve deep into topics and have a fundamental understanding of key concepts in ML. We prefer deep understanding of key concepts over having a shallow knowledge of a wide range of topics. Example of delving deep: you are working with CLIP model and want to visualize the attention maps for each transformer block to understand what’s going on.
• Good communication skills in English.
• Strong knowledge of linear algebra and probability.

• Experience in generative AI, particularly text to image models, is a great plus. Bonus points if you have experience training adapters (LoRA, ControlNet, IP-Adapter etc) for Stable Diffusion models.
• Has strong analytical skills
• Has has an artistic side
• Has experience in computer vision and/or generative AI.
• Published papers or research work related to ML.
• Familiarity with diffusers and diffusion models.

You should still consider applying even if you don’t have a lot of experience with diffusion models. A great candidate is someone who learns machine learning deeply, not only superficially. A candidate who developed deep expertise with any relevant ML topic will fit better than a candidate who only has limited understanding of diffusion.

Flexible Work Time & Remote Work 彈性上下班時間 & 遠端工作

Maker Time 每週一天 Maker Time，不進行不必要的會議和討論

Team Activities and Workshops 各式活動與學習機會 (技術分享、Design Demo, Hack Days等）

Fun, Creative & International 雙語工作環境，團隊成員來自世界各地

Free Lunch and Dinner in the Office 辦公室內團隊午餐晚餐

Sports Classes in Office/Online 和同事一起起來動一動

PicCafe 健康零食水果，永遠餵飽你的胃

Awesome & Fun Office 有趣舒適大稻埕風的辦公室