AI frameworks and libraries

Deep Learning Frameworks

TensorFlow
PyTorch
Keras
Caffe
MXNet
Chainer
DL4J (Deeplearning4j)

General-Purpose Machine Learning

Scikit-learn
XGBoost
LightGBM
H2O
CatBoost

Natural Language Processing

NLTK (Natural Language Toolkit)
SpaCy
Gensim
BERT (from Hugging Face’s Transformers)

Computer Vision

OpenCV
Pillow
ImageAI
Reinforcement Learning

OpenAI Gym
Stable Baselines
Ray’s Rllib

Neural Network Architectures and Pre-trained Models

Hugging Face’s Transformers

Probabilistic Programming

Stan
PyMC3