Build portable, production-ready MLOps pipelines
Data loaders and abstractions for text and NLP
Streamline your ML workflow
A GPU-accelerated library containing highly optimized building blocks
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Fundamentals of Machine Learning and Deep Learning
A scientific machine learning (SciML) wrapper for the FEniCS
A machine learning library for detecting anomalies in signals
Nx-powered Neural Networks
ktrain is a Python library that makes deep learning AI more accessible
Proofs, cases, concept supplements, and reference explanations
A Python library for audio data augmentation
Detecting silent model failure. NannyML estimates performance
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Petastorm library enables single machine or distributed training
A cross-platform Python library for differentiable programming
A general-purpose probabilistic programming system
Spatiotemporal Signal Processing with Neural Machine Learning Models
A lightweight vision library for performing large object detection