PyTorch

TRLhuggingface/trl

A comprehensive library for post-training foundation models using reinforcement learning and fine-tuning techniques.

Sentence TransformersUKPLab/sentence-transformers

A Python framework for computing and training state-of-the-art text embeddings, rerankers, and sparse encoders.

torchvisionpytorch/vision

A PyTorch library providing datasets, model architectures, and image transformations for computer vision tasks.

cvatopencv/cvat

An industry-leading open-source data engine for interactive video and image annotation to power machine learning.

FinRLAI4Finance-LLC/FinRL-Library

An open-source framework for applying deep reinforcement learning to quantitative finance, featuring a train-test-trade pipeline for stock and crypto trading.

TorToiSeneonbjb/tortoise-tts

A multi-voice text-to-speech system that produces highly realistic prosody and intonation using autoregressive and diffusion decoders.

Flair embeddings from PubMedzalandoresearch/flair

A simple Python framework for state-of-the-art natural language processing (NLP) tasks like named entity recognition and sentiment analysis.

flairzalandoresearch/flair

A simple Python framework for state-of-the-art natural language processing (NLP) tasks like named entity recognition and sentiment analysis.

dgldmlc/dgl

A Python package for deep learning on graphs, framework-agnostic and optimized for performance and scalability.

GitHub repositoryintel-isl/Open3D

An open-source library for rapid development of software dealing with 3D data, with support for C++ and Python.

Stable Baselines3DLR-RM/stable-baselines3

A reliable PyTorch implementation of reinforcement learning algorithms for research and industry.

pytorch-grad-camjacobgil/pytorch-grad-cam

A PyTorch library providing state-of-the-art methods for generating visual explanations (Class Activation Maps) for computer vision models.

Ludwigludwig-ai/ludwig

A low-code declarative framework for building custom LLMs, neural networks, and other AI models with YAML configurations.

Ludwiguber/ludwig

A low-code declarative framework for building custom LLMs, neural networks, and other AI models with YAML configurations.

segmentation_models.pytorchqubvel/segmentation_models.pytorch

A PyTorch library providing 12+ semantic segmentation model architectures with 800+ pretrained convolutional and transformer-based encoders.

korniakornia/kornia

A differentiable computer vision library for PyTorch, providing geometric vision and image processing algorithms for AI workflows.

LAVISsalesforce/LAVIS

A Python library for language-vision intelligence research, providing unified access to state-of-the-art models, datasets, and tasks.

Tianshouthu-ml/tianshou

An elegant PyTorch-based deep reinforcement learning library with modular APIs for both research and application development.

pyannote.audiopyannote/pyannote-audio

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

Yolov3ultralytics/yolov3

A PyTorch implementation of YOLOv3 for real-time object detection, supporting export to ONNX, CoreML, and TFLite.

cleanrlvwxyzjn/cleanrl

A deep reinforcement learning library offering high-quality, single-file implementations of algorithms like PPO, DQN, and SAC for research and education.

EspNetespnet/espnet

An end-to-end speech processing toolkit for speech recognition, text-to-speech, translation, enhancement, and more.

GitHub repositoryfacebookresearch/pytorch3d

A PyTorch library providing efficient, reusable components for deep learning with 3D data, including mesh operations and differentiable rendering.

Dartsunit8co/darts

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

pyrouber/pyro

A flexible, scalable deep probabilistic programming library built on PyTorch for universal probabilistic modeling.

pyropyro-ppl/pyro

A flexible, scalable deep probabilistic programming library built on PyTorch for universal representation of computable probability distributions.

BigDLintel-analytics/BigDL

An LLM acceleration library for Intel XPU (GPU, NPU, CPU) to speed up local inference and finetuning of popular models.

MONAIProject-MONAI/MONAI

A PyTorch-based open-source framework for deep learning in healthcare imaging, providing domain-specific tools and workflows.

stanzastanfordnlp/stanza

A Python NLP library from Stanford for tokenization, sentence segmentation, NER, and dependency parsing across 60+ languages.

DeepChemdeepchem/deepchem

An open-source Python library for applying deep learning to drug discovery, materials science, quantum chemistry, and biology.

DeepPavlovdeepmipt/DeepPavlov

An open-source NLP framework for building and deploying deep learning dialog systems and chatbots with PyTorch and transformers.

mmdetection3dopen-mmlab/mmdetection3d

An open-source PyTorch toolbox for general 3D object detection, supporting LiDAR, camera, and multi-modal models.

MedicalGPTshibing624/MedicalGPT

An open-source pipeline for training medical domain GPT models using PT, SFT, RLHF, DPO, ORPO, and GRPO methods.

PCDetsshaoshuai/PCDet

A PyTorch-based toolbox for LiDAR-based 3D object detection, supporting multiple state-of-the-art models and datasets.

OpenPCDetopen-mmlab/OpenPCDet

A PyTorch-based toolbox for LiDAR-based 3D object detection, supporting multiple state-of-the-art models and datasets.

machine-learning-bookrasbt/machine-learning-book

Code repository for the 'Machine Learning with PyTorch and Scikit-Learn' book, providing practical examples and notebooks.