Alternatives and similar repositories for tensorflow:

Users that are interested in tensorflow are comparing it to the libraries listed below

• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆502Updated 3 weeks ago
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,677Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,719Updated last year
• NVIDIA / cuda-python
  CUDA Python: Performance meets Productivity
  ☆1,103Updated this week
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…
  ☆2,190Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆942Updated 2 weeks ago
• CVCUDA / CV-CUDA
  CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.
  ☆2,440Updated this week
• NVIDIA / CUDALibrarySamples
  CUDA Library Samples
  ☆1,776Updated last week
• NVIDIA / nvidia-container-toolkit
  Build and run containers leveraging NVIDIA GPUs
  ☆2,855Updated this week
• NVIDIA / DALI
  A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…
  ☆5,282Updated this week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆905Updated this week
• intel / llvm
  Intel staging area for llvm.org contribution. Home for Intel LLVM-based projects.
  ☆1,290Updated this week
• NVIDIA / DeepLearningExamples
  State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…
  ☆13,951Updated 6 months ago
• NVIDIA / cuda-samples
  Samples for CUDA Developers which demonstrates features in CUDA Toolkit
  ☆6,932Updated this week
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,468Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,260Updated this week
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆767Updated last week
• pytorch / cppdocs
  PyTorch C++ API Documentation
  ☆216Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,424Updated this week
• NVIDIA / nccl
  Optimized primitives for collective multi-GPU communication
  ☆3,473Updated 3 weeks ago
• intel / intel-extension-for-pytorch
  A Python package for extending the official PyTorch that can easily obtain performance on Intel platform
  ☆1,740Updated this week
• rapidsai / raft
  RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-a…
  ☆838Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆534Updated this week
• inducer / pycuda
  CUDA integration for Python, plus shiny features
  ☆1,903Updated last week
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆561Updated 3 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…
  ☆11,200Updated 2 weeks ago
• tensorflow / tensorrt
  TensorFlow/TensorRT integration
  ☆740Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆303Updated this week
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆347Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,025Updated 10 months ago