Machine Learning

Client library to download and publish models, datasets and other repos on the huggingface.co hub

A set of python modules for machine learning and data mining

Deep learning for humans.

State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow

Tensors and Dynamic neural networks in Python with strong GPU acceleration

TensorBoard lets you watch Tensors Flow

CUBLAS native runtime libraries

HuggingFace community-driven open-source library of datasets

CUSPARSE native runtime libraries

NVIDIA Collective Communication Library (NCCL) Runtime

Nvidia JIT LTO Library

cuDNN runtime libraries

CUFFT native runtime libraries

CUDA solver native runtime libraries

CUDA profiling tools runtime libs.

NVRTC native runtime libraries

CURAND native runtime libraries

CUDA Runtime native Libraries

TensorBoard lets you watch Tensors Flow

NVIDIA Tools Extension

TensorFlow is an open source machine learning framework for everyone.

Python library for easily interacting with trained machine learning models

MLflow is an open source platform for the complete machine learning lifecycle

MLflow is an open source platform for the complete machine learning lifecycle

NVIDIA cuSPARSELt

Toolbox for imbalanced dataset in machine learning

ONNX Runtime is a runtime accelerator for Machine Learning models

Fast data loading for TensorBoard

cuFile GPUDirect libraries

TensorFlow Estimator.

A new flavour of deep learning operations

Ray provides a simple, universal API for building distributed applications.

A library for efficient similarity search and clustering of dense vectors.

Python library for easily interacting with trained machine learning models

TensorFlow Serving Python API.

PyTorch native Metrics

TensorFlow IO

A unified approach to explain the output of any machine learning model.

TF.Text is a TensorFlow library of text related ops, modules, and subgraphs.

tensorflow/datasets is a library of datasets ready to use with TensorFlow.

Orbax Checkpoint

Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.

Fast, flexible, and advanced augmentation library for deep learning, computer vision, and medical imaging. Albumentations offers a wide range of transformations for both 2D (images, masks, bboxes, keypoints) and 3D (volumes, volumetric masks, keypoints) data, with optimized performance and seamless integration into ML workflows.

Containers for machine learning

TensorFlow is an open source machine learning framework for everyone.

Efficiently computes derivatives of NumPy code.

CUBLAS native runtime libraries

HuggingFace community-driven open-source library of evaluation

cuDNN runtime libraries

CUDA Runtime native Libraries