Python package built to ease deep learning on graph
Standalone, small, language-neutral
Reference implementations of MLPerf™ training benchmarks
TFX is an end-to-end platform for deploying production ML pipelines
TimeGPT-1: production ready pre-trained Time Series Foundation Model
Pretrained (Language) Models for Probabilistic Time Series Forecasting
A high performance implementation of HDBSCAN clustering
Python library for decision tree visualization & model interpretation
SkyPilot: Run AI and batch jobs on any infra
Open-source tool designed to enhance the efficiency of workloads
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Library for OCR-related tasks powered by Deep Learning
The open-source tool for building high-quality datasets
Fast image augmentation library and an easy-to-use wrapper
Graph Neural Network Library for PyTorch
Environments and algorithms for research in general reinforcement
An unsupervised and free tool for image and video dataset analysis
AIMET is a library that provides advanced quantization and compression
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
NVIDIA Federated Learning Application Runtime Environment
A library for scientific machine learning & physics-informed learning
Training data (data labeling, annotation, workflow) for all data types
A lightweight vision library for performing large object detection
Tool for visualizing and tracking your machine learning experiments
The most intuitive, flexible, way for researchers to build models