Geometric deep learning extension library for PyTorch
Python-based neural networks API
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Hub of ready-to-use datasets for ML models
Industrial-strength Natural Language Processing (NLP)
2D and 3D Face alignment library build using pytorch
PyTorch3D is FAIR's library of reusable components for deep learning
Han Language Processing
Ongoing research training transformer models at scale
Deep universal probabilistic programming with Python and PyTorch
Audiocraft is a library for audio processing and generation
Making large AI models cheaper, faster and more accessible
Package of deep-learning based CTR models
ktrain is a Python library that makes deep learning AI more accessible
A game theoretic approach to explain the output of ml models
A set of Docker images for training and serving models in TensorFlow
A python library for easy manipulation and forecasting of time series
Deep learning optimization library: makes distributed training easy
Photorealistic Synthetic Dataset for Holistic Indoor Scene
AutoML library for deep learning
PaddlePaddle End-to-End Development Toolkit
Determined, deep learning training platform
A ranked list of awesome machine learning Python libraries
A python library for self-supervised learning on images
Jittor is a high-performance deep learning framework