Computer Vision

• easyocr - Ready-to-use OCR with 40+ languages supported.

Deep Learning

• keras - A high-level neural networks library and capable of running on top of either TensorFlow or Theano.
• pytorch-lightning - Deep learning framework to train, deploy, and ship AI products Lightning fast.
• stable-baselines3 - PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.
• vllm - A high-throughput and memory-efficient inference and serving engine for LLMs.

Machine Learning

• diffusers - A library that provides pretrained diffusion models for generating and editing images, audio, and video.
• Transformers - A framework that lets you easily use pretrained transformer models for NLP, vision, and audio tasks.

Natural Language Processing

• PyTorch-NLP - A toolkit enabling rapid deep learning NLP prototyping for research.
• Stanza - The Stanford NLP Group's official Python library, supporting 60+ languages.

Projects

AI and Agents

• funasr - Industrial-grade speech recognition toolkit with 170x realtime speed, 50+ languages, speaker diarization, and emotion detection.
• SenseVoice - A multilingual speech understanding model for speech recognition, language identification, emotion recognition, and audio event detection.
• peft - A library for parameter-efficient fine-tuning of large pretrained models.

Computer Vision

• ultralytics - Ultralytics YOLO for object detection, segmentation, pose estimation, and classification with state-of-the-art accuracy and speed.

Quantum Computing

• PennyLane - classical machine learning library with automatic differentiation support.

Recommender Systems

• spotlight - Deep recommender models using PyTorch.