Machine Learning Engineer

Ideogram is on a mission to make professional-quality design accessible to everyone. We’re building an AI-native application that helps creative individuals and teams produce high-quality content faster and at a lower cost. We are pushing the boundaries 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 growing presence in NYC. Read our 3.0, Canvas and Character blog posts, and try our 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

• 5+ 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, William Chan, and Chitwan Saharia. This team has previously led transformative AI projects at Google Brain, CMU, and the University of Toronto. Our fundamental work in generative media includes Imagen and Imagen Video: Google’s text-to-image and text-to-video systems.

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.