The easiest way to use deep metric learning in your application
Graph Neural Network Library for PyTorch
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
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