TensorLayerX是一個跨平臺開發框架,可以運行在各類操作系統和AI硬件上,并支持混合框架的開發。目前支持TensorFlow、MindSpore、PaddlePaddle框架常用神經網絡層以及算子,PyTorch支持特性正在開發中。
TensorLayerX相比于之前開發的 TensorLayer有著更加強大的功能,它兼容多種計算框架后端。TensorLayer各個版本對比:
TensorLayerX繼承了之前版本的特性,包括簡單性,靈活性和低級抽象。 TensorLayerX支持多后端,如TensorFlow, MindSpore, PaddlePaddle和PyTorch。它允許用戶在不同的硬件上運行相同的代碼,比如Nvidia-GPU和HuaWei-Ascend。TensorLayerX的更多功能正在開發中。
模型庫: 構建包含經典模型和sota模型的系列模型庫,涵蓋CV、NLP、RL等領域。
模型部署: TensorLayerX將支持ONNX協議,支持模型導出、導入和部署。
并行訓練: 為了支持并行訓練,數據并行已列入開發計劃中。