Radixark
Member of Technical Staff — Diffusion Model
Palo Alto, CA · Staff+
Sponsorship not specifiedDetected 8 hours ago
AlgorithmsGitAWSMachine LearningDeep LearningPyTorchNLPLLMsStatisticsResearch
About the role
- RadixArk is seeking a Member of Technical Staff - Diffusion Model to advance the frontier of generative modeling.
- You will work on cutting-edge diffusion and flow-based models for image, video, and multimodal generation, pushing model quality, efficiency, and scalability.
- This role combines deep research thinking with strong engineering execution - from designing novel algorithms to training and deploying models at scale.
Responsibilities
- Design and develop next-generation diffusion and generative models
- Research and implement novel training and sampling methods
- Optimize models for large-scale distributed training
- Collaborate with systems teams to scale training and inference
- Our team has optimized kernels serving billions of tokens daily, designed distributed training systems coordinating 10,000+ GPUs, and contributed to infrastructure that powers leading AI companies and research labs.
- We're backed by well-known infrastructure investors and partner with Nvidia, Google, AWS, and frontier AI labs.
- Join us in building infrastructure that gives real leverage back to the AI community.
Requirements
- 5+ years of experience in ML research or applied ML engineering
- Strong expertise in diffusion models or generative models (DDPM, DDIM, latent diffusion, flow matching, etc.)
- Proven experience training large-scale models on GPUs/TPUs
- Strong proficiency in PyTorch or JAX
- Experience implementing research ideas into working systems
- Ability to move from research prototypes to production-quality models
- Experience with large-scale distributed training
- Experience in multimodal generation (text-to-image, video, audio)
- Experience improving sampling speed and generation efficiency
- Experience scaling models to billions of parameters
Compensation
- We offer competitive compensation with meaningful equity, comprehensive benefits, and flexible work arrangements.
- Compensation depends on location, experience, and level.
Benefits
- We offer competitive compensation with meaningful equity, comprehensive benefits, and flexible work arrangements.
Company info
- We're on a mission to democratize frontier-level AI infrastructure by building world-class open systems for inference and training.
Equal opportunity
- RadixArk is an Equal Opportunity Employer and is proud to offer equal employment opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.
Visa & Work Authorization
- t opportunity to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more
Apply directly at Radixark →Create a free account for alerts like thisView Radixark immigration profile
This listing is sourced directly from Radixark's careers page and normalized into a canonical job model.