深度学习

通用

名称	状态	说明
pytorch-lightning		高级 PyTorch 接口，帮助写出更高质量、更简洁的代码。

日志

名称	状态	说明
TensorBoard		日志可视化工具，TensorFlow 自带，可单独搭配 PyTorch 等。
Comet		训练到生产的可视化与管理，支持外网访问与分享。
ClearML		开源日志可视化与协作，更适合团队使用。

优化

名称	状态	说明
Optuna		超参数优化框架，自动搜索，框架无关。
Ax		基于 PyTorch 的超参搜索框架，支持可视化分析。

NLP

名称	状态	说明
transformers		预训练模型库，涵盖完整 NLP Pipeline，基于 PyTorch。
SentenceTransformers		句子级嵌入库，基于 PyTorch。
fairseq		PyTorch 团队的 seq2seq 库，支持翻译、摘要等任务。

GNN

名称	状态	说明
DGL		深度图神经网络库，支持异构图与经典模型，含数据集。
pytorch_geometric / tf_geometric		同风格 GNN 库，分别支持 PyTorch / TensorFlow，涵盖经典与新模型。
PyGCL		PyTorch 图对比学习框架。
Neo4J		图数据库，社区版免费单实例，支持图算法与可视化。

CV

名称	状态	说明
Ultralytics		YOLO 系列官方团队，提供预训练模型与完整 CV Pipeline。
Open MMLab		开放 CV 社区，提供通用训练框架、模型与工具箱，覆盖 FewShot、Flow、3D 等。