Alternatives and similar repositories for atex

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

• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆55Updated 5 months ago
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆32Updated 2 weeks ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆177Updated last year
• bytedance / xpu-perf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆298Updated 2 weeks ago
• PaddlePaddle / CINN
  Compiler Infrastructure for Neural Networks
  ☆147Updated 2 years ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆114Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆380Updated this week
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆254Updated last week
• NVIDIA-Merlin / HierarchicalKV
  HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…
  ☆192Updated this week
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆111Updated last week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆109Updated last year
• iree-org / iree-nvgpu
  ☆49Updated last year
• NVlabs / cub
  THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.
  ☆85Updated last year
• meta-pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆182Updated last month
• openucx / ucc
  Unified Collective Communication Library
  ☆290Updated last week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆148Updated 8 months ago
• BBuf / tensorrt-llm-moe
  ☆34Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 10 months ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆283Updated last month
• 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.
  ☆44Updated 11 months ago
• Oneflow-Inc / DLPerf
  DeepLearning Framework Performance Profiling Toolkit
  ☆296Updated 3 years ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆105Updated 7 years ago
• rapidsai / ucx-py
  Python bindings for UCX
  ☆139Updated 4 months ago
• triton-inference-server / onnxruntime_backend
  The Triton backend for the ONNX Runtime.
  ☆172Updated this week
• billmuch / matmul_perf_test
  ☆15Updated 3 years ago
• amd / ZenDNN
  ☆137Updated last week
• InternLM / turbomind
  ☆96Updated 10 months ago
• OpenPPL / ppl.llm.kernel.cuda
  ☆152Updated last year
• OpenPPL / ppl.llm.serving
  ☆130Updated last year
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆457Updated this week