Spellbrush · Tokyo; San Francisco
AI researcher role designing and developing large-scale diffusion models, primarily for images with secondary work on 3D, text, video, and audio. Requires proficiency in PyTorch, with JAX also used for large-scale TPU training. Models move from research into production for niji・journey.
The Research Role:
We're looking for an experienced AI Researcher to join our team and help us push the boundaries of AI and anime. You'll design and develop large-scale diffusion models with a primary focus on images, and secondary focus on other modalities like 3d, text, video and audio.
You may be a good fit if:
This one of the only labs in the world where you can combine your expertise on anime and deep learning to work on state-of-the-art anime models.
We're looking for experienced researchers who aren't afraid of jumping headfirst into experimenting and designing massive models.
We have a very large Pytorch codebase, and an equally large JAX codebase for large-scale TPU training.
We pride ourselves on getting models from the lab into production quickly. Our research team works closely with our internal studio to both develop game-ready models, and deploy regular quality updates to niji・journey.
We currently have a team of four on AI. You'll be working closely alongside some of the best AI researchers in the world, on the literal best image model in the world.
We also believe in the unmatched speed of in-person teams, and prefer on-site collaboration in either our primary research office in Tokyo (downtown Akihabara), or San Francisco. Visa sponsorships are available.