PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Elegant and Performant Deep Learning
Graph Neural Network Library for PyTorch
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
A GPU-accelerated library containing highly optimized building blocks
Geometric deep learning extension library for PyTorch
Jittor is a high-performance deep learning framework
Making large AI models cheaper, faster and more accessible
OpenVINO™ Toolkit repository
Determined, deep learning training platform
Medical imaging toolkit for deep learning
Fast image augmentation library and an easy-to-use wrapper
Spatiotemporal Signal Processing with Neural Machine Learning Models
A python library for self-supervised learning on images
Deep Learning API and Server in C++14 support for Caffe, PyTorch
Trainable models and NN optimization tools
A distributed system for embedding-based vector retrieval
On-device wake word detection powered by deep learning
Deep Learning Visualization Toolkit
Open source framework for deep learning satellite and aerial imagery
ktrain is a Python library that makes deep learning AI more accessible
AI Toolkit for Healthcare Imaging
Python package built to ease deep learning on graph
Decentralized deep learning in PyTorch. Built to train models
oneAPI Deep Neural Network Library (oneDNN)