Principal Machine Learning Engineer, 3D

$146,900 - $237,600/Yr

Autodesk - San Francisco, CA

Full-time - Principal

Remote - San Francisco, CA

Publishing Industries

About the position

As a Principal Machine Learning Engineer at Autodesk Research, you will collaborate with a global team of researchers and engineers to develop and enhance machine learning-powered features for Autodesk products. This role focuses on applying modern deep learning techniques to solve complex problems in the fields of design and manufacturing, ultimately helping customers create a better world. You will work closely with AI researchers to implement generative AI features, contributing to both research and product development.

Responsibilities

• Collaborate on projects at the intersection of research and product with a diverse, global team of researchers and engineers
• Develop new or improve existing ML models used in CAD software
• Process data and analyze feature extractions
• Design solutions based on error analysis
• Present results to collaborators and leadership
• Review relevant AI/ML literature to identify emerging methods, technologies, and best practices

Requirements

• BSc or MSc in Computer Science, or equivalent industry experience
• 3+ years of software development experience
• Proficiency with modern deep learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) as well as frameworks (e.g. PyTorch, Lightning, Ray etc.)
• Experience with deploying machine learning models in production settings
• Experience with version control, reproducibility, and writing reusable, testable code
• Experience with data modeling, architecture, and processing using varied data representations including 2D and 3D geometry
• Experience with cloud services and architectures (e.g. AWS, Azure)
• Excellent written documentation skills to document code, architectures, and experiments

Nice-to-haves

• Experience scaling ML training and data pipelines
• Experience working with distributed systems
• Knowledge of the design, manufacturing, AEC, or media & entertainment industries
• Experience with Autodesk or similar products (CAD, CAE, CAM, etc.)

Benefits

• Health and financial benefits
• Time away and everyday wellness
• Annual cash bonuses
• Stock grants
• Comprehensive benefits package