Principal Machine Learning Engineer - Foundation Models in AEC

idisciplinary team of scientists, engineers, and designers working on learning-based design systems, computer vision, graphics, robotics, human-computer interaction, sustainability, simulation, manufacturing, and construction. In this role, you will take ownership of advancing foundation model research in the AEC domain, leading efforts in experiment design, distributed training, optimization, and large-scale deployment to ensure efficiency and scalability.

You will report to the Machine Learning Manager within the AEC Solutions organization.

Responsibilities

• Collaborate with Research Scientists to translate novel model architectures and experimental ideas into robust, scalable implementations

• Develop, optimize, and deploy new ML models and AI techniques at scale

• Own the end-to-end training workflow: distributed training, debugging, and performance optimization

• Identify and apply best practices in large model training (e.g., parallelization, mixed precision, gradient checkpointing)

• Drive engineering efforts within a global team of scientists and engineers, ensuring reproducibility and efficiency of experiments

• Drive throughput by identifying bottlenecks in training pipelines and implementing improvements

• Partner with infrastructure and platform teams to leverage large-scale compute clusters and cloud services

Minimum Qualifications

• Master's or PhD in a field related to AI/ML such as: Computer Science, Mathematics, Statistics, Physics, Computational Linguistics, Mechanical Engineering, or related disciplines

Strong background in deep learning, including:

• Implementing custom architectures

• Optimizing model performance

• Developing novel loss functions

• Deploying production-ready solutions

• Familiarity with transformer-based models across various data modalities. Strong expertise in PyTorch (TensorFlow, JAX also valuable)

• Strong coding abilities in Python, with emphasis on debugging and performance profiling

• Hands-on experience with distributed training frameworks (e.g., PyTorch Distributed, DeepSpeed, Megatron-LM, FSDP, Horovod)

Preferred Qualifications

• Experience training foundation models on 2D, 3D, or multimodal data at scale

• Expert-level knowledge of transformers, scaling laws, and distributed training

• Demonstrated success in optimizing training or deployment pipelines for large models, including familiarity with: Ray, DeepSpeed, Megatron, Triton, CUDA, Metaflow/MLflow

• Familiarity using compute clusters and cloud services for large-scale ML pipelines

• Experience with multi-GPU and large-scale training in HPC or cloud environments

• Contributions to PyTorch or large-scale ML frameworks

• Significant post-graduate research experience, or 5+ years of industry experience, title and level commensurate with experience

• Knowledge of A****EC-related data modalities (3D geometry, CAD/BIM models, construction text corpus) is a strong plus

Ideal Candidate

• Strong builder mindset with ability to quickly implement and scale research ideas

• Detail-oriented, with strong debugging instincts and persistence in resolving training failures

• Passion for enabling cutting-edge research through high-quality engineering

• Ability to thrive in a fast-paced, collaborative research environment

Description du poste

Autodesk stimule l'innovation dans le secteur de l'architecture, de l'ingénierie et de la construction (AEC) en intégrant une IA avancée dans ses plateformes cloud natives de nouvelle génération. À travers des produits phares tels qu'AutoCAD, Revit, Construction Cloud et Forma, nous exploitons l'IA générative et les modèles de base pour transformer la manière dont les professionnels conçoivent, construisent et façonnent le monde.

En tant qu'ingénieur principal en apprentissage automatique chez Autodesk,

vous assurerez le leadership technique à la croisée de la recherche appliquée et de l'ingénierie à grande échelle, en partenariat avec des chercheurs scientifiques et des équipes interfonctionnelles afin d'accélérer l'innovation qui aide nos clients à imaginer, concevoir et créer un monde meilleur. Vous contribuerez à guider une équipe multidisciplinaire de scientifiques, d'ingénieurs et de concepteurs travaillant sur des systèmes de conception basés sur l'apprentissage, la vision par ordinateur, le graphisme, la robotique, l'interaction homme-machine, la durabilité, la simulation, la fabrication et la construction. À ce poste, vous serez chargé de faire progresser la recherche sur les modèles de base dans le domaine de l'AEC, en dirigeant les efforts en matière de conception d'expériences, de formation distribuée, d'optimisation et de déploiement à grande échelle afin de garantir l'efficacité et l'évolutivité.

Vous rendrez compte au responsable de l'apprentissage automatique au sein de l'organisation AEC Solutions.

Responsabilités

• Collaborer avec les chercheurs scientifiques afin de traduire les nouvelles architectures de modèles et les idées expérimentales en implémentations robustes et évolutives

• Développer, optimiser et déployer à grande échelle de nouveaux modèles d'apprentissage automatique et techniques d'IA

• Gérer le flux de travail de formation de bout en bout : formation distribuée, débogage et optimisation des performances

• Identifier et appliquer les meilleures pratiques en matière de formation de modèles à grande échelle (par exemple, parallélisation, précision mixte, vérification des gradients).

• Diriger les efforts d'ingénierie au sein d'une équipe mondiale de scientifiques et d'ingénieurs, en garantissant la reproductibilité et l'efficacité des expériences.

• Améliorer le débit en identifiant les goulots d'étranglement dans les pipelines de formation et en mettant en œuvre des améliorations.

• Collaborer avec les équipes chargées de l'infrastructure et de la plateforme afin de tirer parti des clusters de calcul à grande échelle et des services cloud.

Qualifications minimales

• Maîtrise ou doctorat dans un domaine lié à l'IA/ML, tel que : informatique, mathématiques, statistiques, physique, linguistique computationnelle, génie mécanique ou disciplines connexes

Solide expérience en apprentissage profond, notamment :

• Mise en œuvre d'architectures personnalisées

• Optimisation des performances des modèles

• Développement de nouvelles fonctions de perte

• Déploiement de solutions prêtes à la production

• Connaissance des modèles basés sur des transformateurs dans diverses modalités de données. Solide expertise enPyTorch (TensorFlow, JAX également apprécié)

• Solides compétences en codage en Python, avec un accent particulier sur le débogage et le profilage des performances

• Expérience pratique des cadres de formation distribués (par exemple, PyTorch Distributed, DeepSpeed, Megatron-LM, FSDP, Horovod)

Qualifications souhaitées

• Expérience dans la formation de modèles de base sur des données 2D, 3D ou multimodales à grande échelle

• Connaissance approfondie des transformateurs, des lois d'échelle et de la formation distribuée

• Succès avéré dans l'optimisation des pipelines de formation ou de déploiement pour les grands modèles, y compris une bonne connaissance de : Ray, DeepSpeed, Megatron, Triton, CUDA, Metaflow/MLflow

• Connaissance de l'utilisation des clusters de calcul et des services cloud pour les pipelines ML à grande échelle

• Expérience avec multi-GPU et formation à grande échelle dans des environnements HPC ou cloud

• Contributions à PyTorch ou à des cadres ML à grande échelle

• Expérience significative en recherche post-universitaire, ou 5 ans ou plus d'expérience dans l'industrie, titre et niveau correspondant à l'expérience

• La connaissance des modalités de données liées à l'AEC (géométrie 3D, modèles CAD/BIM, corpus de textes sur la construction) est un atout considérable

Candidat idéal

• Esprit constructif avec une capacité à mettre en œuvre et à adapter rapidement des idées de recherche

• Souci du détail, instinct de débogage et persévérance dans la résolution des échecs de formation

• Passion pour la recherche de pointe grâce à une ingénierie de haute qualité

• Capacité à s'épanouir dans un environnement de recherche collaboratif et dynamique

Learn More

About Autodesk

Welcome to Autodesk! Amazing things are created every day with our software - from the greenest buildings and cleanest cars to the smartest factories and biggest hit movies. We help innovators turn their ideas into reality, transforming not only how things are made, but what can be made.

We take great pride in our culture here at Autodesk - it's at the core of everything we do. Our culture guides the way we work and treat each other, informs how we connect with customers and partners, and defines how we show up in the world.

When you're an Autodesker, you can do meaningful work that helps build a better world designed and made for all. Ready to shape the world and your future? Join us!

Salary transparency
Salary is one part of Autodesk's competitive compensation package. For Canada-BC based roles, we expect a starting base salary between $141,600 and $194,700. Offers are based on the candidate's experience and geographic location, and may exceed this range. In addition to base salaries, our compensation package may include annual cash bonuses, commissions for sales roles, stock grants, and a comprehensive benefits package.

Diversity & Belonging
We take pride in cultivating a culture of belonging where everyone can thrive. Learn more here: https://https://www.autodesk.com/company/diversity-and-belonging

Are you an existing contractor or consultant with Autodesk?

Please search for open jobs and apply internally (not on this external site).