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
Open source machine learning framework
Enabling PyTorch on Google TPU
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
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
Visualizer for neural network, deep learning, machine learning models
Probabilistic time series modeling in Python