Lead efforts to drive the design, development, prototype, and production of ML systems that address 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 fine-tuning, dataset engineering, tooling, open-source efforts, etc.
Work on the commercial side, productionizing generative model-based solutions, by partnering with platform and research teams, building the infrastructure to serve them at scale
Write production-quality code and influence the next generation of Stability AI’s machine learning solutions
Advocate for engineering best practices and upskill the team
Requirements & Skills:
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 skills 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 top-tier machine learning conferences
