← All roles

Member of Technical Staff — Inference-Multimodal & Diffusion

RadixArk· Member of Technical Staff
Apply on company site
Palo Alto, CA$200k - $400kfull-timePosted 5mo ago

About the role

Role focused on advancing diffusion and flow-based generative models for image, video, and multimodal generation, combining research with engineering to train and deploy models at scale.

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

Qualifications

  • 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
  • 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

Skills

Required:Diffusion models or generative modelsDeep learning fundamentals and optimizationGPU/TPU training at scalePyTorch or JAX

Benefits

  • Equity (as part of compensation)
Last updated Aug 18, 2026. Interested? Apply on the company site.
RadixArk logo

About RadixArk

SHIP AI FOR ALL

RadixArk is an infrastructure-first, deep-tech company building large-scale inference and training systems for the entire AI community.

AI systemsMilesSGLangbessemer-ai-infrastructure-2026inferenceinference-inflection-pointinfrastructureopen infrastructure
Company profileWebsite

More open roles at RadixArk