Research Engineer / Research Scientist (Post-training)

TL;DR

Research Engineer / Research Scientist (AI): Developing post-training pipelines for text-to-image foundation models from data strategy to deployment with an accent on advancing techniques like RLHF and RLAIF. Focus on implementing and maintaining high-throughput fine-tune/eval pipelines and pushing the state-of-the-art in generative AI.

Location: On-site in Toronto and NYC

Company

hirify.global builds proprietary generative media models and AI native creative workflows, tackling unsolved challenges in graphic design, backed by Andreessen Horowitz and Index Ventures.

What you will do

• Develop end-to-end post-training pipelines for text-to-image foundation models.
• Advance techniques such as RLHF, RLAIF, personalization, and customization.
• Contribute to post-training research to drive measurable gains.
• Implement and maintain high-throughput fine-tune/eval pipelines.
• Work with a creative and ambitious team building the future of the creative economy.

Requirements

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

Culture & Benefits

• Competitive compensation and equity.
• 4 weeks of vacation.
• Comprehensive health, vision, and dental coverage starting on day one.
• RRSP/401(k) with employer match up to 4%.
• Top-of-the-line tools and tech.
• Daily in-office lunches and dinners at Toronto HQ.
• Autonomy to explore and experiment with access to compute/resources.
• Culture of learning, growth, curiosity, and mentorship.