Member of Technical Staff - Training Platform
Prime Intellect · Remote · San Francisco ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Fully remote or SF
- Kubernetes + Python
- Visa sponsorship
- AI infrastructure
- $150k-$300k
You will help build the hosted training product that allows users to launch fine-tuning runs on managed GPU clusters via API or UI. The position spans both the developer-facing platform and the underlying Kubernetes-based infrastructure that orchestrates these jobs.
The team works on unifying compute, environments, and evaluations into a full-stack system for post-training at scale. You will interface with research teams to productize new training capabilities and model architectures.
What you'll do
- Design and operate Kubernetes orchestration across multi-cloud GPU fleets
- Build Helm charts for reproducible training stacks and services
- Develop Python control-plane agents to monitor pods and sync clusters
- Implement scheduling and autoscaling for heterogeneous hardware
- Build FastAPI backend services and REST APIs for the platform
- Ship product UI features using Next.js, React, and TypeScript
- Create real-time monitoring tools for logs, metrics, and debugging
- Interface with RL trainers and inference servers inside clusters
What you'll need
- Strong knowledge of modern AI stack and finetuning techniques
- Familiarity with GPU hardware tradeoffs and memory hierarchy
- Understanding of distributed training fundamentals and NCCL
- Strong Kubernetes operations experience with Helm and operators
- Comfortable debugging production clusters and node issues
- Cloud platform experience, preferably GCP and GKE
- Strong Python backend development skills with FastAPI
- Modern frontend development skills with React and TypeScript
Stack
- Kubernetes
- Python
- FastAPI
- React
- Next.js
- TypeScript
- Helm
- Prometheus
- Grafana
- GCP
How you'll work
- Flexible work arrangement available
- Remote work or San Francisco office
- Full visa sponsorship provided
- Relocation support available
What you get
- Cash compensation between $150K and $300K
- Significant equity package included
- Professional development budget for courses
- Regular team off-sites and conference attendance
- Opportunity to contribute to open-source AI