Habana Catalog
TensorFlow
TensorFlow is a free and open-source software library for machine learning.
Habana’s SynapseAI software suite is integrated with the publicly available TensorFlow versions. Users can start training models on Gaudi by simply loading the SynapseAI library.
Habana has already ported many TensorFlow models to Gaudi. These reference models are readily available on Habana GitHub, and cover popular domains such as computer vision and natural language processing.
PyTorch
PyTorch is a free and open-source machine learning library based on the Torch library.
Habana’s SynapseAI software suite is integrated with the publicly available PyTorch versions. Habana’s custom PyTorch package provides everything needed to integrate PyTorch with SynapseAI.
Habana has already ported many PyTorch models to Gaudi. These reference models are readily available on Habana GitHub, and cover popular domains such as computer vision and natural language processing.