Member of Technical Staff - Training Platform
Prime Intellect · San Francisco, CA · 2 mo ago
On-siteEngineering$150k–$300k/yrFull-time
Core Technical Responsibilities
- Hosted Training Infrastructure Design and operation of Kubernetes-based training and inference orchestration across multi-cluster, multi-cloud GPU fleets
- Build and maintain Helm charts that compose trainers, inference servers, environment servers, and supporting services into reproducible "Training stacks"
- Develop the Python control-plane agents that watch pods, report run state to the platform, and keep clusters in sync
- Implement scheduling and autoscaling for heterogeneous hardware (H100/H200/B200) using KEDA, LeaderWorkerSet, taints/tolerations, and gang scheduling
- Run a tight GitOps workflow - every change ships through PRs, Helm values, and CI
- Build node-local model caches, checkpoint pipelines, and shared storage for fast cold starts
- Operate the observability stack (Prometheus, Grafana, Loki, DCGM) and make GPU cluster debugging fast
Platform Development
- Build the developer-facing surfaces for hosted training: job submission, live run monitoring, logs, metrics, model/adapter management, comparisons
- Develop FastAPI backend services and REST APIs that bridge the platform to running clusters
- Build real-time monitoring and debugging tools (streaming logs, step-level metrics, failure analysis)
- Ship product UI in Next.js / React / TypeScript with shadcn, Tailwind, tRPC, and TanStack Query
Technical Requirements
- AI & GPU Landscape: Strong working knowledge of the modern AI stack - open model families, finetuning techniques (LoRA, QLoRA, full FT, RLHF/RLAIF), inference engines (vLLM, SGLang, TensorRT-LLM)
- Familiarity with GPU hardware tradeoffs (H100 / H200 / B200, NVLink, interconnects, memory hierarchy) and what they mean for training and inference workloads
- Understanding of distributed training fundamentals (data/tensor/pipeline/expert parallelism, NCCL, multi-node scheduling)
- Awareness of what's happening at the frontier - new models, training methods, infra patterns - and the ability to translate that into product decisions
- Kubernetes & Infrastructure: Kubernetes operations experience - Helm, CRDs, operators, KEDA, gang scheduling, GPU operator
- Comfortable debugging real production clusters (kubectl, pod lifecycle, node issues, networking)
- Cloud platform experience (GCP preferred - GCS, GKE, Cloud Run, Cloud Tasks)
- Infrastructure automation (Helm, Terraform, Ansible) and a GitOps mindset
- Observability: Prometheus, Grafana, Loki, OpenTelemetry, DCGM
- Linux fundamentals: networking, namespaces, performance tuning
- Programming & Platform: Strong Python backend development (FastAPI, async, SQLAlchemy)
- Comfortable building Python control-plane agents that talk to Kubernetes APIs
- Modern frontend development (TypeScript, React/Next.js, Tailwind, shadcn) - enough to ship product surfaces end-to-end
- REST and tRPC API design
- Experience building developer tools, dashboards, and live-monitoring UIs