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协议,支持模型导出、导入和部署。
并行训练: 为了支持并行训练,数据并行已列入开发计划中。