packages
- pyTorch
- tensorflow
- cudnn
- nccl
- keras
- horovod
- tensorboard
- xgboost
- torchmetrics
- torchtext
- torchvision
- torch-sparse
- torch-scatter
- torch-geometric
- pytorch-lightning
- tensorflow-text
- tensorflow-addons
- transformers
- mpi4py
- onnx
- dali
- jax/jaxlib
- mpi4jax
- pybind11
- huggingface-hub
- deepspeed
- scikit-learn