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
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
High-performance library for gradient boosting on decision trees
Library for OCR-related tasks powered by Deep Learning
Graph Neural Network Library for PyTorch
Environments and algorithms for research in general reinforcement
Fast image augmentation library and an easy-to-use wrapper
An unsupervised and free tool for image and video dataset analysis
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
NVIDIA Federated Learning Application Runtime Environment
An MLOps framework to package, deploy, monitor and manage models
A library for scientific machine learning & physics-informed learning
Training data (data labeling, annotation, workflow) for all data types
Tool for visualizing and tracking your machine learning experiments
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Machine Learning Systems: Design and Implementation
Sacred is a tool to help you configure, andorganize IDSIA experiments
An easy-to-use & supercharged open-source experiment tracker
A very simple framework for state-of-the-art NLP
Hyperparameter Optimization for TensorFlow, Keras and PyTorch