PyTorch Lightning

pyannote.audiopyannote/pyannote-audio

An open-source Python toolkit for speaker diarization with state-of-the-art pretrained models and pipelines.

Dartsunit8co/darts

A Python library for user-friendly forecasting and anomaly detection on time series, from ARIMA to deep neural networks.

lightly - A computer vision framework for self-supervised learninglightly-ai/lightly

A Python library for self-supervised learning on images, providing a modular PyTorch-like framework with support for modern SSL models.

pytorch_geometric_temporalbenedekrozemberczki/pytorch_geometric_temporal

A PyTorch library for spatiotemporal signal processing with dynamic and temporal graph neural networks.

Atlasmagicleap/Atlas

An end-to-end deep learning system for reconstructing complete 3D scenes (geometry and semantics) from posed 2D images.

aitextgenminimaxir/aitextgen

A Python package for fine-tuning and generating text with GPT-2 and GPT Neo models using PyTorch and Hugging Face Transformers.

PyTorch Lightning BoltsPyTorchLightning/pytorch-lightning-bolts

A collection of models, callbacks, and datasets to extend PyTorch Lightning for applied AI/ML research and production.

pytorch_tabularmanujosephv/pytorch_tabular

A unified framework for implementing and training deep learning models on tabular data using PyTorch and PyTorch Lightning.

totalVIscverse/scvi-tools

A Python library for deep probabilistic modeling and analysis of single-cell and spatial omics data.