pytorch / TensorRT

Related projects: ⓘ

• onnx / onnx-tensorrt
  ONNX-TensorRT: TensorRT backend for ONNX
  ☆2,910Updated last week
• NVIDIA-AI-IOT / torch2trt
  An easy to use PyTorch to TensorRT converter
  ☆4,557Updated last month
• daquexian / onnx-simplifier
  Simplify your onnx model
  ☆3,777Updated 2 weeks ago
• CVCUDA / CV-CUDA
  CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.
  ☆2,325Updated 2 weeks ago
• quic / aimet
  AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.
  ☆2,088Updated this week
• onnx / tensorflow-onnx
  Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
  ☆2,296Updated 2 weeks ago
• ZhangGe6 / onnx-modifier
  A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.
  ☆1,284Updated 2 months ago
• NVIDIA / TensorRT
  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
• NVIDIA / DALI
  A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…
  ☆5,065Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,149Updated 2 weeks ago
• facebookresearch / fvcore
  Collection of common code that's shared among different research projects in FAIR computer vision team.
  ☆1,981Updated last month
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…
  ☆1,811Updated this week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,303Updated 2 weeks ago
• onnx / optimizer
  Actively maintained ONNX Optimizer
  ☆634Updated 6 months ago
• webdataset / webdataset
  A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
  ☆2,206Updated this week
• onnx / onnx-tensorflow
  Tensorflow Backend for ONNX
  ☆1,269Updated 5 months ago
• IntelLabs / distiller
  Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…
  ☆4,341Updated last year
• open-mmlab / mmdeploy
  OpenMMLab Model Deployment Framework
  ☆2,705Updated last month
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆5,773Updated 5 months ago
• openvinotoolkit / nncf
  Neural Network Compression Framework for enhanced OpenVINO™ inference
  ☆906Updated this week
• tensorflow / tensorrt
  TensorFlow/TensorRT integration
  ☆737Updated 9 months ago
• sovrasov / flops-counter.pytorch
  Flops counter for convolutional networks in pytorch framework
  ☆2,783Updated 2 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆994Updated 5 months ago
• facebookresearch / pycls
  Codebase for Image Classification Research, written in PyTorch.
  ☆2,133Updated 5 months ago
• mit-han-lab / once-for-all
  [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment
  ☆1,863Updated 9 months ago
• NVIDIA / trt-samples-for-hackathon-cn
  Simple samples for TensorRT programming
  ☆1,477Updated 2 weeks ago
• onnx / tutorials
  Tutorials for creating and using ONNX models
  ☆3,338Updated 2 months ago
• zerollzeng / tiny-tensorrt
  Deploy your model with TensorRT quickly.
  ☆756Updated 9 months ago
• intel / neural-compressor
  SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX R…
  ☆2,146Updated this week
• Lightning-AI / torchmetrics
  Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,077Updated this week