PyTorch Support

We now officially support PyTorch in AI Lab pipelines and training scripts when running on Python 3.11 or higher.

New capabilities include:

• Training and using PyTorch models directly inside Script and Model steps
• Passing and manipulating torch.Tensor objects between pipeline steps
• Storing PyTorch models through context.set_model()
• Models must be saved as full TorchScript .pt objects (scripted modules), ensuring safe, portable, framework-agnostic deployment across all execution environments
• Support for tensor‐based feature extraction, embeddings, neural networks, etc.
• GPU acceleration where available

No additional configuration is required — simply import PyTorch in your pipeline scripts and start building.