PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A Python toolbox for scalable outlier detection
Open deep learning compiler stack for cpu, gpu, etc.
A ranked list of awesome machine learning Python libraries
Jittor is a high-performance deep learning framework
Powering Amazon custom machine learning chips
Enabling PyTorch on Google TPU
Python-based neural networks API
Open source machine learning framework
Han Language Processing
AutoML library for deep learning
Ongoing research training transformer models at scale
A unified framework for scalable computing
Fast image augmentation library and an easy-to-use wrapper
Probabilistic time series modeling in Python
Geometric deep learning extension library for PyTorch
Pre-trained Deep Learning models and demos
Probabilistic reasoning and statistical analysis in TensorFlow
Industrial-strength Natural Language Processing (NLP)
MII makes low-latency and high-throughput inference possible
Decentralized deep learning in PyTorch. Built to train models
A python library for easy manipulation and forecasting of time series
Deep learning library
A python library for self-supervised learning on images
Petastorm library enables single machine or distributed training