1,007 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Fast and Accurate ML in 3 Lines of Code | 914K | |
| Fast and Accurate ML in 3 Lines of Code | 859K | |
| A scalable generative AI framework built for researchers and developers working ... | 798K | |
| A lightweight library for PyTorch training tools and utilities | 671K | |
| 🔎 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS,... | 487K | |
| Load tensorboard event logs as pandas DataFrames for scientific plotting; Suppor... | 375K | |
| Packaged version of ultralytics/yolov5 + many extra features | 332K | |
| An Open Source Machine Learning Framework for Everyone | 305K | |
| 🐸 - A general purpose model trainer, as flexible as it gets | 261K | |
| An Open Source Machine Learning Framework for Everyone | 177K | |
| Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.6, DeepSeek-R1, ... | 171K | |
| Open Audio Watermarking Tool | 157K | |
| A Model Compression Toolkit for neural networks | 144K | |
| spotriver - Sequential Parameter Optimization Interface to River | 136K | |
| 🐸 - A general purpose model trainer, as flexible as it gets | 135K | |
| A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs). | 132K | |
| An Open Source Machine Learning Framework for Everyone | 128K | |
| A Python toolkit/library for reality-centric machine/deep learning & data mining... | 122K | |
| verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework | 122K | |
| NeuralProphet: A simple forecasting package | 114K | |
| An elegant PyTorch deep reinforcement learning library. | 84K | |
| Deep probabilistic analysis of single-cell and spatial omics data | 77K | |
| Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NV... | 73K | |
| Transformers for Information Retrieval, Text Classification, NER, QA, Language M... | 73K | |
| 56K | ||
| A PyTorch native platform for training generative AI models | 54K | |
| SuperGradients | 42K | |
| A Model Compression Toolkit for neural networks | 41K | |
| The net:cal calibration framework is a Python 3 library for measuring and mitiga... | 41K | |
| All-in-one repository for state-of-the-art NeRFs | 34K | |
| 🤗 AutoTrain Advanced | 34K | |
| SchNetPack - Deep Neural Networks for Atomistic Systems | 33K | |
| RetroChimera: a frontier retrosynthesis model built on ensembling | 31K | |
| General utilities for developing deep learning projects using PyTorch | 30K | |
| Python Package for Airborne RGB machine learning | 30K | |
| Training library for Megatron-based models with bidirectional Hugging Face conve... | 29K | |
| AIMET is a library that provides advanced quantization and compression technique... | 29K | |
| Implementation of Alphafold 3 from Google Deepmind in Pytorch | 28K | |
| Research and development (R&D) is crucial for the enhancement of industrial prod... | 27K | |
| Production-grade adaptive meta-learning framework for continual model improvemen... | 26K | |
| Modular, scalable library to train ML models | 24K | |
| FAIR Sequence Modeling Toolkit 2 | 23K | |
| Open Source Neural Machine Translation and (Large) Language Models in PyTorch | 23K | |
| RapidFire AI: Rapid AI Customization from RAG to Fine-Tuning | 22K | |
| Go ahead and axolotl questions | 20K | |
| Isaac Lab API, powered by MuJoCo-Warp, for RL and robotics research | 20K | |
| 18K | ||
| A lightweight library for PyTorch training tools and utilities | 17K | |
| Ascend MindStudio Probe Utils | 15K | |
| Unified Training of Universal Time Series Forecasting Transformers | 15K |
