
Member of Technical Staff - Compute Platform
Prime Intellect2 months ago
Responsibilities
- Build web interfaces for AI workload management and monitoring.
- Develop Python REST APIs, backend services, automation scripts, and platform support tools.
- Create real-time monitoring and debugging tools and user-facing resource-management and job-control features.
- Design and implement distributed training infrastructure, high-performance networking, and coordination components in Rust.
- Build infrastructure automation pipelines and manage cloud resources and Kubernetes container orchestration.
- Implement scheduling systems for heterogeneous CPU, GPU, and TPU hardware.
- Collaborate on backend features and integrate them into existing infrastructure while maintaining reliability and security.
Requirements
- Strong Python backend development experience, including FastAPI and asynchronous programming.
- Modern frontend development experience with TypeScript, React or Next.js, and Tailwind.
- Experience building developer tools, dashboards, RESTful APIs, and backend services.
- Systems programming experience with Rust.
- Experience with Ansible, Terraform, Kubernetes, and cloud platforms, with GCP preferred.
- Experience with observability tools such as Prometheus and Grafana.
- Experience with GPU computing, ML infrastructure, AI/ML model architecture and training, high-performance networking, open-source infrastructure, or WebSocket and real-time systems is preferred.
Benefits
- Opportunity to work with experienced engineers and researchers on frontier AI infrastructure.
- Encouragement to contribute to the broader AI community through research and open-source contributions.
Tech Stack
AnsibleFastAPIGoogle Cloud PlatformGrafanaKubernetesNext.jsPrometheusPythonReactRustTailwind CSSTerraformTypeScript