Skip to main content

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
  • https://pypi.org/classifiers/