We’re looking for a Machine learning engineer (Python) ****who is an expert at productionizing, optimizing, and customizing new and existing models.

About:

HeadshotPro is the leading AI headshot provider helping customers around the world get their professional headshots without having to hire a physical photographer.

We’re 100% remote and async. We focus on shipping instead of meeting.

You can read more about the way we work here:

Working at HeadshotPro

Some of your day-to-day tasks:

• Optimizing our existing headshot generation pipeline code to be as photorealistic as possible.
• Experiment with Dreambooth and all variables like user uploads, training steps, etc.
• Implementing new image-gen pipelines like LoRa.
• General code and speed improvements.
• Experimenting with newly published models and how these could be utilized.

Qualities that we’re looking for:

• Extensive knowledge of Python, diffusers, Cog, and Replicate.
• A balance of software engineering and machine learning skills.
• You can squeeze every last drop of performance out of a GPU.
• Ideally, you're involved in the generative AI community and familiar with diffusion models and similar techniques.