最流行的 AI 开发库

TensorFlow

Google 开源的机器学习框架

PyTorch

Facebook 开源的机器学习库

Keras

高层神经网络API

Scikit-learn

简单高效的机器学习库

Pandas

数据分析和操作库

Numpy

支持大量维度数组与矩阵运算

Apache MXNet

高效的开源深度学习框架

Theano

优化CPU/GPU数学运算

Caffe

快速的深度学习框架

Dlib

机器学习工具集

Fast.ai

面向实践的深度学习库

Transformers

自然语言处理库

Gymnasium

开源的强化学习库

Ray

为Python提供分布式执行

Chainer

灵活的深度学习框架

Sonnet

灵活的神经网络库

MILvus

开源向量数据库

Neptune

机器学习实验记录工具

Optuna

自动化超参数优化框架

MMdnn

深度学习模型互操作性解决方案

Hugging Face

自然语言处理模型库

AllenNLP

由华盛顿大学开发的NLP库

SpaCy

工业级自然语言处理

TextBlob

简易的NLP库

gensim

主题建模和文档相似性分析

NLTK

领先的平台，用于构建Python程序以处理人类语言数据

Rasa

构建自动化文本和语音助手

PaddlePaddle

百度开源深度学习平台

MindSpore

华为开源的全场景AI计算框架

ONNX

开放神经网络交换格式

Detectron2

Facebook AI研究的下一个版本

DeepSpeed

深度学习优化库

MMDetection

开放源代码的对象检测工具箱

OpenNMT

开源的神经机器翻译系统

Ludwig

用于深度学习模型的库

DALI

NVIDIA的数据处理库

Polyaxon

机器学习模型的生命周期管理

Apache Airflow

工作流管理平台

Prefect

工作流管理系统

Apache Beam

统一的模型进行批处理和流处理

JAX

高性能数值计算

Flax

由Google Research开发的神经网络库