Geometric deep learning extension library for PyTorch
Python-based neural networks API
Fast Python collaborative filtering for implicit feedback datasets
An unsupervised and free tool for image and video dataset analysis
Tensor Learning in Python
Kaggle Python docker image
Running large language models on a single GPU
Graph Neural Network Library for PyTorch
Multilingual Automatic Speech Recognition with word-level timestamps
HivisionIDPhotos: a lightweight and efficient AI ID photos tools
Core ML tools contain supporting tools for Core ML model conversion
AIMET is a library that provides advanced quantization and compression
A unified, comprehensive and efficient recommendation library
Python package built to ease deep learning on graph
A unified framework for scalable computing
Jittor is a high-performance deep learning framework
Decomposable Multiscale Mixing for Time Series Forecasting
A Python library for audio data augmentation
Spatiotemporal Signal Processing with Neural Machine Learning Models
Models and examples built with TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
A cross-platform Python library for differentiable programming
Reference implementations of MLPerf™ training benchmarks
Python toolbox to create adversarial examples
High quality, fast, modular reference implementation of SSD in PyTorch