简介#

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.