ML Libs Awesome

numpy
torch
- touchvision
- torchtext
- torchaudio
- torchvision
peft - Parameter-Efficient Fine-Tuning
- 用于高效参数微调的库，专注于在有限的计算资源下对大规模预训练模型进行微调。
transformers
diffusers
- PyTorch Diffusion Models
- by Hugging Face
- Denoising Diffusion Probabilistic Models (DDPM)
- Denoising Diffusion Implicit Models (DDIM)
accelerate
- by Hugging Face
- 简化和加速深度学习模型在不同硬件环境（如 CPU、GPU、TPU）上的分布式训练和推理
gradio - 快速构建交互界面
PIL - Image
openvino
tqdm - 显示进度条
einops - 用于操作张量的库
flash_attn - Flash Attention
- 不支持 macOS/Apple Silicon
timm - PyTorch Image Models
pypi.org/classifiers

关联信息