États-Unis
CSC Leasing Co.
SpaceXSI
Ingénieur Logiciel - Noyaux/CUDA (C++)
États-Unis
Palo Alto, California, États-Unis · Seattle, Washington, États-Unis
C++
Traduction automatique. Consultez l’original.
À propos du poste
SpaceX a été fondée sur la conviction qu'un avenir où l'humanité explore les étoiles est fondamentalement plus passionnant qu'un avenir où elle ne le fait pas. Aujourd'hui, SpaceX développe activement les technologies pour rendre cela possible, avec l'objectif ultime de rendre la vie multiplanétaire et d'étendre la conscience aux étoiles.
SpaceXSI construit une Super Intelligence pour aider l'humanité à comprendre l'univers. Nos équipes construisent la pile complète de Super Intelligence : de nos supercalculateurs Colossus, aux modèles Grok, en passant par notre suite d'applications, y compris Grok Bot, Grok Build, Grok Imagine, Grok Voice et X. Nous créons un puissant cercle vertueux où plus de calcul entraîne de meilleurs modèles, de meilleurs modèles alimentent des produits plus utiles, et ce que nous apprenons des personnes qui utilisent ces produits façonne ce que nous construisons ensuite.
À PROPOS DU POSTE
Nous construisons l'un des plus grands supercalculateurs d'IA au monde à partir de zéro. En tant que membre de l'équipe d'infrastructure de calcul, vous serez responsable à la fois du supercalculateur GPU brut et de la couche plateforme qui s'exécute par-dessus. Vous travaillerez sur l'ensemble du full stack — des optimisations de noyaux GPU de bas niveau et des internes du noyau Linux à l'orchestration et à la virtualisation à grande échelle — pour rendre l'entraînement et l'inférence chez SpaceXSI aussi rapides, fiables et évolutifs que possible.
Il s'agit d'un rôle large et à fort impact qui combine un travail acharné sur les supercalculateurs et l'infrastructure de calcul. Vos contributions accéléreront directement la vitesse d'entraînement de Grok et le progrès global de l'IA.
RESPONSABILITÉS
- Concevoir, construire et optimiser des clusters GPU massifs pour des charges de travail d'entraînement et d'inférence à échelle extrême
- Développer et optimiser les noyaux CUDA de bas niveau (GeMM, Attention, etc.), en utilisant CUTLASS, Tensor Cores et Nsight pour des performances maximales
- Profiler, déboguer et éliminer les goulots d'étranglement dans la hiérarchie mémoire GPU, le tissu réseau, les systèmes de fichiers et les opérations multi-GPU
- Collaborer étroitement avec les équipes de recherche en IA pour fournir des performances et une évolutivité de qualité production
COMPÉTENCES ET EXPÉRIENCES SOUHAITÉES
- Programmation système approfondie de bas niveau (C/C++/PTX/SASS)
- Solide expérience avec des clusters GPU à grande échelle ou une infrastructure de calcul distribuée à l'échelle de production
- Expérience pratique avec l'optimisation des noyaux GPU (CUTLASS, noyaux personnalisés, profilage Nsight)
- Expérience avérée dans la construction ou l'exploitation d'infrastructures haute performance pour les charges de travail d'IA (plateformes d'entraînement ou d'inférence)
- Capacité à raisonner à partir de principes fondamentaux et à optimiser pour les scénarios liés à la mémoire et au calcul
RÉMUNÉRATION ET AVANTAGES
$180,000 - $440,000 USD
Le salaire de base n'est qu'une partie de notre programme de rémunération globale chez SpaceXSI, qui comprend également des actions, une couverture médicale, de vision et dentaire complète, l'accès à un plan de retraite 401(k), une assurance invalidité courte et longue durée, une assurance-vie, ainsi que diverses autres réductions et avantages.
SpaceXSI est un employeur qui souscrit au principe de l'égalité des chances. Pour plus de détails sur le traitement des données, consultez notre Avis de confidentialité de recrutement x.ai/legal/recruitment-privacy-notice.
Plus d’opportunités avec un seul profil
Vous souhaitez accéder à plus de postes sans répéter tout le processus ? Créez votre profil pour être trouvé par les entreprises utilisant Nort.
Créer mon profil