Member of Technical Staff — Developer Technology
About the Role
Key Responsibilities
-
Accelerate AI workloads. Profile and optimize GPU performance for real production workloads on current and next-generation hardware, root-causing bottlenecks from kernels to distributed multi-node systems.
-
Go deep in one or two focus areas. The team collectively covers the full stack; each engineer specializes in one or two tracks:
-
Inference performance: engine tuning, benchmarking, long-context and multi-turn optimization, parallelism strategy, production debugging
-
Kernels and model/hardware enablement: custom CUDA/ROCm/Triton kernels, low-precision quantization, day-0 support for new models on new silicon
-
Speculative decoding: draft-model training, acceptance-rate tuning, cross-platform kernel adaptation
-
Training systems: RL post-training with Miles, FP8 training, elasticity, long-rollout and long-context efficiency
-
-
Partner directly with the ecosystem. Turn ambiguous, high-stakes problems from expert engineers at our key partners into concrete wins, clear technical guidance, and reproducible cookbooks.
-
Enhance SGLang and Miles. Feed user-driven improvements back into our open-source systems and roadmap, so every win compounds across the ecosystem.
Qualifications
-
4+ years of experience in GPU systems, LLM infrastructure, or performance engineering.
-
Strong profiling and debugging skills: able to root-cause performance and correctness issues across the stack.
-
Hands-on GPU programming experience in at least one of CUDA, ROCm, or Triton, and willingness to work across platforms.
-
Strong programming skills in Python plus C++ or CUDA.
-
Comfortable making progress on hard, ambiguous problems with little context to start from, and fast to ramp into unfamiliar systems, codebases, and domains.
-
Ability to translate ambiguous asks into clear technical plans, verified cookbooks, and actionable recommendations, and to communicate credibly with expert engineering audiences.
-
Deep familiarity with LLM inference internals: distributed serving, parallelism, routing, KV-cache management, scheduling.
-
Experience with low-precision quantization and inference/training (FP8, INT8/INT4; NVFP4 or MXFP4 a strong plus).
-
Experience writing and optimizing custom GPU kernels.
-
Practical familiarity with speculative decoding methods such as Eagle, DFlash, or DSpark.
-
Working knowledge of large-scale distributed training: pre-training, SFT, RL post-training, elasticity, long-context workloads.
-
Experience optimizing across both NVIDIA and AMD platforms.
-
Hands-on experience with SGLang, Miles, vLLM, TensorRT-LLM, Megatron, or comparable frameworks; contributions to open-source AI/ML projects.
About RadixArk
RadixArk is an infrastructure-first company built by engineers who've shipped production AI systems, created SGLang (20K+ GitHub stars, the fastest open LLM serving engine), and developed Miles (our large-scale RL framework). 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. 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.
Compensation
We offer competitive compensation with meaningful equity, comprehensive benefits, and flexible work arrangements. Compensation depends on location, experience, and level.
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.
Apply for this job
*
indicates a required field