Alternatives and similar repositories for tensorflow:

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

• NVIDIA / CUDALibrarySamples
  CUDA Library Samples
  ☆1,882Updated this week
• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆543Updated last month
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,600Updated this week
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…
  ☆2,365Updated 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,362Updated this week
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,743Updated last year
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆794Updated this week
• NVIDIA / cuda-python
  CUDA Python: Performance meets Productivity
  ☆2,284Updated this week
• pytorch / extension-cpp
  C++ extensions in PyTorch
  ☆1,080Updated 2 months ago
• CVCUDA / CV-CUDA
  CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.
  ☆2,491Updated 3 weeks ago
• NVIDIA / nccl
  Optimized primitives for collective multi-GPU communication
  ☆3,659Updated last 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…
  ☆14,158Updated 8 months ago
• NVIDIA / nvidia-container-toolkit
  Build and run containers leveraging NVIDIA GPUs
  ☆3,111Updated this week
• openxla / xla
  A machine learning compiler for GPUs, CPUs, and ML accelerators
  ☆3,100Updated this week
• pytorch / TensorRT
  PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
  ☆2,725Updated this week
• intel / llvm
  Intel staging area for llvm.org contribution. Home for Intel LLVM-based projects.
  ☆1,316Updated this week
• uxlfoundation / oneDNN
  oneAPI Deep Neural Network Library (oneDNN)
  ☆3,772Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,123Updated last year
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆620Updated this week
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆933Updated this week
• NVIDIA / multi-gpu-programming-models
  Examples demonstrating available options to program multiple GPUs in a single node or a cluster
  ☆684Updated 2 months ago
• tensorflow / tensorrt
  TensorFlow/TensorRT integration
  ☆741Updated last year
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆858Updated 3 months ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,299Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,498Updated this week
• ROCm / MIOpen
  AMD's Machine Intelligence Library
  ☆1,143Updated this week
• rapidsai / rmm
  RAPIDS Memory Manager
  ☆569Updated 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…
  ☆373Updated 2 weeks ago
• NVIDIA / cutlass
  CUDA Templates for Linear Algebra Subroutines
  ☆7,294Updated last week
• NVIDIA / cuCollections
  ☆535Updated this week