未整理的资料#

Torch 模块#

可以查阅 前期资料 对 PyTorch 的粗浅介绍。

Captum

模型可解释模块。

PyTorch Geometric

PyG(PyTorch Geometric)是基于 PyTorch 构建的库,可以轻松地编写和训练图神经网络(GNN),用于各种与结构化数据相关的应用程序。如,graph,点云和流体。

TorchMetrics

TorchMetrics 是分布式、可伸缩的 PyTorch 模型的机器学习度量集合,以及易于使用的 API 来创建自定义度量。

transformers

State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.