Research Engineer / Research Scientist (Post-training) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Ideogram · 1 month ago

Research Engineer / Research Scientist (Post-training)

Ideogram is a company dedicated to making world-class design accessible through generative media models and AI workflows. The role involves developing the post-training pipeline for text-to-image models, contributing to research, and implementing high-throughput pipelines in a collaborative environment.

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

Responsibilities

Develop the post-training pipeline for text-to-image foundation models end to end, from data strategy to deployment
Advance 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

Machine Learning modelsJAXPyTorchTensorFlowGenerative media applicationsDeep 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 team lunches every Tuesday and dinners every Thursday.
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.
Fully remote flexibility across North America, with regular in-person team meetups and collaboration opportunities.

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