Stability AI Logo

Stability AI

Senior Machine Learning Engineer, Applied

Posted 14 Days Ago
Remote
2 Locations
Mid level
Remote
2 Locations
Mid level
Lead the design, development, and optimization of ML systems using generative models. Collaborate with research to improve models and productionize solutions.
The summary above was generated by AI

Senior Machine Learning Engineer, Applied - United States or Canada

About the role: 

We are looking for Machine Learning engineers 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 who are 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 state-of-the-art application solutions based on diffusion models. You will have access to high-performance 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 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

Qualifications: 

  • MS or PhD in Computer Science or related technical discipline
  • +3 years working on machine learning projects
  • Experience working 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 top-tier machine learning conferences

Top Skills

Colab
Jupyter
Python
PyTorch

Similar Jobs

2 Hours Ago
Easy Apply
Remote
13 Locations
Easy Apply
Senior level
Senior level
Consumer Web • Enterprise Web • Mobile • Productivity • Software
As a Senior Backend Engineer, you'll design and develop backend systems that support revenue growth, collaborate across teams, and ensure code quality while working with various technologies.
Top Skills: GoGCPKubernetesPostgresRedisStripe
2 Hours Ago
Remote
Hybrid
6 Locations
Mid level
Mid level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
As a UI Engineer, you will build and maintain single-page web applications with Ember.js, collaborate with diverse teams, and drive technical aspects of product delivery.
Top Skills: AWSCSSEmberGitHTMLJavaScriptMochaPostcssQunitSassTailwind Css
5 Hours Ago
Remote
Hybrid
Toronto, ON, CAN
Senior level
Senior level
Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
Design, build, and maintain cloud infrastructures for Motorola Solutions’ platforms. Ensure security best practices and automate processes for scalability and performance in a remote environment.
Top Skills: ArgocdAWSAzureBashCi/CdElastic StackElasticsearchGCPGitGoJavaScriptKubernetesOpentofuPrometheusPythonSQLTerraformThanos

What you need to know about the Vancouver Tech Scene

Raincouver, Vancity, The Big Smoke — Vancouver is known by many names, and in recent years, it has gained a reputation as a growing hub for both tech and sustainability. Renowned for its natural beauty, the city has become a magnet for professionals eager to create environmental solutions, and with an emphasis on clean technology, renewable energy and environmental innovation, it's attracted companies across various industries, all working toward a shared goal: advancing clean technology.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account