Models and examples built with TensorFlow
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Fundamentals of Machine Learning and Deep Learning
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
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
An MLOps framework to package, deploy, monitor and manage models
Nx-powered Neural Networks
A Python library for audio data augmentation
Detecting silent model failure. NannyML estimates performance
Pytorch domain library for recommendation systems
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Training data (data labeling, annotation, workflow) for all data types
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 refreshing functional take on deep learning
Probabilistic time series modeling in Python
AI Toolkit for Healthcare Imaging