NVIDIA / DALI
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,299Updated this week
Alternatives and similar repositories for DALI:
Users that are interested in DALI are comparing it to the libraries listed below
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,688Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,566Updated this week
- An easy to use PyTorch to TensorRT converter☆4,682Updated 6 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,585Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,772Updated 3 weeks ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,371Updated last year
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,535Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,256Updated last month
- 🐍 Geometric Computer Vision Library for Spatial AI☆10,267Updated this week
- PyTorch extensions for high performance and large scale training.☆3,269Updated last month
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆8,826Updated this week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,899Updated 8 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,917Updated last month
- ONNX-TensorRT: TensorRT backend for ONNX☆3,028Updated last week
- Count the MACs / FLOPs of your PyTorch model.☆4,962Updated 7 months ago
- MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…☆5,804Updated 9 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,083Updated 3 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,403Updated last month
- Tutorials for creating and using ONNX models☆3,459Updated 7 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,087Updated 11 months ago
- Simplify your onnx model☆3,993Updated 6 months ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,215Updated this week
- Open standard for machine learning interoperability☆18,525Updated this week
- [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment☆1,899Updated last year
- Optimized primitives for collective multi-GPU communication☆3,503Updated last month
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,269Updated last year
- Codebase for Image Classification Research, written in PyTorch.☆2,149Updated 11 months ago
- A small package to create visualizations of PyTorch execution graphs☆3,307Updated 2 months ago
- Compiler for Neural Network hardware accelerators☆3,271Updated 9 months ago