PyTorch

Intel® Gaudi® software is integrated with PyTorch and optimized for Intel Gaudi AI accelerator. The software works under the hood, so as a developer you can still enjoy the same experience in PyTorch that you are accustomed to today.

Training on PyTorch

To start training on PyTorch, refer to Training on PyTorch

Inference on PyTorch

To start performing inference with PyTorch, refer to Inference on PyTorch

DeepSpeed

To start using DeepSpeed, refer to Large Models on PyTorch Using DeepSpeed

Reference

For additional documentation, refer to Reference