542 dependents
| Package | Description | Downloads/month |
|---|---|---|
| State-of-the-art speaker diarization toolkit | 2.2M | |
| Fast and Accurate ML in 3 Lines of Code | 914K | |
| Fast and Accurate ML in 3 Lines of Code | 859K | |
| NequIP is a code for building E(3)-equivariant interatomic potentials | 371K | |
| Forecasting timeseries with PyTorch - dataloaders, normalizers, metrics and mode... | 370K | |
| TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial ... | 232K | |
| Wrappers that facilitate enabling training job telemetry for a set of supported ... | 224K | |
| spotriver - Sequential Parameter Optimization Interface to River | 136K | |
| A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs). | 132K | |
| PyTorch-based framework that handles segmentation/regression/classification usin... | 94K | |
| Deep probabilistic analysis of single-cell and spatial omics data | 77K | |
| Message Passing Neural Networks for Molecule Property Prediction | 55K | |
| anomalib - Anomaly Detection Library | 37K | |
| End-to-End Speech Processing Toolkit | 30K | |
| Implementation of Alphafold 3 from Google Deepmind in Pytorch | 28K | |
| MatGL is a framework for graph deep learning for materials science. | 27K | |
| PVNet | 18K | |
| OCR/HTR engine for all the languages | 15K | |
| Unified Training of Universal Time Series Forecasting Transformers | 15K | |
| 20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale... | 14K | |
| Sequential Parameter Optimization in Python | 13K | |
| Training and prediction scripts for Chemprop models trained on ADMET datasets | 11K | |
| A tool for developing remote sensing datasets and models. | 11K | |
| so-vits-svc fork with realtime support, improved interface and more features. | 11K | |
| A library for interacting with the Datamint API, designed for efficient data man... | 11K | |
| Stable Pretraining Library for Self-Supervised Learning | 10K | |
| [ACM MM Award] AV-Deepfake1M: A Large-Scale LLM-Driven Audio-Visual Deepfake Dat... | 9K | |
| Napari Visualization tool for Trackmate > 6.0 XML files for 2/3D + time tracks. ... | 9K | |
| Implementation of Nougat Neural Optical Understanding for Academic Documents | 8K | |
| MAGDI data | 8K | |
| Segmentation tool for biological cells of irregular size and shape in 3D and 2D. | 8K | |
| PyTorch Wildlife: a Collaborative Deep Learning Framework for Conservation. | 7K | |
| This package handles downloading, cleaning, analyzing street view imagery in a o... | 7K | |
| A direct Google Cloud Storage integration for PyTorch | 6K | |
| PyTorch Lightning Optical Flow models, scripts, and pretrained weights. | 6K | |
| Fish Speech | 6K | |
| 6K | ||
| Accurate and efficient spot detection for microscopy data | 6K | |
| TrustMark - Universal Watermarking for Arbitrary Resolution Images | 6K | |
| Shared utilities and training infrastructure for biomolecular structure predicti... | 6K | |
| 🍵 Matcha-TTS: A fast TTS architecture with conditional flow matching | 5K | |
| Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™ | 5K | |
| Bioacoustic Recognizer Toolkit | 5K | |
| Nucleobench optimizers and tasks. | 5K | |
| 5K | ||
| computer vision for image-based phenotyping of single cells | 5K | |
| A framework for training and evaluating foundation models, using the MEDS ecosys... | 5K | |
| The GenAI Forecasting Agent · LLMs × Time Series Foundation Models | 4K | |
| A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial O... | 4K | |
| A Comprehensive Framework for Building End-to-End Recommendation Systems with St... | 4K |
