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,065Updated this week
Related projects: ⓘ
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,303Updated 2 weeks ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,341Updated last year
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,499Updated this week
- MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…☆5,784Updated 3 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,512Updated this week
- Tutorials for creating and using ONNX models☆3,338Updated 2 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,362Updated this week
- An easy to use PyTorch to TensorRT converter☆4,557Updated last month
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆10,552Updated last week
- Google Brain AutoML☆6,208Updated 5 months ago
- Open standard for machine learning interoperability☆17,638Updated this week
- Geometric Computer Vision Library for Spatial AI☆9,806Updated this week
- Gluon CV Toolkit☆5,802Updated 5 months ago
- PyTorch extensions for high performance and large scale training.☆3,149Updated 2 weeks ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,158Updated 2 weeks ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,852Updated 2 weeks ago
- Reference models and tools for Cloud TPUs.☆5,210Updated 3 weeks ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆11,602Updated this week
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,446Updated this week
- Compiler for Neural Network hardware accelerators☆3,206Updated 4 months ago
- Differentiable architecture search for convolutional and recurrent networks☆3,900Updated 3 years ago
- ONNX-TensorRT: TensorRT backend for ONNX☆2,910Updated last week
- oneAPI Deep Neural Network Library (oneDNN)☆3,579Updated this week
- cuML - RAPIDS Machine Learning Library☆4,158Updated this week
- NumPy & SciPy for GPU☆8,124Updated this week
- Make huge neural nets fit in memory☆2,688Updated 4 years ago
- Serve, optimize and scale PyTorch models in production☆4,155Updated this week
- An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.☆2,783Updated last year
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,249Updated 6 months ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,286Updated last year