A comprehensive set of fairness metrics for datasets
Data loaders and abstractions for text and NLP
Probabilistic reasoning and statistical analysis in TensorFlow
Embed images and sentences into fixed-length vectors
Solve end to end problems using Llama model family
A TensorFlow library for applied reinforcement learning
AI discovers 520000 stable inorganic crystal structures for research
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
NVIDIA Federated Learning Application Runtime Environment
Deep learning optimization library making distributed training easy
Detecting silent model failure. NannyML estimates performance
PyTorch extensions for fast R&D prototyping and Kaggle farming
Python package built to ease deep learning on graph
An Easy-to-use, Scalable and High-performance RLHF Framework
Standalone, small, language-neutral
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
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Fast image augmentation library and an easy-to-use wrapper
A library for accelerating Transformer models on NVIDIA GPUs
A lightweight 3D Morphable Face Model library in modern C++
Machine Learning Pipelines for Kubeflow
Training PyTorch models with differential privacy