Research Engineer / Research Scientist (Post-training)
On-site · New York City, New York, United States or Toronto, Ontario, Canada
New York City, New York, United States or Toronto, Ontario, CanadaOn-siteFull TimeMid LevelNot SpecifiedAI SoftwareStartup
Type
Full Time
Level
Mid Level
Education
Not Specified
Company size
Startup
Industry
AI Software
Job Summary
Develop the post-training pipeline for text-to-image foundation models end to end, from data strategy to deployment, advancing techniques such as RLHF and RLAIF, and working on personalization/customization. Contribute to post-training research that drives measurable gains and implement high-throughput fine-tune/eval pipelines. Collaborate with a creative and ambitious team of engineers and researchers shaping the future of the creative economy.
Required Qualifications
- 5+ years of experience developing machine learning models in JAX, PyTorch, or TensorFlow
- Experience implementing ML foundations (e.g., Transformer, VAE, Diffusion models) from scratch
- Track record in machine learning innovation and familiarity with deep learning and advanced ML
- End-to-end understanding of generative media applications and excitement for advancing state-of-the-art in generative AI
- Ability to debug ML models to iteratively improve quality and performance
- Nice to have: Kubernetes and Docker familiarity
- Optional: CUDA kernel code experience
Apply with one swipe on Sorce. We auto-fill applications and apply on your behalf — no cover letters, no 40-minute forms.
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.