DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Repository Index
47,133 repositories indexed
1–24 of 83 · Most stars
💫 Industrial-strength Natural Language Processing (NLP) in Python
Graph Neural Network Library for PyTorch
- pytorch
- geometric-deep-learning
- graph-neural-networks
- deep-learning
A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python
- face-recognition
- vgg-face
- facenet
- openface
A Library for Advanced Deep Time Series Models for General Time Series Analysis.
Easy-to-use and powerful LLM and SLM library with awesome model zoo.
- nlp
- embedding
- bert
- ernie
🐍 Geometric Computer Vision Library for Spatial AI
Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.
- weak-supervision
- data-cleaning
- data-quality
- noisy-labels
PyMuPDF is a high performance Python library for data extraction, analysis, conversion & manipulation of PDF (and other) documents.
- mupdf
- xps
- pdf-documents
- epub
An elegant PyTorch deep reinforcement learning library.
- pytorch
- policy-gradient
- dqn
- double-dqn
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
TensorFlow-based neural network library
A python library for user-friendly forecasting and anomaly detection on time series.
- python
- time-series
- forecasting
- machine-learning
Deep learning library featuring a higher-level API for TensorFlow.
AutoML library for deep learning
- automl
- neural-architecture-search
- automated-machine-learning
- python
Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
An open source python library for automated feature engineering
- feature-engineering
- machine-learning
- data-science
- automated-machine-learning
Deep Learning and Reinforcement Learning Library for Scientists and Engineers
- tensorlayer
- deep-learning
- tensorflow
- neural-network
Open-source offline translation library written in Python
- python
- machine-translation
- transformers
- translation
Python Audio Analysis Library: Feature Extraction, Classification, Segmentation and Applications
- audio
- machine-learning
- signal-processing
- audio-data
High-performance TensorFlow library for quantitative finance.
A library of extension and helper modules for Python's data analysis and machine learning libraries.
- python
- machine-learning
- data-science
- data-mining
Image augmentation library in Python for machine learning.
- augmentation
- machine-learning
- deep-learning
- neural-networks
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.