Python examples of popular machine learning algorithms
Petastorm library enables single machine or distributed training
Open deep learning compiler stack for cpu, gpu, etc.
A unified framework for scalable computing
A library for scientific machine learning & physics-informed learning
Deep learning optimization library making distributed training easy
AutoML library for deep learning
End-to-End Library for Continual Learning based on PyTorch
Train machine learning models within Docker containers
TensorFlow is an open source library for machine learning
Fast image augmentation library and an easy-to-use wrapper
JAX-based neural network library
Graph Neural Network Library for PyTorch
A cross-platform Python library for differentiable programming
AIMET is a library that provides advanced quantization and compression
Python package built to ease deep learning on graph
Library for OCR-related tasks powered by Deep Learning
A python library for easy manipulation and forecasting of time series
Geometric deep learning extension library for PyTorch
A python library for self-supervised learning on images
Python library for defining and optimizing mathematical expressions
Models and examples built with TensorFlow
Fast Differentiable Tensor Library in JavaScript & TypeScript with Bun
A unified, comprehensive and efficient recommendation library
The Unified Machine Learning Framework