Senior Principal Machine Learning Engineer, Foundational Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

Autodesk · 22 hours ago

Senior Principal Machine Learning Engineer, Foundational Models

Autodesk is transforming the Architecture, Engineering, and Construction (AEC) industry by embedding advanced AI and foundation models into cloud-native platforms. As a Senior Principal Machine Learning Engineer, you will lead complex ML initiatives, setting technical direction and ensuring the delivery of scalable ML systems.

3D TechnologyArchitectureConstructionManufacturingSoftware Engineering
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Define the long-term technical vision for Generative AI and Foundation Model infrastructure within the AEC Solutions team. Influence architectural decisions across the broader organization
Lead the design, development, and delivery of complex ML systems. Own the full lifecycle from model architecture selection and data strategy to distributed training and production deployment
Drive the development of large-scale training pipelines. Collaborate with Research Scientists to translate experimental ideas (custom architectures, novel loss functions) into scalable, performant code
Architect solutions for distributed training (e.g., FSDP, Megatron-LM, DeepSpeed) on massive compute clusters. Identify and resolve bottlenecks in data processing and model parallelism to maximize training throughput
Mentor Principal and Senior engineers, fostering a culture of technical ownership, rigorous experimentation, and best practices. Act as a technical partner to Product Management and Engineering leadership
Partner effectively with Data Engineering, Platform, and Research teams to integrate large-scale multimodal AEC data (3D geometry, images, text) into model development workflows
Establish standards for model evaluation, versioning, monitoring, and MLOps best practices to ensure reproducibility and reliability in a high-stakes production environment

Qualification

Machine LearningDeep LearningPyTorchDistributed TrainingPythonReinforcement LearningMLOpsData EngineeringOperational ExcellenceMentorshipCross-Functional CollaborationTechnical Leadership

Required

Master's or PhD in a field related to AI/ML such as Computer Science, Mathematics, Statistics, Physics, Computational Linguistics, or related disciplines
10+ years of experience in machine learning, AI, or related fields, with a proven track record of technical leadership and hands-on implementation
Demonstrated experience mentoring engineers and leading technical projects in cross-functional environments
Proven history of leading the delivery of large-scale ML systems from conception to production
Expert-level understanding of deep learning architectures (Transformers, Diffusion models) and modern frameworks (PyTorch is required)
Hands-on experience with distributed training frameworks and techniques (e.g., PyTorch Distributed, Ray, DeepSpeed, Megatron, CUDA optimization) in HPC or cloud environments (AWS/Azure)
Strong proficiency in Python, with an emphasis on performance profiling, debugging, and writing robust, maintainable production code
Excellent ability to translate complex technical concepts into clear insights for executive leadership and cross-functional partners

Preferred

Experience with large foundation model training in distributed compute environments
Experience designing data pipelines for multimodal datasets at the terabyte/petabyte scale (using Spark, Iceberg, etc.)
Experience constructing internal developer platforms for ML, utilizing tools like Kubernetes, Slurm, or Metaflow
A portfolio demonstrating the successful translation of academic research papers into tangible product features
Background in AEC, computational geometry, or experience working with 3D data representations (BIM, CAD, meshes, point clouds)

Company

Autodesk

company-logo
Autodesk develops 3D design software for use in the architecture, engineering, construction, and media industries.

H1B Sponsorship

Autodesk has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (233)
2024 (243)
2023 (269)
2022 (359)
2021 (298)
2020 (260)

Funding

Current Stage
Public Company
Total Funding
$1.5B
Key Investors
CitibankOK Blockchain Capital
2025-07-01Acquired
2025-05-08Post Ipo Debt· $1.5B
2020-08-25Seed

Leadership Team

leader-logo
Andrew Anagnost
President and Chief Executive Officer
linkedin
leader-logo
Janesh Moorjani
CFO
linkedin
Company data provided by crunchbase