PyTorch Framework engineer

About This Role

• The Role and Impact: As an AI Frameworks Engineer, you will play a pivotal role in shaping the future of artificial intelligence software development.
• You will contribute to the design, development, and optimization of AI frameworks, such as PyTorch.
• Your work will directly impact Intel's ability to deliver cutting-edge AI solutions, enhance distributed algorithms, and optimize software performance across various hardware backends.
• By collaborating with researchers, engineers, and other stakeholders, you will be instrumental in advancing Intel's mission to enable transformative AI capabilities worldwide.
• Key Responsibilities: - Design and develop AI software and frameworks to optimize performance across computing hardware backends. - Implement distributed algorithms, such as model/data parallel frameworks, parameter servers, and asynchronous data communication in machine learning and deep learning frameworks. - Transform computational graph representations of neural network models and develop machine learning and deep learning primitives in mathematical libraries. - Profile distributed deep learning models to identify performance bottlenecks and propose solutions across component teams. - Collaborate with machine learning and deep learning researchers to integrate advancements into framework optimization. - Contribute code to open-source projects that enhance Intel's AI capabilities while maintaining industry standards.