Fast and memory-efficient exact attention
RAPIDS Machine Learning Library
Fast, flexible and easy to use probabilistic modelling in Python
Making large AI models cheaper, faster and more accessible
Effortless data labeling with AI support from Segment Anything
The GPU-powered AI application database
A generic, simple and fast implementation of Deepmind's AlphaZero
Lightning fast C++/CUDA neural network framework
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Fast Forward Computer Vision (and other ML workloads!)
Experiment tracking, ML developer tools
TensorFlowOnSpark brings TensorFlow programs to Apache Spark clusters
CPU/GPU inference server for Hugging Face transformer models
Tez is a super-simple and lightweight Trainer for PyTorch
YOLO ROS: Real-Time Object Detection for ROS
Easily train your own text-generating neural network
A Neural Net Training Interface on TensorFlow, with focus on speed
Deep Learning toolkit for Computer Vision