The MITRE Corporation
Mindbeam
Ingénieur en pré-entraînement Machine Learning
À propos de Mindbeam
Nous construisons l'infrastructure IA de nouvelle génération pour l'open source et les entreprises. Notre travail est profondément axé sur la recherche et passionné par le développement d'innovations révolutionnaires pour faire passer les applications d'IA de pointe au niveau supérieur.
Ce qui nous anime, ce n'est pas seulement de faire progresser la technologie, mais aussi de responsabiliser les personnes qui la façonnent. Nous sommes une communauté de chercheurs, d'ingénieurs et de visionnaires qui croient que la collaboration, la curiosité et l'ouverture alimentent le progrès. Si vous êtes motivé par l'impact et inspiré à construire des outils sur lesquels d'autres peuvent s'appuyer, vous serez au bon endroit.
Mission
Concevoir et optimiser des systèmes de pré-entraînement à grande échelle qui alimentent les modèles d'IA générative de Mindbeam.
Attentes du rôle
- Construire des pipelines de pré-entraînement évolutifs pour les modèles fondamentaux, en optimisant le débit et l'efficacité.
- Implémenter des stratégies d'entraînement distribué sur des clusters GPU/TPU et haute performance.
- Collaborer avec les chercheurs pour traduire les configurations expérimentales en flux de travail prêts pour la production.
- Développer des systèmes de surveillance et de tolérance aux pannes pour garantir un entraînement fiable à grande échelle.
- Benchmarker et ajuster continuellement les performances sur les piles matérielles et logicielles.
Profil
- Licence, Master ou Doctorat en informatique, ingénierie ou domaine connexe, ou expérience équivalente.
- 2+ années d'expérience dans l'entraînement de modèles à grande échelle et les systèmes distribués.
- Solides compétences en codage en Python et familiarité avec les frameworks de ML (PyTorch, TensorFlow, JAX).
- Expérience avec la planification GPU, l'optimisation de la mémoire et les stratégies de parallélisme.
- À l'aise avec les environnements conteneurisés et orchestrés (Docker/Kubernetes).
- Compréhension du calcul haute performance et des goulots d'étranglement réseau.
À propos de vous
Vous prospérez dans l'échelle et la complexité. Vous aimez résoudre les goulots d'étranglement au niveau du système, pousser le matériel et le logiciel à leurs limites, et travailler en étroite collaboration avec les chercheurs pour accélérer le développement de l'IA de pointe.
