PyTorchΒΆ

Habana SynapseAI is integrated with PyTorch and optimized for Habana Gaudi HPU. SynapseAI works under the hood, so as a developer you can still enjoy the same experience in PyTorch that you are accustomed to today.