Deloitte
SpaceXSI
Ingeniero de Software - Kernels/CUDA (C++)
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 lo hacemos. Hoy en día, SpaceX desarrolla activamente las tecnologías para hacer esto posible, con el objetivo final de hacer que la vida sea multiplanetaria y expandir la conciencia hacia las estrellas.
SpaceXSI está construyendo una Superinteligencia para ayudar a la humanidad a comprender el universo. Nuestros equipos están construyendo toda la pila de Superinteligencia: desde nuestros superordenadores Colossus hasta los modelos Grok y nuestro conjunto de aplicaciones, que incluye Grok Bot, Grok Build, Grok Imagine, Grok Voice y X. Estamos creando un potente volante de inercia donde más capacidad de computación entrena mejores modelos, mejores modelos impulsan productos más útiles, y lo que aprendemos de las personas que utilizan esos productos da forma a lo que construiremos a continuación.
SOBRE EL PUESTO
Estamos construyendo uno de los superordenadores de IA más grandes del mundo desde cero. Como parte del equipo de Infraestructura de Computación, serás responsable tanto del superordenador de GPU en bruto como de la capa de plataforma que se ejecuta sobre él. Trabajarás en todo el full stack —desde optimizaciones de kernels de GPU de bajo nivel y los aspectos internos del kernel de Linux hasta la 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 puesto amplio y de gran impacto que combina un trabajo intenso de supercomputación e infraestructura de computación. 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 de GPU masivos para cargas de trabajo de entrenamiento e inferencia a escala extrema
- Desarrollar y ajustar kernels de CUDA de bajo nivel (GeMM, Attention, etc.), utilizando CUTLASS, Tensor Cores y Nsight para obtener el máximo rendimiento
- Perfilar, depurar y eliminar cuellos de botella en la jerarquía de memoria de la GPU, el tejido de red, 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 una escalabilidad de nivel de producción
HABILIDADES Y EXPERIENCIA PREFERIDAS
- Profundos conocimientos de programación de sistemas de bajo nivel (C/C++/PTX/SASS)
- Sólida experiencia con clústeres de GPU a gran escala o infraestructura de computación distribuida a escala de producción
- Trabajo práctico con optimización de kernels de GPU (CUTLASS, kernels personalizados, perfilado con 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 los primeros principios y optimizar tanto para escenarios limitados por la memoria como por la computación
COMPENSACIÓN Y BENEFICIOS
180,000 - 440,000 USD
El salario base es solo una parte de nuestro paquete de compensación total en SpaceXSI, que también incluye capital (equity), cobertura médica, de visión y dental integral, acceso a un plan de jubilación 401(k), seguro de invalidez a corto y largo plazo, seguro de vida y otros diversos descuentos y beneficios.
SpaceXSI es un empleador que ofrece igualdad de oportunidades. Para obtener detalles sobre el procesamiento de datos, consulta nuestro Aviso de Privacidad de Selección de Personal x.ai/legal/recruitment-privacy-notice.
