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.
Jittor is a high-performance deep learning framework
A ranked list of awesome machine learning Python libraries
OneFlow is a deep learning framework designed to be user-friendly
Powering Amazon custom machine learning chips
Python-based neural networks API
Enabling PyTorch on Google TPU
Open source machine learning framework
Medical imaging toolkit for deep learning
Deep universal probabilistic programming with Python and PyTorch
A python library for self-supervised learning on images
Industrial-strength Natural Language Processing (NLP)
Deep learning library
MII makes low-latency and high-throughput inference possible
Ongoing research training transformer models at scale
Trainable models and NN optimization tools
AutoML library for deep learning
Visualizer for neural network, deep learning, machine learning models
ktrain is a Python library that makes deep learning AI more accessible
A Python library for audio data augmentation
Python package built to ease deep learning on graph
A unified framework for scalable computing
Probabilistic time series modeling in Python