Generative AI Research Engineer — Remote & Equity

Ideogram; Toronto; Hybrid; CAD 100,000 - 130,000; Full time

Job summary

A cutting-edge AI company is seeking an experienced machine learning engineer to develop post-training pipelines for text-to-image models. The role involves working with advanced techniques and contributing to research that drives improvements. Ideal candidates will have extensive experience in developing ML models and familiarity with generative media applications. This position offers competitive compensation, flexibility, and a strong commitment to team collaboration and mentorship.

Benefits

Competitive compensation and equity

4 weeks of vacation

Comprehensive health, vision, and dental coverage

RRSP/401(k) with employer match

Top-of-the-line tools and tech

Team lunches and dinners

Autonomy to explore

Culture of learning and growth

Fully remote flexibility with in-person meetups

Qualifications

• 5+ years of experience in developing machine learning models.
• Experience in implementing Machine Learning foundations from scratch.
• Familiarity with deep learning and advanced machine learning.

Responsibilities

• Develop the post-training pipeline for text-to-image models.
• Contribute to post-training research for measurable gains.
• Implement and maintain high-throughput fine-tune/eval pipelines.

Skills

Developing machine learning models

Implementation of machine learning foundations

Machine learning innovation

Understanding of generative media applications

Debugging machine learning models

Familiarity with Kubernetes

CUDA kernel code optimization

Tools

JAX

PyTorch

TensorFlow

Docker

Job description

A cutting-edge AI company is seeking an experienced machine learning engineer to develop post-training pipelines for text-to-image models. The role involves working with advanced techniques and contributing to research that drives improvements. Ideal candidates will have extensive experience in developing ML models and familiarity with generative media applications. This position offers competitive compensation, flexibility, and a strong commitment to team collaboration and mentorship.