A library for accelerating Transformer models on NVIDIA GPUs
Library for OCR-related tasks powered by Deep Learning
High-level training, data augmentation, and utilities for Pytorch
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
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
NVIDIA Federated Learning Application Runtime Environment
Toolkit for conversational AI
Optax is a gradient processing and optimization library for JAX
A very simple framework for state-of-the-art NLP
Making Enterprise Data Intelligent and Responsive for AI
Tool for visualizing and tracking your machine learning experiments
Train machine learning models within Docker containers
Open Source Differentiable Computer Vision Library
Superfast AI decision making and processing of multi-modal data
An MLOps framework to package, deploy, monitor and manage models
Flower: A Friendly Federated Learning Framework
A system for quickly generating training data with weak supervision
Uniform Manifold Approximation and Projection
PyTorch version of Stable Baselines
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Determined, deep learning training platform
Unified Model Serving Framework