Alternatives and similar repositories for apex:

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

• ROCm / TransformerEngine
  ☆26Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆91Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆60Updated 3 weeks ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆56Updated 2 weeks ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆172Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆54Updated 3 weeks ago
• ROCm / pytorch-micro-benchmarking
  ☆20Updated last week
• ROCm / rocmProfileData
  ☆21Updated last month
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆87Updated 6 years ago
• iree-org / iree-nvgpu
  ☆49Updated last year
• muriloboratto / NCCL
  Sample examples of how to call collective operation functions on multi-GPU environments. A simple example of using broadcast, reduce, all…
  ☆32Updated last year
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆243Updated 5 months ago
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆62Updated 3 weeks ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆93Updated 8 months ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆114Updated this week
• codeplaysoftware / cutlass-fork
  CUDA Templates for Linear Algebra Subroutines
  ☆16Updated this week
• yifuwang / symm-mem-recipes
  ☆60Updated 3 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆30Updated 2 months ago
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆79Updated last week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆95Updated this week
• HabanaAI / Habana_Custom_Kernel
  Provides the examples to write and build Habana custom kernels using the HabanaTools
  ☆21Updated 4 months ago
• ColfaxResearch / cutlass-kernels
  ☆193Updated 8 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆63Updated last week
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆227Updated this week
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆129Updated last year
• ROCm / rocSHMEM
  rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.
  ☆66Updated this week
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆111Updated 4 months ago
• ROCm / rdc
  RDC
  ☆27Updated this week
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆163Updated this week
• AlibabaPAI / FLASHNN
  ☆91Updated 6 months ago