Principal Machine Learning Operations for AI Research

extra holidays - extra parental leave - fully flexible
Work set-up: 
Full Remote
Contract: 
Experience: 
Senior (5-10 years)
Work from: 

Offer summary

Qualifications:

Bachelor's or Master's degree in Computer Science or related field, or equivalent industry experience., Experience with distributed systems for machine learning and deep learning at scale., Proficiency in Python and strong software engineering practices., Knowledge of ML infrastructure and model parallelism techniques, including frameworks like PyTorch, Lightning, Megatron, DeepSpeed, and FSDP..

Key responsibilities:

  • Build scalable ML training pipelines and infrastructure for foundation model development.
  • Design efficient data processing workflows for large-scale design datasets and industry-specific file formats.
  • Optimize distributed training systems and develop solutions for model parallelism, checkpointing, and resource management.
  • Collaborate with a global team of researchers and engineers, and present results to stakeholders.

Autodesk logo
Autodesk Computer Software / SaaS Large http://www.autodesk.com
10001 Employees
See all jobs

Job description

Job Requisition ID #

25WD90678

Job Title: Principal Machine Learning Operations Developer for AI Research

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 to solve problems that matter.

As a MLOps Developers at Autodesk Research, you will be working sidebyside with worldclass AI researchers to build and scale foundation models trained on design data. You will focus on overcoming the challenges associated with largescale model training and processing of vast amounts of diverse design data. Your expertise in distributed systems, ML infrastructure, and data engineering will be crucial in developing the next generation of MLpowered product features that will help our customers imagine, design, and make a better world.

Youll be joining a rapidly growing team working on a project that aims to revolutionize the design of nearly every aspect of the built environment. Your contributions will directly influence how designers, architects, and engineers interact with AI tools in the future.

This role is fully remotefriendly. Our team operates primarily remotely with team members distributed across the globe, with offices in London, Boston, Toronto and other locations worldwide. At Autodesk, we embrace remote work while fostering connection through regular team offsites for collaborative planning and relationship building. This balanced approach ensures you can work where youre most productive while maintaining meaningful connections with colleagues.

Responsibilities

  • Support AI researchers by building scalable ML training pipelines and infrastructure for foundation model development

  • Design efficient data processing workflows for largescale design datasets and industryspecific file formats

  • Optimize distributed training systems and develop solutions for model parallelism, checkpointing, and efficient resource management

  • Analyze performance bottlenecks and provide solutions to scaling problems

  • Implement and maintain robust, testable code that is well documented and easy to understand

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

  • Present results to collaborators and leadership

    • Minimum Qualifications

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

      • Experience with distributed systems for machine learning and deep learning at scale

      • Strong knowledge of ML infrastructure and model parallelism techniques, including frameworks like PyTorch, Lightning, Megatron, DeepSpeed, and FSDPProficiency in Python and strong software engineering practices

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

      • Familiarity with version control, CICD, and deployment pipelines

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

        • Preferred Qualifications

          • Experience with AEC data formats (e.g., BIM models, IFC files, CAD files, Drawing Sets)

          • Knowledge of the AEC industry and its specific data processing challenges

          • Experience scaling ML training and data pipelines for large datasets

          • Experience with distributed data processing and ML infrastructure (e.g., Apache Spark, Ray, Docker, Kubernetes)

          • Experience with performance optimization, monitoring, and efficiency in largescale ML systems

          • Experience with Autodesk or similar products (Revit, Sketchup, Forma)

            • The Ideal Candidate

              • A selfstarter who can solve problems with minimal supervision while collaborating effectively with a global, remotefirst team

              • Adaptable and creative, comfortable building new infrastructure or working within existing codebases

              • Thrives in ambiguous, rapidly evolving areas where learning and flexibility are essential

              • Excellent communicator who can convey complex technical concepts clearly to diverse audiences

                • 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. Offers are based on the candidate’s experience and geographic location. 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:www.autodesk.comcompanydiversityandbelonging

                  Are you an existing contractor or consultant with Autodesk?

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

Required profile

Experience

Level of experience: Senior (5-10 years)
Industry :
Computer Software / SaaS
Spoken language(s):
English
Check out the description to know which languages are mandatory.

Other Skills

  • Problem Solving
  • Communication
  • Collaboration

AI Operations (AI Ops) Engineer Related jobs