Ametek, Inc.
Figure
Ingénieur Staff en inférence et accélération IA
À propos du poste
Figure est une entreprise de robotique IA qui développe des robots humanoïdes autonomes à usage général. L'objectif de l'entreprise est de commercialiser des robots humanoïdes dotés d'une intelligence de niveau humain. Ses robots sont conçus pour effectuer une variété de tâches sur les marchés domestiques et commerciaux. Figure a son siège social à San Jose, en Californie. Figure est une entreprise de robotique IA qui développe des robots humanoïdes autonomes à usage général. L'objectif de l'entreprise est de commercialiser des robots humanoïdes dotés d'une intelligence de niveau humain. Ses robots sont conçus pour effectuer une variété de tâches sur les marchés domestiques et commerciaux. Figure a son siège social à San Jose, en Californie.
Nous recherchons un ingénieur Staff en inférence et accélération IA pour rejoindre l'équipe Logiciel Plateforme et prendre en charge l'architecture d'inférence embarquée pour les robots humanoïdes de Figure. Vous serez l'autorité technique sur la manière dont les charges de travail IA sont mappées, optimisées et exécutées sur le matériel informatique du robot, afin de réduire la consommation d'énergie et les coûts tout en répondant aux exigences strictes de latence et de fiabilité d'un système autonome en temps réel.
Responsabilités
- Prendre en charge l'architecture d'inférence embarquée : mapper les modèles aux accélérateurs disponibles (NPU, GPU, DSP, CPU) en fonction des budgets de latence, de puissance et de mémoire.
- Répartir les charges de travail d'inférence entre les ressources informatiques hétérogènes, en équilibrant les performances en temps réel avec les contraintes de puissance et thermiques.
- Définir et maintenir un budget informatique au niveau du système pour toutes les tâches d'inférence exécutées sur le robot.
- Évaluer le matériel d'accélération de nouvelle génération et contribuer à la définition des exigences futures de la plateforme informatique.
- Optimiser les chaînes d'outils d'inférence de bout en bout, de l'exportation du modèle à l'exécution du runtime, pour le matériel cible.
- Appliquer la quantification (INT8, INT4, précision mixte), l'élagage, la fusion d'opérateurs et d'autres techniques de compression pour réduire l'empreinte informatique, mémoire et énergétique.
- Profiler les pipelines d'inférence pour identifier et éliminer les goulots d'étranglement en matière de latence, de bande passante mémoire et de consommation d'énergie.
- Optimiser la planification des noyaux, la disposition de la mémoire et le transfert de données à travers la hiérarchie informatique.
- Collaborer étroitement avec l'équipe IA/ML pour définir dès le départ les contraintes d'architecture de modèle adaptées au matériel.
- Travailler avec l'équipe Logiciel Plateforme sur l'intégration du runtime, la planification et la gestion de l'énergie.
- Collaborer avec les fournisseurs de silicium et les équipes de recherche pour suivre le paysage des accélérateurs et influencer les feuilles de route matérielles.
Exigences
- Master ou doctorat en génie informatique, génie électrique, informatique ou dans un domaine connexe, ou expérience équivalente dans l'industrie.
- Au moins 8 ans d'expérience dans l'industrie en accélération matérielle, systèmes ML ou architecture informatique.
- Compréhension approfondie de l'inférence IA/ML : formats de modèles (ONNX, TFLite, etc.), runtimes d'inférence et pipelines de déploiement.
- Expérience pratique de l'optimisation de modèles pour le matériel edge ou embarqué en utilisant la quantification, l'élagage et le réglage au niveau de l'opérateur.
- Solide compréhension de l'architecture informatique : hiérarchies de mémoire, transfert de données et informatique hétérogène.
- Expérience en profilage et benchmarking de charges de travail d'inférence sur CPU, GPU, NPU, DSP.
- Familiarité avec les chaînes d'outils de bas niveau et les frameworks de compilation (par exemple TVM, MLIR, TensorRT, Torch, SNPE/QNN, JAX, CUDA, ROCm).
- Solides compétences en génie logiciel en C++ et Python.
- Solides compétences en communication interfonctionnelle : capacité à travailler efficacement avec les équipes matérielles, logicielles et IA/ML.
Qualifications bonus
- Connaissance des contraintes d'exploitation en temps réel et de leur impact sur la planification de l'inférence.
- Expérience avérée dans la co-conception d'architectures de modèles avec les équipes ML pour répondre aux contraintes matérielles.
La fourchette de salaire de base aux États-Unis pour ce poste à temps plein se situe entre $180,000 - $275,000 par an.
La rémunération proposée pour ce poste peut varier en fonction de plusieurs facteurs individuels, notamment les connaissances, les compétences et l'expérience liées au poste. Le package de rémunération globale peut également inclure des éléments/avantages supplémentaires en fonction du rôle spécifique. Ces informations seront partagées si une offre d'emploi est faite.
