Research Engineer / Research Scientist (Post-training) jobs in United States
info-icon
This job has closed.
company-logo

Ideogram · 2 months ago

Research Engineer / Research Scientist (Post-training)

Ideogram is a company focused on making world-class design accessible through generative media models and AI workflows. In this role, you will develop the post-training pipeline for text-to-image foundation models and contribute to research that drives measurable gains in generative AI applications.

Artificial Intelligence (AI)Generative AIInformation TechnologyInternet
check
H1B Sponsor Likelynote

Responsibilities

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
Contribute to post-training research that drives measurable gains
Implement and maintain high-throughput fine-tune/eval pipelines

Qualification

JAXPyTorchTensorFlowMachine Learning modelsGenerative AIDeep LearningDebugging modelsKubernetesDockerCUDA optimization

Required

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

Preferred

Familiarity with kubernetes and docker
Experience in low-level machine learning optimization, e.g., writing CUDA kernel code

Benefits

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.

Company

Ideogram

twittertwitter
company-logo
Ideogram is an artificial intelligence startup that provides generative text-to-image technology.

H1B Sponsorship

Ideogram has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)

Funding

Current Stage
Early Stage
Total Funding
$96.46M
Key Investors
Andreessen Horowitz
2024-02-28Series A· $80M
2023-08-22Seed· $16.46M

Leadership Team

leader-logo
Mohmmad Norouzi
Chief Executive Officer
linkedin
Company data provided by crunchbase