Staff Software Engineer, GPU Infrastructure (HPC)

As a Staff Software Engineer in GPU Infrastructure (HPC) at Cohere, you will design and operate ML-optimized, Kubernetes-based GPU/TPU superclusters across multiple clouds, focusing on stability, scalability, and performance for AI workloads. Your responsibilities include optimizing infrastructure with cloud providers, troubleshooting complex issues, and enabling researchers with self-service tools to manage their AI training jobs. You will collaborate closely with AI researchers to innovate and implement solutions that enhance machine learning infrastructure while championing best practices like automation and observability.