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
The open-source tool for building high-quality datasets
Graph Neural Network Library for PyTorch
Fast image augmentation library and an easy-to-use wrapper
AIMET is a library that provides advanced quantization and compression
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
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
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Machine Learning Systems: Design and Implementation
The Operator Splitting QP Solver
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
Multilingual Automatic Speech Recognition with word-level timestamps
Superfast AI decision making and processing of multi-modal data