About IBM Infrastructure & Technology

We design and operate the systems that keep the world running. From high-resiliency mainframes and hybrid cloud platforms to networking, automation, and site reliability, our teams ensure the performance, security, and scalability that clients and industries depend on every day. Working in Infrastructure & Technology means tackling complex challenges with curiosity and collaboration. You’ll work with diverse technologies and colleagues worldwide to deliver resilient, future-ready solutions that power innovation. With continuous learning, career growth, and a supportive culture, IBM provides the opportunities to build expertise and shape the infrastructure that drives progress.

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 Qualifications

• Education: Bachelor's Degree
• AI Development Experience: Experience with designing, coding, debugging, testing PyTorch Core & Internals.
• PyTorch Internals Contribution: 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: Contributions to PyTorch or related projects.
• Accelerator Experience: Experience with proprietary or custom AI accelerators and integrating AI frameworks with them.
• Programming Skills: Advanced skills in C++ and Python, with strong debugging ability.
• GPU Programming: Experience with GPU programming or accelerator SDKs.
• 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 Qualifications

• Education: Master's Degree
• 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.
• Communication: Excellent communication skills.

About the Business Unit

IBM Systems helps IT leaders think differently about their infrastructure. IBM servers and storage are no longer inanimate - they can understand, reason, and learn so our clients can innovate while avoiding IT issues. Our systems power the world’s most important industries and our clients are the architects of the future. Join us to help build our leading-edge technology portfolio designed for cognitive business and optimized for cloud computing.

Your Life @ IBM

In a world where technology never stands still, we understand that dedication to our clients' success, innovation that matters, and trust and personal responsibility in all our relationships, lives in what we do as IBMers as we strive to be the catalyst that makes the world work better.

Being an IBMer means you’ll be able to learn and develop yourself and your career, you’ll be encouraged to be courageous and experiment every day, all whilst having continuous trust and support in an environment where everyone can thrive whatever their personal or professional background.

Our IBMers are growth-minded, always staying curious, open to feedback and learning new information and skills to constantly transform themselves and our company. They are trusted to provide on-going feedback to help other IBMers grow, as well as collaborate with colleagues keeping in mind a team focused approach to include different perspectives to drive exceptional outcomes for our customers. The courage our IBMers have to make critical decisions every day is essential to IBM becoming the catalyst for progress, always embracing challenges with resources they have to hand, a can-do attitude and always striving for an outcome-focused approach within everything that they do.