Member of Technical Staff — Inference-Multimodal & Diffusion

RadixArk • Palo Alto, CA

17h

About The Position

RadixArk is seeking a Member of Technical Staff to advance generative modeling, with a focus on inference, multimodal systems, and diffusion. The role works on diffusion and flow-based models for image, video, and multimodal generation, improving model quality, efficiency, and scalability. It combines research with engineering execution, including designing algorithms and training and deploying models at scale. The work will contribute to generative AI systems for researchers, developers, and real-world applications.

Requirements

• At least 5 years of experience in ML research or applied ML engineering.
• Strong expertise in diffusion models or generative models, such as DDPM, DDIM, latent diffusion, or flow matching.
• Deep understanding of deep learning fundamentals and optimization.
• Proven experience training large-scale models on GPUs or TPUs.
• Strong proficiency in PyTorch or JAX.
• Experience implementing research ideas into working systems.
• Strong mathematical foundation in probability, statistics, and optimization.
• Ability to move from research prototypes to production-quality models.

Nice To Haves

• Publications in top-tier conferences such as NeurIPS, ICML, ICLR, or CVPR.
• Experience with large-scale distributed training.
• Experience in multimodal generation, including text-to-image, video, or audio.
• Familiarity with transformer architectures and hybrid models.
• Experience improving sampling speed and generation efficiency.
• Contributions to open-source generative model projects.
• Experience scaling models to billions of parameters.

Responsibilities

• Design and develop next-generation diffusion and generative models.
• Improve model quality, controllability, and sample efficiency.
• Research and implement novel training and sampling methods.
• Optimize models for large-scale distributed training.
• Collaborate with systems teams to scale training and inference.
• Translate research ideas into practical production systems.
• Evaluate models using rigorous metrics and benchmarks.
• Contribute to long-term research and product direction in generative AI.

Benefits

• Equity

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed