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
A unified framework for scalable computing
A python library for self-supervised learning on images
A python library for easy manipulation and forecasting of time series
AutoML library for deep learning
Deep universal probabilistic programming with Python and PyTorch
A framework for real-life data science
Industrial-strength Natural Language Processing (NLP)
Deep learning library
Probabilistic time series modeling in Python
Medical imaging toolkit for deep learning
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Audiocraft is a library for audio processing and generation
Geometric deep learning extension library for PyTorch
Visualizer for neural network, deep learning, machine learning models
Ongoing research training transformer models at scale