Intern, Large Deep Learning Model Optimization

Posted:
11/25/2024, 7:12:03 AM

Location(s):
Toronto, Ontario, Canada ⋅ Ontario, Canada

Experience Level(s):
Internship

Field(s):
AI & Machine Learning

Workplace Type:
Hybrid

Job Requisition ID #

24WD83299

About the Canada Intern Program

The 2025 Canada program runs for 12 weeks (May 5 – August 22). All internships are paid. As an intern, you will contribute to meaningful projects, be mentored by industry leaders, and participate in tech talks and other activities designed to support your personal and professional development. Our internships align with Autodesk’s Flexible Workplace approach, which is designed to meet the needs of our business while providing flexibility in support of office, remote and hybrid work preferences. 

Position Overview

The intern will join our AI and machine learning team, collaborating on optimizing the inference and training processes for large deep learning models. Reporting to a senior ML researcher/engineer, the role involves investigating and implementing cutting-edge techniques to improve model serving and training efficiency while balancing speed, accuracy, and resource utilization.

Responsibilities

  • Analyze deep learning models in production and training environments to identify bottlenecks in inference and training times

  • Explore and implement optimization techniques such as quantization, batch inference, model partitioning, and caching

  • Investigate methods to accelerate training workflows, including mixed precision training, gradient accumulation, and distributed training

  • Evaluate the trade-offs between speed, accuracy, and resource usage for different optimization strategies

  • Utilize tools and libraries like TensorRT, PyTorch Distributed, and asynchronous processing frameworks to enhance performance

  • Document findings and provide actionable recommendations for integrating optimized models into production and training pipelines

Minimum Qualifications

  • Currently pursuing a Master's or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field

  • Hands-on experience with deep learning model optimization for inference and training

  • Proficiency in using frameworks and tools such as PyTorch, TensorFlow, TensorRT, or similar libraries

  • Strong programming skills in Python and familiarity with techniques like quantization, mixed precision training, and distributed training strategies

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, educational level, and geographic location.

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

Autodesk

Website: http://www.autodesk.com/

Headquarter Location: San Francisco, California, United States

Employee Count: 10001+

Year Founded: 1982

IPO Status: Public

Last Funding Type: Seed

Industries: 3D Technology ⋅ Architecture ⋅ Construction ⋅ Manufacturing ⋅ Software ⋅ Software Engineering