Models and examples built with TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
AI agents autonomously run and improve ML experiments overnight
Python examples of popular machine learning algorithms
A scientific machine learning (SciML) wrapper for the FEniCS
NVIDIA Federated Learning Application Runtime Environment
A machine learning library for detecting anomalies in signals
An MLOps framework to package, deploy, monitor and manage models
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 general-purpose probabilistic programming system
Spatiotemporal Signal Processing with Neural Machine Learning Models
A lightweight vision library for performing large object detection
A refreshing functional take on deep learning
Probabilistic time series modeling in Python
Tool for visualizing and tracking your machine learning experiments
Resources for deep learning with satellite & aerial imagery