Skip to content

AI Toolkit for IBM Z and LinuxONE

AI Toolkit for IBM Z and LinuxONE

The AI Toolkit for IBM Z® and LinuxONE is a family of popular open-source AI frameworks with IBM Elite Support and adapted for IBM Z and LinuxONE hardware.

This is a select set of containerized software that has been vetted by IBM. The pre-built images are available through the IBM Z and LinuxONE Container Registry.

These images can be run in Linux on Z environents (including z/OS Container Extensions). The software is optimized for IBM Z and LinuxONE platforms, including the ability to leverage the IBM Telum Integrated Accelerator for AI.

As of 2025, the AI Toolkit includes the following components:

  • IBM Z Deep Learning Compiler
  • IBM Snap ML
  • PyTorch
  • TensorFlow
  • TensorFlow Serving
  • Triton Inference Server

Learn more

Learn more about this Elite support offering here