PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A Python toolbox for scalable outlier detection
Jittor is a high-performance deep learning framework
Open deep learning compiler stack for cpu, gpu, etc.
A ranked list of awesome machine learning Python libraries
Powering Amazon custom machine learning chips
Open source machine learning framework
Python-based neural networks API
Enabling PyTorch on Google TPU
A python library for self-supervised learning on images
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Medical imaging toolkit for deep learning
Probabilistic time series modeling in Python
Ongoing research training transformer models at scale
A unified framework for scalable computing
AutoML library for deep learning
A Python library for audio data augmentation
Visualizer for neural network, deep learning, machine learning models
Determined, deep learning training platform
Industrial-strength Natural Language Processing (NLP)
Pre-trained Deep Learning models and demos
Petastorm library enables single machine or distributed training
Decentralized deep learning in PyTorch. Built to train models
A python library for easy manipulation and forecasting of time series
Graph Neural Network Library for PyTorch