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
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
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
HuggingFace community-driven open-source library of datasets
TensorFlow is an open source machine learning framework for everyone.
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.
Ray provides a simple, universal API for building distributed applications.
A new flavour of deep learning operations
A library for efficient similarity search and clustering of dense vectors.
TensorFlow Serving Python API.
Python library for easily interacting with trained machine learning models
PyTorch native Metrics
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.
Azure ML Data Preparation SDK is used to load, transform, and write data for machine learning workflows
Python library for easily interacting with trained machine learning models
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.
TensorFlow is an open source machine learning framework for everyone.
Containers for machine learning
Efficiently computes derivatives of NumPy code.
CUBLAS native runtime libraries
HuggingFace community-driven open-source library of evaluation
cuDNN runtime libraries