Machine Learning Engineer

Ideogram’s mission is to help people become more creative. Our thesis is that everyone has an innate desire to create. We are developing state-of-the-art AI tools that will make creative expression more accessible and efficient. We are pushing the limits of what’s possible with AI, with a focus on creativity and a high standard for trust and safety. Our headquarters is in downtown Toronto, and we have a small presence in NYC. Read Ideogram 2.0 blog post and check out our text to image product at ideogram.ai to get a glimpse of what we're building.

About the role

As a Machine Learning Engineer, you will build and deploy state of the art machine learning models at scale. You will work with a creative and ambitious team of engineers and researchers who are building the future of the creative economy with highly complex challenges. You will work directly with some of the best machine learning experts in the world, who have an exceptional track record of setting the vision and executing generative AI projects.

What we're looking for

• 2+ years of experience in developing machine learning models in JAX, PyTorch, or TensorFlow

• Experience in implementing Machine Learning foundations (e.g., Transformer, VAE, Denoising Diffusion models) from scratch

• Track record in machine learning innovation and familiarity with Deep Learning and advanced Machine Learning

• End-to-end understanding of generative media applications and excitement for pushing the state-of-the-art in generative AI

• Ability to debug machine learning models to iteratively improve model quality and performance

• Nice to have: Familiarity with kubernetes and docker

• Optional: Experience in low-level machine learning optimization, e.g., writing CUDA kernel code

Founding team

Our founding team consists of world-renowned AI experts including Mohammad Norouzi, Jonathan Ho, William Chan, and Chitwan Saharia. This team has previously led transformative AI projects at Google Brain, UC Berkeley, CMU, and the University of Toronto. Our fundamental work in AI includes: Imagen: Google’s text-to-image system, Imagen Video for video synthesis, Denoising Diffusion Models, which is the foundation of the recent generative media transformation.

Company Culture

We're a single flat team that transcends engineering, research, product, and operation roles. Everyone is willing to do whatever is necessary to make our company and customers successful. We believe that a small, dedicated team with a collaborative culture can move faster and build better and more coherent products than large hierarchical organizations. At Ideogram, we provide mentorship and support to help our employees grow with the company and achieve their ambitious career goals.

Ideogram is committed to welcoming everyone, regardless of gender identity, orientation, or expression. Our mission is to remove exclusivity and barriers and encourage new thinking and perceptions, in a space of belonging. It is not about race, gender, or age, it is about the people.