Member of Technical Staff — Inference-Multimodal & Diffusion
RadixArkAI Infrastructure company
Palo Alto, United States$200,000 - $400,000 USD + equityLead
Accel
Spark Capital
NVentures
Salience Capital
A&E Investments
HOF Capital
Data & AI
About the role
TL;DR
Advance generative modeling with diffusion and flow-based models for multimodal generation.
- •RadixArk is seeking a Member of Technical Staff
- •Inference-Multimodal & Diffusion 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.
- •Key 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 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.) Deep understanding of deep learning fundamentals and optimization Proven experience training large-scale models on GPUs/TPUs Strong proficiency in PyTorch or JAX
Required skills
PythonPyTorchJAXComputer VisionLLMs
Nice-to-have skills
TensorFlowscikit-learnHugging Face
Domain expertise
ai
Benefits & perks
equity
Tech stack
PythonPyTorchJAXTensorFlowscikit-learnComputer VisionLLMs