Senior Diffusion AI Scientist — Scalable Generative Models

RadixArk; Palo Alto (CA); Hybrid; USD 180,000 - 280,000; Full time

Benefits offered by this job

Competitive compensation

Meaningful equity

Comprehensive benefits

Flexible work arrangements

Job summary

A leading AI infrastructure company in California is seeking a Member of Technical Staff for Diffusion Model. This role involves designing cutting-edge generative models and collaborating with teams to optimize large-scale training processes. Candidates should have over 5 years of experience in machine learning, particularly with diffusion models, and proficiency in tools like PyTorch or JAX. This position offers competitive compensation, flexible work arrangements, and the opportunity to contribute to innovative generative AI systems.

Qualifications

• 5+ years of experience in ML research or applied ML engineering.
• Strong expertise in diffusion and generative models.
• Proven experience training large-scale models on GPUs/TPUs.

Responsibilities

• Design and develop next-generation diffusion and generative models.
• Collaborate with teams to scale training and inference.
• Translate research ideas into practical production systems.

Skills

ML research experience

Expertise in diffusion models

Proficiency in PyTorch or JAX

Mathematical foundation in probability

Experience training large-scale models

Job description

A leading AI infrastructure company in California is seeking a Member of Technical Staff for Diffusion Model. This role involves designing cutting-edge generative models and collaborating with teams to optimize large-scale training processes. Candidates should have over 5 years of experience in machine learning, particularly with diffusion models, and proficiency in tools like PyTorch or JAX. This position offers competitive compensation, flexible work arrangements, and the opportunity to contribute to innovative generative AI systems.