Alternatives and similar repositories for apex

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

• ROCm / TransformerEngine
  ☆38Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆68Updated last month
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆67Updated last week
• yifuwang / symm-mem-recipes
  ☆90Updated 6 months ago
• codeplaysoftware / cutlass-sycl
  A CUTLASS implementation using SYCL
  ☆27Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆97Updated 2 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆109Updated 11 months ago
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆84Updated this week
• ROCm / pytorch-micro-benchmarking
  ☆20Updated 3 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆99Updated 11 months ago
• ROCm / MAD
  ☆19Updated last week
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆94Updated 6 years ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆40Updated 11 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 4 months ago
• ColfaxResearch / cutlass-kernels
  ☆212Updated 11 months ago
• bdhirsh / pytorch_open_registration_example
  Example of using pytorch's open device registration API
  ☆30Updated 2 years ago
• ROCm / Megatron-LM
  Ongoing research training transformer models at scale
  ☆23Updated 2 weeks ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆174Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆252Updated 8 months ago
• ROCm / rocSHMEM
  rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.
  ☆90Updated this week
• mlcommons / training_results_v1.0
  This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.
  ☆38Updated last year
• sunlex0717 / DissectingTensorCores
  ☆98Updated last year
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆125Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆58Updated this week
• AlibabaPAI / FLASHNN
  ☆97Updated 9 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆339Updated this week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆277Updated 3 weeks ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆116Updated 7 months ago
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆133Updated last year
• intel / torch-xpu-ops
  ☆48Updated this week