Machine Learning Engineer
NYC
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.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: CUDA Deep Learning Diffusion models Docker Engineering Generative AI Imagen JAX Kubernetes Machine Learning ML models PyTorch Research TensorFlow
Perks/benefits: Career development
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.