項目簡介

TensorLayerX是一個跨平臺開發(fā)框架，可以運行在各類操作系統(tǒng)和AI硬件上，并支持混合框架的開發(fā)。目前支持TensorFlow、MindSpore、PaddlePaddle框架常用神經(jīng)網(wǎng)絡(luò)層以及算子，PyTorch支持特性正在開發(fā)中。

TensorLayerX相比于之前開發(fā)的 TensorLayer有著更加強大的功能，它兼容多種計算框架后端。TensorLayer各個版本對比：

TensorLayerX繼承了之前版本的特性，包括簡單性，靈活性和低級抽象。 TensorLayerX支持多后端，如TensorFlow, MindSpore, PaddlePaddle和PyTorch。它允許用戶在不同的硬件上運行相同的代碼，比如Nvidia-GPU和HuaWei-Ascend。TensorLayerX的更多功能正在開發(fā)中。

• 模型庫: 構(gòu)建包含經(jīng)典模型和sota模型的系列模型庫，涵蓋CV、NLP、RL等領(lǐng)域。

• 模型部署: TensorLayerX將支持ONNX協(xié)議，支持模型導出、導入和部署。

• 并行訓練: 為了支持并行訓練，數(shù)據(jù)并行已列入開發(fā)計劃中。