1、TensorFlow - 由谷歌开发,用于构建和训练深度学习模型。
2、PyTorch - 开源框架,支持动态计算图,适合快速开发。
3、ONNX - 模型转换工具,支持将TensorFlow、PyTorch等模型转换为ONNX格式。
4、Hugging Face Transformers - 提供预训练语言模型,支持NLP任务。
5、PyTorch Lightning - 高效训练框架,简化深度学习开发。
6、Keras - 高级深度学习API,基于TensorFlow。
7、OpenCV - 计算机视觉库,用于图像和视频处理。
8、Torchvision - PyTorch的补充库,提供预训练模型和数据集。
9、 horovod - 分布式训练优化库,提升训练效率。
10、aimagerecognition - 提供预训练视觉模型,支持快速部署。
这些工具广泛应用于AI模型构建、训练、优化和推理,满足不同需求。