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 library for scientific machine learning & physics-informed learning
A curated list of practical financial machine learning tools and apps
A unified framework for scalable computing
JAX-based neural network library
AutoML library for deep learning
Fast image augmentation library and an easy-to-use wrapper
TensorFlow is an open source library for machine learning
End-to-End Library for Continual Learning based on PyTorch
Deep learning optimization library making distributed training easy
Graph Neural Network Library for PyTorch
Library for OCR-related tasks powered by Deep Learning
Train machine learning models within Docker containers
A cross-platform Python library for differentiable programming
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 package built to ease deep learning on graph
Python library for defining and optimizing mathematical expressions
A unified, comprehensive and efficient recommendation library
The Unified Machine Learning Framework
Models and examples built with TensorFlow
AIMET is a library that provides advanced quantization and compression