Research Engineer / Research Scientist (Post-training)

SupportFinity™; New York (NY); On-site; USD 150,000 - 190,000; Full time

Benefits offered by this job

Competitive compensation & equity

4 weeks vacation

Health, vision, and dental coverage

401(k) with employer match

Top-of-the-line tools & tech

Job summary

Ideogram, a pioneering AI design company in New York, seeks a Research Engineer/Scientist to advance post-training pipelines for text-to-image foundation models. You will push RLHF, RLAIF, and personalization work, building high-throughput fine-tune/eval pipelines with a collaborative, mentorship-driven team.

Ideal candidates have 5+ years in ML model development with JAX/PyTorch/TensorFlow, broad ML foundations, and a passion for state-of-the-art generative AI.

Qualifications

• 5+ years of experience developing machine learning models in JAX, PyTorch, or TensorFlow.
• Experience implementing ML foundations (e.g., Transformer, VAE, Denoising Diffusion models) from scratch.
• Track record in machine learning innovation and familiarity with Deep Learning and advanced ML.
• 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.

Skills

ML model development

Transformer models

VAE knowledge

DDM knowledge

Kubernetes

Docker

Debug ML models

End-to-end ML

JAX

PyTorch

TensorFlow

CUDA

Tools

JAX

PyTorch

TensorFlow

CUDA

Job description

Research Engineer / Research Scientist (Post-training)

Ideogram |

Ideogram | Posted Mar 9

Full-time

New York

Negotiable

Advanced (5-10 yrs)

About Ideogram

Ideogram's mission is to make world-class design accessible to everyone, multiplying human creativity. We build proprietary generative media models and AI native creative workflows, tackling unsolved challenges in graphic design. Our team includes builders with a track record of technology breakthroughs including early research in Diffusion Models, Google's Imagen, and Imagen Video. We care about design, taste, and craft as much as research and engineering - shipping experiences that creatives actually love.

We've raised nearly $100M, led by Andreessen Horowitz and Index Ventures. Headquartered in Toronto with a growing team in NYC, we're scaling fast, aiming to triple over the next year. We're a flat team with a culture of high ownership, collaboration, and mentorship.

Explore Ideogram 3.0, Canvas, and Character blog posts, and try Ideogram at ideogram.ai.

The Opportunity

In this role, you will develop the post-training pipeline for our text-to-image foundation models end to end, from data strategy to deployment, advancing techniques such as RLHF, RLAIF, and work on personalization/customization. You will contribute to post-training research that drives measurable gains, and implement and maintain high-throughput fine-tune/eval pipelines. You'll work with a creative and ambitious team of engineers and researchers who are building the future of the creative economy.

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.

Our Culture

We're a team of exceptionally talented, curious builders who love solving tough problems and turning bold ideas into reality. We move fast, collaborate deeply, and operate without unnecessary hierarchy, because we believe the best ideas can come from anyone.

Everyone at Ideogram rolls up their sleeves to make our products and our customers successful. We thrive on curiosity, creativity, and shared ownership. We believe that small, dedicated teams working together with trust and purpose can move faster, think bigger, and create amazing things.

Ideogram is committed to welcoming everyone — regardless of gender identity, orientation, or expression. Our mission is to create belonging and remove barriers so everyone can create boldly.

What We Offer

• Competitive compensation and equity designed to recognize the value and impact of your contributions to Ideogram's success.
• 4 weeks of vacation to recharge and explore.
• Comprehensive health, vision, and dental coverage starting on day one.
• RRSP/401(k) with employer match up to 4% to invest in your future from the moment you join.
• Top-of-the-line tools and tech to fuel your creativity and productivity.
• Toronto HQ perks: Steps from Union Station and the PATH, with daily in-office lunches and dinners.
• Autonomy to explore and experiment - whether you're testing new ideas, running large-scale experiments, or diving into research, you'll have access to compute/resources you need when there's a clear business or creative use case. We encourage curiosity and bold thinking.
• A culture of learning and growth, where curiosity is encouraged and mentorship is part of the journey.

About the company

Ideogram