AutoGluon: AutoML for Image, Text, and Tabular Data
The most intuitive, flexible, way for researchers to build models
Solve end to end problems using Llama model family
C++ DataFrame for statistical, Financial, and ML analysis
Data loaders and abstractions for text and NLP
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
A library for audio and music analysis, feature extraction
Standalone, small, language-neutral
Reference implementations of MLPerf™ training benchmarks
TFX is an end-to-end platform for deploying production ML pipelines
Pretrained (Language) Models for Probabilistic Time Series Forecasting
SkyPilot: Run AI and batch jobs on any infra
RAGFlow is an open-source RAG (Retrieval-Augmented Generation) engine
Time series forecasting with PyTorch
Implementation of Imagen, Google's Text-to-Image Neural Network
NVIDIA Federated Learning Application Runtime Environment
An MLOps framework to package, deploy, monitor and manage models
Train machine learning models within Docker containers
NMA Computational Neuroscience course
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Spatiotemporal Signal Processing with Neural Machine Learning Models
Python package built to ease deep learning on graph
RAPIDS Machine Learning Library
A library for accelerating Transformer models on NVIDIA GPUs