Machine Learning Engineer

Key Skills

AWS, AWS Sagemaker, English Proficiency, Google Cloud Platform (GCP), HuggingFace, Japanese, Jupyter Notebook, Large Language Models - LLMs, Linux Operating system, Machine learning techniques, PyTorch

Job Description

About Stability:

Stability AI is a community and mission driven, open-source artificial intelligence company that cares deeply about real-world implications and applications. Our most considerable advances grow from our diversity in working across multiple teams and disciplines. We are unafraid to go against established norms and explore creativity. We are motivated to generate breakthrough ideas and convert them into tangible solutions. Our vibrant communities consist of experts, leaders and partners across the globe who are developing cutting-edge open AI models for Image, Language, Audio, Video, 3D and Biology.

About the role:

We are looking for a versatile ML Engineer who will train and deploy generative models for Japanese partners, clients, and the broader community. You will adapt quickly as we try various approaches to various industries in a fast-changing environment. We will focus especially on image/video models, large language models, and chatbots.

You will have access to state-of-the-art high-performance computing resources and you will be able to work alongside top researchers and engineers to truly make an impact in the fast-growing world of generative AI.

Responsibilities:

• Lead efforts to drive the design, development, and productionization of ML models and systems, and present the solutions to partners and clients in Japan
• Work on the commercial side - productionizing generative models and building the infrastructure to serve them at scale; collaborate with other engineers and researchers to build customized models for commercial applications
• Be a strategic thought partner both internally and externally on driving business impact through machine learning
• Conduct experiments on e.g. fine-tuning image generation models or LLMs for the Japan market
• Prototype and productionize model architecture improvements and new features
• Provide technical advice to partners/clients on generative models

Qualifications:

• Good communication skills with fluency in Japanese and business-level English proficiency
• 5+ years working on machine learning projects, including training, fine tuning and refining models
• Familiarity with recent, important papers and projects in the generative machine learning space
• Experience with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks
• Experience with the open-source ML ecosystem (HuggingFace, W&B, etc.)
• Experience with training and/or deploying ML models with Amazon AWS (Sagemaker a plus) or Google Cloud
• Experience with Linux and command line tools

Equal Employment Opportunity:

We are an equal opportunity employer and do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability or other legally protected statuses.