Alternatives and similar repositories for apex

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

• ROCm / TransformerEngine
  ☆34Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆63Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆64Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆97Updated 3 weeks ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆122Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆109Updated 10 months ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆174Updated this week
• 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…
  ☆61Updated 2 months ago
• ROCm / rocmProfileData
  ☆24Updated last week
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆91Updated 6 years ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆248Updated 6 months ago
• ROCm / Megatron-LM
  Ongoing research training transformer models at scale
  ☆20Updated this week
• deepspeedai / DeepSpeed-Kernels
  ☆69Updated last month
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆76Updated this week
• yifuwang / symm-mem-recipes
  ☆76Updated 4 months ago
• ColfaxResearch / cfx-article-src
  ☆109Updated last week
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆158Updated last year
• ROCm / pytorch-micro-benchmarking
  ☆20Updated last month
• codeplaysoftware / cutlass-sycl
  A CUTLASS implementation using SYCL
  ☆22Updated this week
• ColfaxResearch / cutlass-kernels
  ☆202Updated 10 months ago
• ROCm / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆11Updated 10 months ago
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆36Updated 2 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆96Updated 10 months ago
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆55Updated last week
• AlibabaPAI / FLASHNN
  ☆94Updated 8 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 3 months ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆185Updated this week
• Bruce-Lee-LY / cuda_hgemv
  Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.
  ☆61Updated 8 months ago
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆89Updated 2 weeks ago