In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger AI and Compute Foundations team. MTIA Software Team has been developing a comprehensive software platform strategy that delivers a highly flexible platform to train & serve new deep learning model architectures, combined with auto-tuned high performance for production environments across specialized hardware architectures.
You will be working on one of the core areas such as PyTorch & GenAI framework components, AI compiler and runtime, high-performance deep learning kernels and tooling to accelerate machine learning workloads on the current & next generation of MTIA AI hardware platforms. You will work closely with AI researchers to analyze deep learning models and optimize them efficiently on MTIA hardware. You will also partner with hardware design teams to develop optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning techniques. You will gain valuable experience in developing machine learning compiler frameworks and will help in driving next generation hardware software codesign for AI domain specific problems.
À ce poste, vous ferez partie de l'équipe logicielle MTIA (Meta Training & Inference Accelerator) et intégrerez l'équipe plus large chargée de l'IA et des fondements informatiques. L'équipe logicielle de MTIA élabore actuellement une stratégie globale de plate-forme logicielle visant à offrir une plate-forme extrêmement flexible permettant de former et d'exploiter de nouvelles architectures de modèles d'apprentissage profond, tout en garantissant des performances optimisées automatiquement pour les environnements de production sur diverses architectures matérielles spécialisées.
Show the full description
Vous travaillerez sur l'un des domaines clés, tels que les composants des frameworks PyTorch et GenAI, le compilateur et le moteur d'exécution d'IA, les noyaux d'apprentissage profond haute performance et les outils destinés à accélérer les charges de travail d'apprentissage automatique sur les plates-formes matérielles d'IA MTIA actuelles et de nouvelle génération. Vous travaillerez en étroite collaboration avec des chercheurs en IA pour analyser les modèles d’apprentissage profond et les optimiser efficacement sur le matériel MTIA. Vous travaillerez également en collaboration avec les équipes de conception matérielle afin de mettre au point des optimisations visant à atteindre des performances élevées. Vous mettrez en œuvre les meilleures pratiques en matière de développement logiciel pour concevoir des fonctionnalités, optimiser les applications et mettre en œuvre des techniques d'optimisation des performances. Vous bénéficierez d'une expérience précieuse dans le développement de compilateurs d'apprentissage automatique et contribuerez à la conception de logiciels matériels de nouvelle génération pour les problèmes spécifiques du domaine d'IA.
