Python library for decision tree visualization & model interpretation
SkyPilot: Run AI and batch jobs on any infra
A library for accelerating Transformer models on NVIDIA GPUs
High-level training, data augmentation, and utilities for Pytorch
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
Open source framework for deep learning satellite and aerial imagery
Toolkit for conversational AI
Data manipulation and transformation for audio signal processing
Making Enterprise Data Intelligent and Responsive for AI
An unsupervised and free tool for image and video dataset analysis
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling
AI discovers 520000 stable inorganic crystal structures for research
NVIDIA Federated Learning Application Runtime Environment
Training data (data labeling, annotation, workflow) for all data types
Tool for visualizing and tracking your machine learning experiments
Train machine learning models within Docker containers
Library for training machine learning models with privacy for data
Optax is a gradient processing and optimization library for JAX
An easy-to-use & supercharged open-source experiment tracker
A very simple framework for state-of-the-art NLP
Easy-to-use deep learning framework with 3 key features
Open Source Differentiable Computer Vision Library