Machine Learning: Performance Engineer Remote or Hybrid United Kingdom Full-time

Job Requisition ID #

25WD85975

Machine Learning Performance Engineer Remote or Hybrid United Kingdom

Position Overview

The work we do at Autodesk touches nearly every person on the planet. By creating software tools for making buildings, machines, and even the latest movies, we influence and empower some of the most creative people in the world. As a Machine Learning Engineer at Autodesk Research, you will work side-by-side with world-class researchers and engineers to build new ML-powered product features to help our customers imagine, design, and make a better world. You are a software engineer who is passionate about solving problems and building things. You are excited to collaborate with AI researchers to implement generative AI features in Autodesk products.

You will report to a Research Manager in the Research Engineering organization of Autodesk Research.

Location:

We are a global team, located in London, San Francisco, Toronto, and remotely. For this role we support both in-person, hybrid, and remote work.

Responsibilities

• Implement GPU kernels to adapt to our models with low precision inference

• Write custom low balancing algorithms to optimize serving

• Profile and optimize machine learning tasks and code

• Write efficient code for machine learning tasks, focusing on software rather than hardware

• Split Jobs across multiple GPUs

• Understanding of the right type of GPUs for inference tasks

• Prepare appropriate containers and instances for various machine learning tasks

• Train and optimize machine learning models

• Collaborate on projects at the intersection of research and product with a diverse, global team of researchers and engineers

• Support research through the construction of ML pipelines, prototypes, and reusable, testable code

• Process data and analyze feature extractions

• Analyze errors and provide solutions to problems

• Present results to collaborators and leadership

Minimum Qualifications

• BSc or MSc in Computer Science, or equivalent industry experience

• 3+ years of software development experience

• Knowledge of Internals of PyTorch, CUDA and C++ for parallel processing

• Experience with version control, reproducibility, and deploying machine learning models

• Experience with cloud services and architectures (e.g. AWS, Azure)

• GPU compute programming models & optimization techniques

• GPU compute framework development, maintenance, and optimization

• Excellent written documentation skills to document code, architectures, and experiments

Preferred Qualifications

• Experience with data modeling, architecture, and processing using varied data representations including 2D and 3D geometry

• Experience with high performance parallel programming, or GPU programming

• Experience with adding computational graph support, runtime or device backend to Machine learning libraries (PyTorch or Lightning Ray) support

• 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.)

#LI-JK3

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 – our Culture Code is at the core of everything we do. Our values and ways of working help our people thrive and realize their potential, which leads to even better outcomes for our customers.

When you’re an Autodesker, you can be your whole, authentic self and do meaningful work that helps build a better future for all. Ready to shape the world and your future? Join us!

Salary transparency
Salary is one part of Autodesk’s competitive compensation package. Offers are based on the candidate’s experience and geographic location. In addition to base salaries, we also have a significant emphasis on discretionary annual cash bonuses, commissions for sales roles, stock or long-term incentive cash grants, and a comprehensive benefits package.

Diversity & Belonging
We take pride in cultivating a culture of belonging and an equitable workplace where everyone can thrive. Learn more here: 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).