AI back-end Engineer

Role overview

AI back-end Engineer at IBM

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.

Responsibilities

• Collaborate with Teams: Work in an agile, collaborative environment to understand requirements and deliver high-quality solutions that meet client needs.
• 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.

Required qualifications

• Ensure Solution Quality: Ensure that solutions meet the required standards for quality, resiliency, performance, security, and stability. 5-10 years of industrial experience in working with Unix/Linux based products developed using Python/C/C++ programming language.
• 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.
• 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 India Infrastructure & Technology Hybrid Professional Bangalore, IN Bangalore (0063) IBM India Private Limited Architecture Artificial Intelligence C++ Communication Debugging Design Thinking DevOps Engineering Linux Computer Programming Python Security Technical Leadership Testing Training Unix User Experience Financial and Operational Combined Uniform Single Reporting Resiliency Coding PyTorch C++ Programming Language Software Development Kit Graphics Processing Unit Innovation Parameter Memory Management Artificial Intelligence Development Solution Delivery

Employment details

• Location: Bangalore [India]

About the employer

• At IBM Infrastructure & Technology, we design and operate the systems that keep the world running.
• Our teams ensure the performance, security, and scalability that clients and industries depend on every day.
• 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.

Additional role details

• From high-resiliency mainframes and hybrid cloud platforms to networking, automation, and site reliability.
• 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 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.
• 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.
• AI Development