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
TensorBoard lets you watch Tensors Flow
TensorBoard lets you watch Tensors Flow
CUBLAS native runtime libraries
Nvidia JIT LTO Library
CUSPARSE native runtime libraries
CUDA Runtime native Libraries
CUDA profiling tools runtime libs.
CUFFT native runtime libraries
NVRTC native runtime libraries
CURAND native runtime libraries
CUDA solver native runtime libraries
cuDNN runtime libraries
TensorFlow is an open source machine learning framework for everyone.
NVIDIA Collective Communication Library (NCCL) Runtime
HuggingFace community-driven open-source library of datasets
NVIDIA Tools Extension
Fast data loading for TensorBoard
MLflow is an open source platform for the complete machine learning lifecycle
TensorFlow Estimator.
Toolbox for imbalanced dataset in machine learning
MLflow is an open source platform for the complete machine learning lifecycle
ONNX Runtime is a runtime accelerator for Machine Learning models
TensorFlow Serving Python API.
NVIDIA cuSPARSELt
CUBLAS native runtime libraries
cuDNN runtime libraries
CUDA Runtime native Libraries
NVRTC native runtime libraries
PyTorch native Metrics
A new flavour of deep learning operations
TF.Text is a TensorFlow library of text related ops, modules, and subgraphs.
Python library for easily interacting with trained machine learning models
A unified approach to explain the output of any machine learning model.
A library for efficient similarity search and clustering of dense vectors.
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) data, with optimized performance and seamless integration into ML workflows.
Ray provides a simple, universal API for building distributed applications.
Python library for easily interacting with trained machine learning models
Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
John Snow Labs Spark NLP is a natural language processing library built on top of Apache Spark ML. It provides simple, performant & accurate NLP annotations for machine learning pipelines, that scale easily in a distributed environment.
TensorFlow Addons.
Containers for machine learning
Efficiently computes derivatives of NumPy code.
Azure Machine Learning core packages, modules, and classes
Orbax Checkpoint
HuggingFace community-driven open-source library of evaluation