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