NVIDIA / tensorflow
An Open Source Machine Learning Framework for Everyone
☆1,140Updated 7 months ago
Alternatives and similar repositories for tensorflow
Users that are interested in tensorflow are comparing it to the libraries listed below
Sorting:
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆557Updated last month
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,745Updated this week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…☆2,412Updated this week
- Optimized primitives for collective multi-GPU communication☆3,710Updated 2 weeks ago
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,746Updated last year
- CUDA Library Samples☆1,924Updated this week
- TensorFlow/TensorRT integration☆742Updated last year
- ONNX-TensorRT: TensorRT backend for ONNX☆3,066Updated last week
- CUDA Core Compute Libraries☆1,636Updated this week
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,319Updated this week
- NCCL Tests☆1,096Updated last week
- CUDA Templates for Linear Algebra Subroutines☆7,450Updated last week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,575Updated last week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆808Updated this week
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,496Updated 2 weeks ago
- oneAPI Deep Neural Network Library (oneDNN)☆3,790Updated this week
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆703Updated 2 months ago
- Samples for Intel® oneAPI Toolkits☆1,032Updated last week
- Continuous builder and binary build scripts for pytorch☆352Updated last week
- Sample codes for my CUDA programming book☆1,712Updated 3 months ago
- Transformer related optimization, including BERT, GPT☆6,152Updated last year
- CUDA Kernel Benchmarking Library☆639Updated last week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,169Updated 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,239Updated 9 months ago
- RAPIDS Memory Manager☆579Updated this week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆384Updated this week
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆940Updated this week
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆984Updated 7 months ago
- nvidia-modelopt is a unified library of state-of-the-art model optimization techniques like quantization, pruning, distillation, speculat…☆909Updated last week
- Triton Python, C++ and Java client libraries, and GRPC-generated client examples for go, java and scala.☆620Updated this week