ROCm / apex

Related projects ⓘ

Alternatives and complementary repositories for apex

• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆49Updated 2 weeks ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• ColfaxResearch / cfx-article-src
  ☆47Updated 2 weeks ago
• ColfaxResearch / cutlass-kernels
  ☆162Updated 4 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆87Updated 4 months ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆40Updated this week
• spcl / substation
  Research and development for optimizing transformers
  ☆124Updated 3 years ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆86Updated 2 weeks ago
• ROCm / TransformerEngine
  ☆12Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆96Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆196Updated 2 weeks ago
• yifuwang / symm-mem-recipes
  ☆11Updated last month
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated last week
• libxsmm / parlooper
  PArallelLOOPgEneratoR: Threaded Loops Code Generation Infrastructure targeting Tensor Contraction Applications such as GEMMs, Convolution…
  ☆18Updated last month
• HabanaAI / vllm-fork
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆41Updated this week
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆93Updated this week
• ROCm / pytorch-micro-benchmarking
  ☆16Updated last week
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated last 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…
  ☆57Updated 2 months ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆47Updated last week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆160Updated last week
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆78Updated this week
• NVIDIA / compute-sanitizer-samples
  Samples demonstrating how to use the Compute Sanitizer Tools and Public API
  ☆65Updated last year
• NVIDIA / HMM_sample_code
  CUDA 12.2 HMM demos
  ☆17Updated 3 months ago