55 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Machine learning metrics for distributed, scalable PyTorch applications. | 12.8M | |
| Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code ... | 11.6M | |
| Speed up model training by fixing data loading. | 249K | |
| Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code ... | 143K | |
| anomalib - Anomaly Detection Library | 39K | |
| The Lightning Logger of the Lightning.AI Platform | 22K | |
| PyTorch compiler that accelerates training and inference. Get built-in optimizat... | 10K | |
| Save, load, host, and share AI model checkpoints without slowing down training. ... | 8K | |
| Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code ... | 5K | |
| The GenAI Forecasting Agent · LLMs × Time Series Foundation Models | 5K | |
| dgenerate is a scriptable command line tool (and library) for generating images ... | 2K | |
| 普强内部NLP数据存储分享处理工具 | 2K | |
| Shape Analysis Explainability And Interpretability | 2K | |
| A PyTorch library for multi-modal image translation with diffusion bridges, GANs... | 2K | |
| REAX — Scalable, flexible training for JAX, inspired by the simplicity of PyTorc... | 2K | |
| A package for dipper detection in lightcurves | 1K | |
| Google EMEA gTech Ads Data Science Team's solution to automatically translate an... | 1K | |
| Lightning support for Intel Habana accelerators. | 1K | |
| This library provides tools and utilities for Few Shot Learning in Natural Langu... | 941 | |
| Semantic segmentation for plant roots grown in EcoFABs | 770 | |
| Opinionated Inference Framework for Remote Sensing Deep Learning Applications. | 739 | |
| Lightning support for GraphCore accelerators | 733 | |
| OlmoEarth Runner - Geospatial ML processing tools for distributed inference and ... | 714 | |
| High-quality, single file and distributed implementation of Deep Reinforcement L... | 667 | |
| 一个非常NB的包 | 603 | |
| A torch-based integration method for single-cell multi-omic data. | 601 | |
| Automated machine-learned Potential Landscape explorer | 594 | |
| BackdoorMBTI is an open source project expanding the unimodal backdoor learning ... | 567 | |
| PDF processing pipeline: remove headers/footers, convert to markdown, and genera... | 414 | |
| A benchmarking toolkit for genomic prediction with multiple methods and LLM-powe... | 399 | |
| 390 | ||
| Enterprise-Grade extensions for PyTorch Lightning | 357 | |
| 350 | ||
| 327 | ||
| BEASTsim (BEnchmarking and Analysis of Spatial Transcriptomics simulations) is a... | 311 | |
| Production-ready transcription and diarization pipeline with parallel processing | 302 | |
| Comprehensive Scene Text Recognition Toolkit across 11 Indian Languages | 285 | |
| Admetica is a command-line tool for making ADMET (Absorption, Distribution, Meta... | 274 | |
| A standalone service for transcribing audio files using WhisperX | 240 | |
| An advanced neural network framework with interpretability, generalization, robu... | 233 | |
| Foundational Kit for AI Training | 206 | |
| Foundational Kit for AI Training | 188 | |
| 161 | ||
| Lightning-native library for fine-tuning LLMs using alignment techniques like PP... | 154 | |
| TabStruct is a comprehensive benchmark suite for tabular data generation, predic... | 142 | |
| 130 | ||
| A DSP Toolkit for PyTorch | 110 | |
| Framework for differentiable black-box and gray-box audio effects modeling | 103 | |
| 基于funasr的语音识别服务 | 102 | |
| Speed up model training by fixing data loading. | 93 |