3. 训练框架的支持

MXMACA软件栈支持主流的训练框架,包括PyTorch和PaddlePaddle等。

3.1. PyTorch

PyTorch是一个开源的Python机器学习库,基于Torch,底层由C++实现,应用于人工智能领域,如计算机视觉和自然语言处理。

MXMACA软件栈以Wrapper的方式兼容PyTorch下的GPU代码,从而实现为PyTorch支持MetaX的硬件后端。适配优化后的PyTorch以PIP安装包和源码的形式提供。

3.2. PaddlePaddle

PaddlePaddle(飞桨)深度学习框架有大量的官方模型库,支持大规模训练和端到端的部署,详细信息可参见 官方网站

基于MXMACA软件对PaddlePaddle的适配和优化,使得MXMACA程序开发人员可以在曦云系列GPU上方便高效地使用PaddlePaddle。适配优化后的PaddlePaddle以PIP安装包和源码的形式提供。

3.3. Tensorflow

TensorFlow是一个开源的深度学习框架,完全基于Python语言设计,底层由C++实现。TensorFlow可以训练和运行深度神经网络,可以应用于多个场景,比如图像识别、图像分类、自然语言处理和视频检测等。

基于MXMACA软件对TensorFlow的适配和优化,使得MXMACA程序开发人员可以在曦云系列GPU上方便高效地使用TensorFlow。适配优化后的TensorFlow以PIP安装包和源码的形式提供。