About Stability:
Stability AI is a community and mission driven, open 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, Code, Audio, Video, and 3D, including models specifically made for Japan/Japanese.
About the role:
We are looking for a versatile ML Engineer who will train and deploy generative models for use cases in Japan. You will adapt quickly as we try various approaches to various industries in a fast-changing environment. We will focus especially on image models, language models, and multimodal models.
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:
- Research and development of generative AI models and systems
- 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
- Work on projects related to data and annotation for our models
- Be a strategic thought partner both internally and externally on driving impact through machine learning
Qualifications:
- Good communication skills with fluency in Japanese and business-level English proficiency
- 6+ years working on machine learning and 4+ years working on deep learning projects, including training / fine-tuning models
- Familiarity with recent, important papers and projects in the generative AI space
- 1+ years of hands-on engineering or research work related to generative AI
- Understanding of evaluations of ML models, especially LLMs
- Experience with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks
- Experience with gathering/cleaning/labeling data for ML models
- Experience with the open ML ecosystem (HuggingFace, W&B, etc.)
- Experience with Linux and command line tools
- Experience with training and/or deploying ML models with Amazon AWS or Google Cloud is a plus
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.