Research Scientist or Engineer

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.

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 and evaluation pipelines.

• Develop the post‑training pipeline for text‑to‑image foundation models from data strategy to deployment.
• Advance techniques such as RLHF, RLAIF, personalization and customization.
• Contribute to post‑training research that drives measurable gains.
• Implement and maintain high‑throughput fine‑tune and evaluation pipelines.

• 5+ years of experience developing machine learning models in JAX, PyTorch, or Tensor Flow.
• 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.
• Experience in low‑level machine learning optimization, e.g., writing CUDA kernel code.

• Competitive compensation and equity.
• 4 weeks of vacation.
• Comprehensive health, vision, and dental coverage.
• Daily in‑office lunches and dinners.
• Access to compute resources needed for large‑scale experiments and research.