Machine Learning Engineer
Job Description
- Familiarity with Kubernetes and Docker
- Experience in low-level ML optimization (e.g., CUDA kernel code)
- 5+ years experience in developing ML models in JAX, PyTorch, or TensorFlow
- Experience implementing ML foundations (Transformer, VAE, Denoising Diffusion models) from scratch
- Track record in ML innovation and familiarity with Deep Learning
- End-to-end understanding of generative media applications
- Ability to debug ML models to improve quality and performance
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 Ideogram 3.0 and Canvas 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, 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.
