Research Engineer / Research Scientist (Post-training)
Company:Ideogram
Location:North America
Languages:English
Seniority level:Senior, 5+ years
Experience:5+ years
Skills:
DockerKubernetesMachine LearningPyTorchTensorflow
Requirements:
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. Ability to debug machine learning models to iteratively improve model quality and performance. Familiarity with kubernetes and docker (nice to have). Experience in low-level machine learning optimization, e.g., writing CUDA kernel code (optional).
Responsibilities:
Develop the post-training pipeline for text-to-image foundation models end to end. Advance techniques such as RLHF and RLAIF. Work on personalization and customization of models. Implement and maintain high-throughput fine-tune/eval pipelines.
