AI back-end Engineer

Your Role and Responsibilities:

As a Back-End Developer specializing in Artificial Intelligence for IBM Z, you will work in an agile, collaborative environment to design, code, and deliver creative solutions to client problems and requirements. You will employ IBM's Design Thinking to create solutions that provide a great user experience along with world-class quality, resiliency, performance, security, and stability. Your Primary Responsibilities Will Include:

• Design and Develop Solutions: Design, code, debug, test, and deliver creative solutions to client problems and requirements in various AI-related areas, leveraging DevOps methodologies.

• Collaborate with Teams: Work in an agile, collaborative environment to understand requirements and deliver high-quality solutions that meet client needs.

• Apply Design Thinking: Employ IBM's Design Thinking to create solutions that provide a great user experience along with world-class quality, resiliency, performance, security, and stability.

• Lead Edge Projects: Work on leading-edge projects using technologies like machine learning, deep learning, and other AI-related areas to drive innovation and solution delivery.

• Ensure Solution Quality: Ensure that solutions meet the required standards for quality, resiliency, performance, security, and stability.

Required Education:

Bachelor's Degree

Required Technical and Professional Expertise:

• 5-10 years of industrial experience in working with Unix/Linux based products developed using Python/C/C++ programming language.
• AI Development Experience: Experience with designing, coding, debugging, testing PyTorch Core & Internals.
• Contribution to and extend PyTorch internals and backend architecture, including:
• TorchInductor
• TorchDynamo
• AOTAutograd
• Debug low-level PyTorch runtime issues (autograd, dispatcher, kernels, graph breaks).
• Deep hands-on experience with PyTorch FSDP
• Optimize model execution paths for training and inference.
• Open-source contributions to PyTorch or related projects
• Experience with proprietary or custom AI accelerators
• Experience integrating AI frameworks with custom or proprietary accelerators
• Advanced skills in C++ and Python
• Experience with GPU programming or accelerator SDKs
• Advanced Python skills and strong debugging ability
• DevOps Methodologies: Experience working with DevOps methodologies to deliver high-quality solutions on AI frameworks ensuring world-class quality, resiliency, performance, security, and stability.

Preferred Technical and Professional Experience:

• Deep understanding of:
• AI runtimes
• Memory hierarchies
• Parallel execution models
• PyTorch distributed runtime
• Parameter sharding and memory management
• Hands-on experience with torch.compile and TorchInductor
• Design Thinking Application: Experience employing IBM's Design Thinking to create solutions that provide a great user experience, with a focus on delivering innovative and effective solutions.
• Technical Leadership: Experience leading or contributing to leading-edge projects, driving innovation and solution delivery in Core backend AI areas.
• Excellent communication skills

****Years of Experience:****5 - 10