< Remote Canada >
About the role:
We are looking for Machine Learning engineers who are passionate about generative models and the creative applications of AI to join our Applied ML team. We look for highly innovative ML engineers with a strong blend of research and engineering skills motivated to push the boundaries of image video and 3D foundation models and their applications. In this role you will collaborate with research and platform teams to build improve and optimize workflows that power stateoftheart application solutions based on diffusion models. You will have access to highperformance computing resources and work alongside top researchers and engineers making a real impact in the rapidly growing field of generative AI.
Responsibilities:
- Lead efforts to drive the design develop prototype and productionize ML systems that addresses the customer needs
- Collaborate with the research team on developing the next generation of models where you may assist with areas such as optimization of model training model finetuning dataset engineering tooling opensource efforts and etc.
- Work on the commercial side productionizing generative modelbased solutions by partnering with platform and research teams building the infrastructure to serve them at scale
- Write productionquality code and influence the next generation of Stability AIs machine learning solutions
- Advocate for engineering best practices and upskill the team
Qualifications:
- MS or PhD in Computer Science
- 1 years of experience working on machine learning projects
- Familiar with diffusion models
- Proficiency with Python scientific stack PyTorch creating Jupyter/Colab notebooks
- Strong coding skill and comfortable with coding in production platforms
- Ability to diagnose technical problems debug code and automate manual routine tasks
- Ability to communicate machine learning concepts and results effectively through writing and visualization
- Track record of publication in toptier machine learning conferences