Estados Unidos
CSC Leasing Co.
SpaceXSI
Ingeniero/a de Software - Kernels/CUDA (C++)
Estados Unidos
Palo Alto, California, Estados Unidos · Seattle, Washington, Estados Unidos
C++
Traducción automática. Consulta el original.
Sobre la oportunidad
SpaceX se fundó bajo la creencia de que un futuro en el que la humanidad explora las estrellas es fundamentalmente más emocionante que uno en el que no. Hoy, SpaceX está desarrollando activamente las tecnologías para hacer esto posible, con el objetivo final de hacer la vida multiplanetaria y expandir la conciencia a las estrellas.
SpaceXSI está construyendo Super Inteligencia para ayudar a la humanidad a comprender el universo. Nuestros equipos están construyendo la pila completa de Super Inteligencia: desde nuestros superordenadores Colossus, hasta los modelos Grok, pasando por nuestra suite de aplicaciones, que incluyen Grok Bot, Grok Build, Grok Imagine, Grok Voice y X. Estamos creando un potente ciclo en el que más cómputo entrena mejores modelos, mejores modelos potencian productos más útiles y lo que aprendemos de las personas que usan esos productos da forma a lo que construimos a continuación.
SOBRE EL ROL
Estamos construyendo uno de los superordenadores de IA más grandes del mundo desde cero. Como parte del equipo de Infraestructura de Cómputo, serás responsable tanto del superordenador GPU bruto como de la capa de plataforma que se ejecuta sobre él. Trabajarás en todo el full stack — desde optimizaciones de kernels GPU de bajo nivel e internos de kernels Linux hasta orquestación y virtualización a gran escala — para hacer que el entrenamiento y la inferencia en SpaceXSI sean lo más rápidos, fiables y escalables posible.
Este es un rol amplio y de gran impacto que combina trabajo duro de supercómputo e infraestructura de cómputo. Tus contribuciones acelerarán directamente la velocidad de entrenamiento de Grok y el progreso general de la IA.
RESPONSABILIDADES
- Diseñar, construir y optimizar clústeres masivos de GPU para cargas de trabajo de entrenamiento e inferencia a escala extrema
- Desarrollar y ajustar kernels CUDA de bajo nivel (GeMM, Attention, etc.), utilizando CUTLASS, Tensor Cores y Nsight para un rendimiento máximo
- Perfilar, depurar y eliminar cuellos de botella en la jerarquía de memoria de la GPU, la red de interconexión, los sistemas de archivos y la operación multi-GPU
- Colaborar estrechamente con los equipos de investigación de IA para ofrecer un rendimiento y escalabilidad de nivel de producción
HABILIDADES Y EXPERIENCIA PREFERIDAS
- Programación profunda de sistemas de bajo nivel (C/C++/PTX/SASS)
- Sólida experiencia con clústeres de GPU a gran escala o infraestructura de cómputo distribuido a escala de producción
- Experiencia práctica con optimización de kernels de GPU (CUTLASS, kernels personalizados, perfilado Nsight)
- Historial de construcción o ejecución de infraestructura de alto rendimiento para cargas de trabajo de IA (plataformas de entrenamiento o inferencia)
- Capacidad para razonar desde principios fundamentales y optimizar tanto para escenarios limitados por memoria como por cómputo
COMPENSACIÓN Y BENEFICIOS
$180,000 - $440,000 USD
El salario base es solo una parte de nuestro paquete total de recompensas en SpaceXSI, que también incluye acciones, cobertura médica, de visión y dental integral, acceso a un plan de jubilación 401(k), seguro de incapacidad a corto y largo plazo, seguro de vida y varios otros descuentos y beneficios.
SpaceXSI es un empleador que ofrece igualdad de oportunidades. Para obtener detalles sobre el procesamiento de datos, consulte nuestro Aviso de Privacidad de Reclutamiento x.ai/legal/recruitment-privacy-notice.
Más oportunidades con un solo perfil
¿Quieres optar a más puestos sin repetir todo el proceso? Crea tu perfil para que te encuentren las empresas que usan Nort.
Crear mi perfil