Machine Learning Performance Engineer, Autodesk

• 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

Desirable

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