Build and run Docker containers leveraging NVIDIA GPUs
☆17,498Dec 6, 2023Updated 2 years ago
Alternatives and similar repositories for nvidia-docker
Users that are interested in nvidia-docker are comparing it to the libraries listed below
Sorting:
- Models and examples built with TensorFlow☆77,684Feb 13, 2026Updated 2 weeks ago
- An Open Source Machine Learning Framework for Everyone☆193,905Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,675Dec 1, 2025Updated 2 months ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆97,688Updated this week
- Deep Learning for humans☆63,866Updated this week
- Machine Learning Toolkit for Kubernetes☆15,462Jan 5, 2026Updated last month
- NVIDIA container runtime☆1,124Oct 27, 2023Updated 2 years ago
- Caffe: a fast open framework for deep learning.☆34,778Jul 31, 2024Updated last year
- Open standard for machine learning interoperability☆20,373Updated this week
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,829Oct 25, 2023Updated 2 years ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,413Feb 21, 2026Updated last week
- A library for efficient similarity search and clustering of dense vectors.☆39,195Updated this week
- Deep Learning GPU Training System☆4,181Jan 7, 2025Updated last year
- NVIDIA device plugin for Kubernetes☆3,671Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Feb 16, 2026Updated last week
- A toolkit for developing and comparing reinforcement learning algorithms.☆37,055Oct 11, 2024Updated last year
- Convolutional Neural Networks☆26,435May 3, 2024Updated last year
- A flexible, high-performance serving system for machine learning models☆6,350Dec 18, 2025Updated 2 months ago
- Visualizer for neural network, deep learning and machine learning models☆32,465Updated this week
- Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on sing…☆28,035Updated this week
- Setup and customize deep learning environment in seconds.☆6,281Jan 29, 2023Updated 3 years ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,723Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆14,735Aug 12, 2024Updated last year
- Build and run containers leveraging NVIDIA GPUs☆4,088Updated this week
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆157,071Updated this week
- Library for fast text representation and classification.☆26,496Mar 22, 2024Updated last year
- Open Machine Learning Compiler Framework☆13,142Updated this week
- Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit☆17,612Mar 11, 2023Updated 2 years ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Feb 21, 2026Updated last week
- Caffe2 is a lightweight, modular, and scalable deep learning framework.☆8,397Feb 7, 2023Updated 3 years ago
- TensorFlow-based neural network library☆9,902Feb 10, 2026Updated 2 weeks ago
- NumPy & SciPy for GPU☆10,804Updated this week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,375Feb 21, 2026Updated last week
- A flexible framework of neural networks for deep learning☆5,926Aug 28, 2023Updated 2 years ago
- An all-in-one Docker image for deep learning. Contains all the popular DL frameworks (TensorFlow, Theano, Torch, Caffe, etc.)☆3,864Aug 21, 2019Updated 6 years ago
- The Moby Project - a collaborative project for the container ecosystem to assemble container-based systems☆71,483Updated this week
- Run Kubernetes locally☆31,520Updated this week
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week