Machine Learning

A set of python modules for machine learning and data mining

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

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

HuggingFace community-driven open-source library of datasets

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

NVIDIA Collective Communication Library (NCCL) Runtime

CUBLAS native runtime libraries

CUSPARSE native runtime libraries

cuDNN runtime libraries

CUFFT native runtime libraries

CUDA solver native runtime libraries

Nvidia JIT LTO Library

NVRTC native runtime libraries

CUDA profiling tools runtime libs.

CURAND native runtime libraries

CUDA Runtime native Libraries

NVIDIA Tools Extension

TensorBoard lets you watch Tensors Flow

NVIDIA cuSPARSELt

TensorBoard lets you watch Tensors Flow

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

ONNX Runtime is a runtime accelerator for Machine Learning models

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

cuFile GPUDirect libraries

Deep learning for humans.

ml_dtypes is a stand-alone implementation of several NumPy dtype extensions used in machine learning.

TensorFlow is an open source machine learning framework for everyone.

Fast data loading for TensorBoard

NVSHMEM creates a global address space that provides efficient and scalable communication for NVIDIA GPU clusters.

A new flavour of deep learning operations

Toolbox for imbalanced dataset in machine learning

TensorFlow Estimator.

Python library for easily interacting with trained machine learning models

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

PyTorch native Metrics

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

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

TensorFlow IO

TensorFlow Serving Python API.

Python library for easily interacting with trained machine learning models

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

Orbax Checkpoint

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.

Collection of common python utils

Deep learning for humans.

lancedb

Efficiently computes derivatives of NumPy code.

ONNX Runtime is a runtime accelerator for Machine Learning models

HuggingFace community-driven open-source library of evaluation