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
TensorBoard lets you watch Tensors Flow
Tensors and Dynamic neural networks in Python with strong GPU acceleration
TensorBoard lets you watch Tensors Flow
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
TensorFlow is an open source machine learning framework for everyone.
TensorFlow Estimator.
Fast data loading for TensorBoard
Orbax Checkpoint
MLflow is an open source platform for the complete machine learning lifecycle
Toolbox for imbalanced dataset in machine learning.
TensorFlow Serving Python API.
NVIDIA Collective Communication Library (NCCL) Runtime
CUBLAS native runtime libraries
Nvidia JIT LTO Library
NVRTC native runtime libraries
cuDNN runtime libraries
CUDA Runtime native Libraries
CUDA profiling tools runtime libs.
CUFFT native runtime libraries
CUSPARSE native runtime libraries
CURAND native runtime libraries
CUDA solver native runtime libraries
HuggingFace community-driven open-source library of datasets
NVIDIA Tools Extension
cuDNN runtime libraries
CUBLAS native runtime libraries
NVRTC native runtime libraries
CUDA Runtime native Libraries
Chex: Testing made fun, in JAX!
A gradient processing and optimisation library in JAX.
A unified approach to explain the output of any machine learning model.
ONNX Runtime is a runtime accelerator for Machine Learning models
Python library for easily interacting with trained machine learning models
A new flavour of deep learning operations
TF.Text is a TensorFlow library of text related ops, modules, and subgraphs.
MLflow is an open source platform for the complete machine learning lifecycle
PyTorch native Metrics
Python library for easily interacting with trained machine learning models
A library for efficient similarity search and clustering of dense vectors.
Ray provides a simple, universal API for building distributed applications.
Deep learning for humans.
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.
Containers for machine learning
Distrax: Probability distributions in JAX.
TensorFlow Hub is a library to foster the publication, discovery, and consumption of reusable parts of machine learning models.
Azure Machine Learning core packages, modules, and classes
Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.