top of page

Machine Learning Engineer

Tokyo, Japan

Full time


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

Please let the company know you found this position on Jobdai to support us!

bottom of page