The MITRE Corporation
Mindbeam
Ingeniero de Machine Learning - Pre-entrenamiento
Sobre Mindbeam
Estamos construyendo la infraestructura de IA de próxima generación para código abierto y empresas. Nuestro trabajo está profundamente orientado a la investigación y nos apasiona desarrollar innovaciones pioneras para llevar las aplicaciones de IA de vanguardia al siguiente nivel.
Lo que nos impulsa no es solo avanzar en la tecnología, sino empoderar a las personas detrás de ella. Somos una comunidad de investigadores, ingenieros y visionarios que creen que la colaboración, la curiosidad y la apertura impulsan el progreso. Si te motiva el impacto y te inspira crear herramientas sobre las que otros puedan construir, estarás en el lugar adecuado.
Misión
Diseñar y optimizar sistemas de pre-entrenamiento a gran escala que potencien los modelos de IA generativa de Mindbeam.
Expectativas del puesto
- Construir pipelines de pre-entrenamiento escalables para modelos fundacionales, optimizando el rendimiento y la eficiencia.
- Implementar estrategias de entrenamiento distribuido en GPUs/TPUs y clústeres de alto rendimiento.
- Colaborar con investigadores para traducir configuraciones experimentales en flujos de trabajo listos para producción.
- Desarrollar sistemas de monitorización y tolerancia a fallos para garantizar un entrenamiento fiable a gran escala.
- Evaluar y ajustar continuamente el rendimiento en pilas de hardware y software.
Experiencia
- Grado, Máster o Doctorado en Ciencias de la Computación, Ingeniería o campo relacionado, o experiencia equivalente.
- 2+ años de experiencia con entrenamiento de modelos a gran escala y sistemas distribuidos.
- Sólidas habilidades de codificación en Python y familiaridad con frameworks de ML (PyTorch, TensorFlow, JAX).
- Experiencia con planificación de GPU, optimización de memoria y estrategias de paralelismo.
- Comodidad con entornos contenerizados y orquestados (Docker/Kubernetes).
- Comprensión de la computación de alto rendimiento y los cuellos de botella de red.
Sobre ti
Te desenvuelves bien con la escala y la complejidad. Disfrutas resolviendo cuellos de botella a nivel de sistema, llevando el hardware y el software a sus límites, y trabajando estrechamente con los investigadores para acelerar el desarrollo de IA de vanguardia.
