Alternatives and similar repositories for atex:

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

• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆112Updated 3 weeks ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆149Updated 6 months ago
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆43Updated 2 weeks ago
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆378Updated 3 weeks ago
• NVIDIA-Merlin / HierarchicalKV
  HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…
  ☆139Updated this week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆90Updated 7 months ago
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆79Updated 2 weeks ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆223Updated 3 weeks ago
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆223Updated last month
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆106Updated 3 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆105Updated 5 months ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆20Updated 4 years ago
• amd / ZenDNN
  ☆105Updated 3 months ago
• HabanaAI / SynapseAI_Core
  SynapseAI Core is a reference implementation of the SynapseAI API running on Habana Gaudi
  ☆38Updated last month
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆310Updated this week
• DefTruth / ffpa-attn-mma
  📚FFPA(Split-D): Yet another Faster Flash Prefill Attention with O(1)⚡️GPU SRAM complexity for headdim > 256, ~2x↑🎉vs SDPA EA.
  ☆123Updated last week
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆82Updated last year
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆29Updated last month
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆47Updated this week
• triton-inference-server / onnxruntime_backend
  The Triton backend for the ONNX Runtime.
  ☆139Updated this week
• ColfaxResearch / cfx-article-src
  ☆75Updated last week
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆304Updated this week
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆17Updated 2 years ago
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆210Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆54Updated 2 weeks ago
• iree-org / iree-nvgpu
  ☆49Updated last year
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆350Updated 2 weeks ago
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆71Updated this week
• ROCm / hipTensor
  AMD’s C++ library for accelerating tensor primitives
  ☆38Updated this week
• triton-lang / kernels
  ☆71Updated 3 months ago