ROCm / apex

Related projects ⓘ

Alternatives and complementary repositories for apex

• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆50Updated 3 weeks ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆41Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆86Updated 3 weeks ago
• ROCm / TransformerEngine
  ☆13Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆98Updated last week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆270Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆321Updated last month
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆201Updated 2 years ago
• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆209Updated 3 weeks ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• ROCm / pytorch-micro-benchmarking
  ☆16Updated last week
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 3 years ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆81Updated this week
• HabanaAI / vllm-fork
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆43Updated this week
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆313Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆30Updated 3 months ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 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…
  ☆57Updated 2 months ago
• yifuwang / symm-mem-recipes
  ☆12Updated last month
• manishucsd / py-codegen
  ☆14Updated last month
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆271Updated this week