Repository Index

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

• deep-learning
• pytorch
• gpu
• machine-learning

💫 Industrial-strength Natural Language Processing (NLP) in Python

• natural-language-processing
• data-science
• machine-learning
• python

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python

• face-recognition
• vgg-face
• facenet
• openface

💬 Open source machine learning framework to automate text- and voice-based conversations: NLU, dialogue management, connect to Slack, Facebook, and more - Create chatbots and voice assistants

• nlp
• machine-learning
• machine-learning-library
• bot

🐍 Geometric Computer Vision Library for Spatial AI

• computer-vision
• image-processing
• machine-learning
• pytorch

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

TensorFlow-based neural network library

• machine-learning
• artificial-intelligence
• neural-networks
• deep-learning

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.

• tflearn
• tensorflow
• neural-network
• deep-learning

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

• python
• nlp
• natural-language-processing
• machine-learning

An open source python library for automated feature engineering

• feature-engineering
• machine-learning
• data-science
• automated-machine-learning

Python Audio Analysis Library: Feature Extraction, Classification, Segmentation and Applications

• audio
• machine-learning
• signal-processing
• audio-data

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.

• pytorch
• neural-network
• python
• machine-learning

A python library for self-supervised learning and latent world models on images.

• deep-learning
• self-supervised-learning
• machine-learning
• computer-vision

Visual tracking library based on PyTorch.

• computer-vision
• tracking
• machine-learning
• visual-tracking

:sunglasses: A curated list of robotics libraries and software

• robotics-libraries
• multibody-dynamics
• simulation
• robotics

StellarGraph - Machine Learning on Graphs

• graphs
• machine-learning
• machine-learning-algorithms
• graph-convolutional-networks

An intuitive library to add plotting functionality to scikit-learn objects.

• scikit-learn
• visualization
• machine-learning
• data-science

A modular active learning framework for Python

• scikit-learn
• machine-learning
• python
• active-learning

Optax is a gradient processing and optimization library for JAX.

• machine-learning
• optimization

A Python library for audio data augmentation. Useful for making audio ML models work well in the real world, not just in the lab.

• audio
• sound
• data-augmentation
• augmentation