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.
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
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