Alternatives and similar repositories for apex

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

• ROCm / TransformerEngine
  ☆36Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆97Updated last month
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆64Updated last week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆67Updated 2 weeks ago
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆56Updated 2 weeks ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆97Updated 10 months ago
• ROCm / rocmProfileData
  ☆24Updated last month
• codeplaysoftware / cutlass-sycl
  A CUTLASS implementation using SYCL
  ☆23Updated last week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆122Updated this week
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 4 months ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆189Updated this week
• yifuwang / symm-mem-recipes
  ☆88Updated 5 months ago
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆79Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆38Updated 10 months ago
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆83Updated last week
• AlibabaPAI / FLASHNN
  ☆96Updated 8 months ago
• ROCm / hipBLAS
  ROCm BLAS marshalling library
  ☆142Updated this week
• ROCm / amd_matrix_instruction_calculator
  A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators
  ☆95Updated 2 weeks ago
• ColfaxResearch / cutlass-kernels
  ☆208Updated 10 months ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆173Updated this week
• sunlex0717 / DissectingTensorCores
  ☆97Updated last year
• ROCm / pytorch-micro-benchmarking
  ☆20Updated 2 months ago
• ROCm / rdc
  RDC
  ☆29Updated this week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆105Updated last week
• ROCm / hipBLASLt
  hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…
  ☆97Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆109Updated 10 months ago
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆234Updated 2 weeks ago
• ROCm / Tensile
  Stretching GPU performance for GEMMs and tensor contractions.
  ☆242Updated last week
• ROCm / rocprofiler
  ROC profiler library. Profiling with perf-counters and derived metrics.
  ☆148Updated last week
• ROCm / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆11Updated 11 months ago