NVIDIA / DALILinks
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
☆5,417Updated last week
Alternatives and similar repositories for DALI
Users that are interested in DALI are comparing it to the libraries listed below
Sorting:
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,670Updated 3 weeks ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,768Updated this week
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,395Updated 2 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,665Updated 3 weeks ago
- An easy to use PyTorch to TensorRT converter☆4,746Updated 9 months ago
- PyTorch extensions for high performance and large scale training.☆3,328Updated last month
- ONNX-TensorRT: TensorRT backend for ONNX☆3,084Updated 3 weeks ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,675Updated 2 weeks ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,044Updated last year
- MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…☆5,809Updated last year
- Tutorials for creating and using ONNX models☆3,537Updated 10 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,133Updated last week
- A small package to create visualizations of PyTorch execution graphs☆3,364Updated 5 months ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,499Updated last month
- Flops counter for neural networks in pytorch framework☆2,898Updated 4 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,995Updated 10 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,943Updated last month
- Serve, optimize and scale PyTorch models in production☆4,330Updated this week
- A PyTorch implementation of EfficientNet☆8,111Updated 3 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,956Updated last month
- torch-optimizer -- collection of optimizers for Pytorch☆3,114Updated last year
- Codebase for Image Classification Research, written in PyTorch.☆2,154Updated last year
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,956Updated last week
- Gluon CV Toolkit☆5,888Updated 6 months ago
- 🐍 Geometric Computer Vision Library for Spatial AI☆10,508Updated this week
- Optimized primitives for collective multi-GPU communication☆3,761Updated last week
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,091Updated 3 years ago
- C++ extensions in PyTorch☆1,093Updated 4 months ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,625Updated last week
- An end-to-end PyTorch framework for image and video classification☆1,603Updated 11 months ago