539 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code ... | 11.5M | |
| State-of-the-art speaker diarization toolkit | 2.2M | |
| Pytorch domain library for recommendation systems | 1.4M | |
| Fast and Accurate ML in 3 Lines of Code | 859K | |
| MatterSim: A deep learning atomistic model across elements, temperatures and pre... | 418K | |
| NequIP is a code for building E(3)-equivariant interatomic potentials | 371K | |
| A Neural Framework for MT Evaluation | 272K | |
| TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial ... | 232K | |
| spotriver - Sequential Parameter Optimization Interface to River | 136K | |
| A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs). | 132K | |
| Audiocraft is a library for audio processing and generation with deep learning. ... | 122K | |
| Generative models for conditional audio generation | 118K | |
| NeuralProphet: A simple forecasting package | 114K | |
| ML models + benchmark for tabular data classification and regression | 96K | |
| PyTorch-based framework that handles segmentation/regression/classification usin... | 94K | |
| Deep probabilistic analysis of single-cell and spatial omics data | 77K | |
| MACE - Fast and accurate machine learning interatomic potentials with higher ord... | 69K | |
| SuperGradients | 42K | |
| Supercharge Your Model Training | 40K | |
| anomalib - Anomaly Detection Library | 37K | |
| All-in-one repository for state-of-the-art NeRFs | 34K | |
| 🤗 AutoTrain Advanced | 34K | |
| SchNetPack - Deep Neural Networks for Atomistic Systems | 33K | |
| Python suite for neuroscience research across all modalities. | 32K | |
| A unified framework for Deep Learning Models on tabular data | 29K | |
| MedVAE is a family of six medical image autoencoders that can encode high-dimens... | 25K | |
| Supercharge Your Model Training | 23K | |
| Olive: Simplify ML Model Finetuning, Conversion, Quantization, and Optimization ... | 23K | |
| A project which aims to abstract the analysis pipeline of a Particle Physics Ana... | 20K | |
| Pytorch domain library for recommendation systems | 20K | |
| A sample project that exists for PyPUG's "Tutorial on Packaging and Distributing... | 20K | |
| Generalizable Perception Stack for all things 3D, 4D & Scene Understanding | 18K | |
| OCR/HTR engine for all the languages | 15K | |
| Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural netw... | 15K | |
| PyTorch code and models for the DINOv2 self-supervised learning method. | 15K | |
| Sequential Parameter Optimization in Python | 13K | |
| Implementation of Enformer, Deepmind's attention network for predicting gene exp... | 12K | |
| A simplified framework and utilities for PyTorch | 11K | |
| Bayesian active learning library for research and industrial usecases. | 11K | |
| Code for in-loop evaluation tasks used by the OLMo training team | 11K | |
| A Ray-based High-performance RLHF framework. | 10K | |
| Stable Pretraining Library for Self-Supervised Learning | 10K | |
| [ACM MM Award] AV-Deepfake1M: A Large-Scale LLM-Driven Audio-Visual Deepfake Dat... | 9K | |
| Deep survival analysis made easy | 9K | |
| Mock of the Vuforia Web Services (VWS) API | 8K | |
| Train models with self-supervised learning in a single command | 8K | |
| Post-hoc calibration methods and metrics for classification | 8K | |
| Range Invariant SSIM | 7K | |
| Materials for Deep Learning – ÚFAL course NPFL138 | 7K | |
| High-Resolution Image Synthesis with Latent Diffusion Models. This is a wrapper ... | 7K |
