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

TensorBoard lets you watch Tensors Flow

TensorBoard lets you watch Tensors Flow

TensorFlow is an open source machine learning framework for everyone.

HuggingFace community-driven open-source library of datasets

Toolbox for imbalanced dataset in machine learning.

NVIDIA Collective Communication Library (NCCL) Runtime

CUBLAS native runtime libraries

Fast data loading for TensorBoard

TensorFlow Estimator.

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

CUSPARSE native runtime libraries

Nvidia JIT LTO Library

CUDA Runtime native Libraries

CUDA profiling tools runtime libs.

CUFFT native runtime libraries

NVRTC native runtime libraries

CURAND native runtime libraries

cuDNN runtime libraries

CUDA solver native runtime libraries

NVIDIA Tools Extension

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

ONNX Runtime is a runtime accelerator for Machine Learning models

TensorFlow IO

A collection of chemoinformatics and machine-learning software written in C++ and Python

TensorFlow Serving Python API.

Efficiently computes derivatives of NumPy code.

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

PyTorch native Metrics

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

Python library for easily interacting with trained machine learning models

Fast, flexible, and advanced image augmentation library for deep learning and computer vision. Albumentations offers a wide range of transformations for images, masks, bounding boxes, and keypoints, with optimized performance and seamless integration into ML workflows.

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

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

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

A new flavour of deep learning operations

Python library for easily interacting with trained machine learning models

CUBLAS native runtime libraries

Containers for machine learning

cuDNN runtime libraries

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.

CUDA Runtime native Libraries

NVRTC native runtime libraries

Microsoft Azure Machine Learning Client Library for Python

Azure Machine Learning core packages, modules, and classes

HuggingFace community-driven open-source library of evaluation

TensorFlow Hub is a library to foster the publication, discovery, and consumption of reusable parts of machine learning models.