San Francisco, CAHybrid$310,000 - $510,0001 months ago
full-timeprincipalllamacustom
About the Role
Join Codegen's Diffusion team to push the boundaries of AI research. You'll work on cutting-edge problems in diffusion and contribute to foundational advances in the field.
Key responsibilities:
- Conduct original research in diffusion
- Design and run experiments at scale
- Publish at top-tier venues (NeurIPS, ICML, ICLR)
- Collaborate with engineering teams to productionize research
- Mentor junior researchers and interns
Requirements
- 10+ years of experience in software engineering or ML
- Strong programming skills in Python, JAX, RLHF
- Experience with machine learning frameworks and tools
- Track record of delivering complex projects
- Experience mentoring and leading teams
- Strong communication and collaboration skills
- Publications in top venues preferred
- Industry recognition and thought leadership
- Experience setting technical strategy