720,454open jobs
42,966companies
101,308added this week
Browse all
Salary
$200k – $400k per year
Location
Remote/Hybrid (Palo Alto, United States)
Seniority
Staff · 5+ years exp

Confirmed on the employer's own hiring board on Sep 24, 2026. First seen by Alion on Feb 17, 2026. RadixArk scores C on the Alion truth index.

Overview
Company
Impact
Profile match
RadixArk builds large-scale inference and training systems for the entire AI community, making frontier-level AI infrastructure open and accessible.

About the Role

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.

Your work will directly shape next-generation generative AI systems used by researchers, developers, and real-world applications.

This is a high-impact role for engineers and researchers who want to push the limits of generative models in both theory and practice.

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

  • 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

Strong Plus

  • Publications in top-tier conferences (NeurIPS, ICML, ICLR, CVPR, etc.)

  • Experience with large-scale distributed training

  • Experience in multimodal generation (text-to-image, video, 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

 

About RadixArk

RadixArk is an infrastructure-first company built by engineers who've shipped production AI systems, created SGLang (30K+ GitHub stars, the fastest open LLM serving engine), and developed Miles (our large-scale RL framework). Founded by AI infrastructure veterans from xAI and NVIDIA, we're on a mission to democratize frontier-level AI infrastructure by building world-class open systems for inference and training. 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.

Compensation

Depending on background, skills, and experience, the expected annual salary range for this position is $200,000 - $400,000 USD + equity.

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.

Free account
Stop reading job ads. Get the ones that fit.
One free account turns this page into a shortlist built around your stack, your level and your pay.
Match on every job. Stack, seniority, pay and location, scored against your profile.
720,454 open roles. Read straight off company career pages, refreshed every day.
Unlimited applications. Every one you send is tracked in one place, on-site or on a company board.
3 tailored CVs a month. Rewritten for the exact job you are applying to. Included free.
Create a free account Continue with Google
Free forever. No card. Under a minute.

Your match

How well do you fit this role?
Two answers are enough for a real match. No account needed.
Check my fit
Answers stay in this browser until you create an account.

Recommended for you based on this role

Similar stack
Same company
Palo Alto
$64k – $170k per year (Estimated) • In office • Full-Time • 4+ years exp • Bachelor's Degree • Paris
Python
SQL
Databases
Snowflake
Neo4j
Databricks
Amazon Neptune
AI/ML
LangGraph
AutoGen
LangChain
DSPy
LlamaIndex
Prefect
Embeddings
JAX
Prompt Engineering
AI Agents
Haystack
ONNX
Pydantic AI
Semantic Kernel
AgentOps
Mistral
TensorFlow
PyTorch
CrewAI
Gemini
LLM
RAG
OpenAI
Anthropic
LLMOps
GraphRAG
Human-in-the-Loop
Machine Learning
DevOps
GCP
Azure
CI/CD
AWS
Chips/EDA
PoC Library
Apply
Staff ML Scientist 11 hours ago
$49k – $109k per year (Estimated) • In office • 7+ years exp • Bachelor's Degree • Bengaluru
Python
Python
pySpark
Databases
FAISS
AI/ML
Spark
Triton Inference Server
Embeddings
Scikit-learn
Multimodal AI
AI Agents
Transformers
TensorFlow
Pandas
NumPy
PyTorch
LLM
RAG
Recommender Systems
Machine Learning
Analytics
A/B Testing
Apply
$122k – $318k per year (Estimated) • Remote • Full-Time • Bachelor's Degree
Python
C++
C++
PyTorch C++
AI/ML
Multimodal AI
Computer Vision
VLM
PyTorch
Machine Learning
Apply
Senior Staff Engineer 11 hours ago
$48k – $101k per year (Estimated) • In office • 7+ years exp • Bachelor's Degree • Bengaluru
Python
Python
pySpark
Databases
FAISS
AI/ML
Spark
Triton Inference Server
Embeddings
Scikit-learn
Multimodal AI
AI Agents
Transformers
TensorFlow
Pandas
NumPy
PyTorch
LLM
RAG
Recommender Systems
Machine Learning
Analytics
A/B Testing
Apply
In office • Full-Time • Master's Degree • Wilmington
Python
SQL
C++
C++
TensorFlow C++
PyTorch C++
Databases
Snowflake
AI/ML
OpenCV
Computer Vision
ONNX
TensorFlow
PyTorch
Vision-Language-Action
Machine Learning
DevOps
Rest API
Azure
AWS
Docker
Linux
Robotics
ROS
SLAM
Localization
Sensor Fusion
Perception
robot_localization
Apply
$208k – $426k per year (Estimated) • In office • 4+ years exp • Bachelor's Degree • Palo Alto
AI/ML
SGLang
LLM
DevOps
Rest API
GCP
Azure
CI/CD
AWS
Kubernetes
GitHub
IAM
Linux
Unix
Cybersecurity
OWASP Top 10
SOC 2
HIPAA
Threat Modeling
SIEM
Apply
$200k – $400k per year • In office • 4+ years exp • Bachelor's Degree • Palo Alto
Python
Go
Rust
AI/ML
SGLang
LLM
DevOps
Rest API
gRPC
GCP
Datadog
Prometheus
Azure
CI/CD
AWS
Kubernetes
Grafana
Platform Engineering
GitHub
Apply
$160k – $300k per year • In office • 10+ years exp • Palo Alto
AI/ML
SGLang
LLM
DevOps
GitHub
Apply
Product Manager 7 days ago
$160k – $300k per year • Equity • In office • 3+ years exp • Bachelor's Degree • Palo Alto
Python
C++
AI/ML
SGLang
LLM
OpenAI
DevOps
GitHub
Apply
$160k – $300k per year • In office • 4+ years exp • Bachelor's Degree • Palo Alto
Python
C++
AI/ML
CUDA Toolkit
SGLang
LLM
CUDA
TPU
AWS Trainium
DevOps
CI/CD
AWS
GitHub
Apply
$143k – $301k per year (Estimated) • Remote/Hybrid • Full-Time • 12+ years exp • Bachelor's Degree • Palo Alto
Apex
Apex
MuleSoft
DevOps
SLI/SLO/SLA
Apply
$52k – $143k per year (Estimated) • In office • High School Diploma • Palo Alto
Apply
$59k – $114k per year (Estimated) • In office • 1+ year exp • Palo Alto
Apply
Chief of Staff 7 hours ago
$120k – $150k per year • Equity 0.4–0.7% • In office • Full-Time • 3+ years exp • Palo Alto
AI/ML
AI Agents
Management
Stripe
Apply
$48k – $96k per year • In office • Internship • Palo Alto
Apply
See all jobs
This is one of many
720,454 more open roles from verified company boards, updated every day.