PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
thu-ml
sageattention

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

156K 3K 403
ml-energy
zeus-ml

Measure and optimize the energy consumption of your AI applications!

2K 355 45
ml-energy
zeus

Measure and optimize the energy consumption of your AI applications!

2K 355 45
tanyuqian
redco

NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference

1K 69 7
MLSysOps
alaas

A scalable & efficient active learning/data selection system for everyone.

369 218 15
SymbioticLab
fedscale

FedScale is a scalable and extensible open-source federated learning (FL) platform.

135 416 121
    • Data from PyPI, GitHub, ClickHouse, and BigQuery