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
- timm - PyTorch Image Models
- ml-tooling/best-of-ml-python
- https://pypi.org/classifiers/